o th @sUdZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZd Zd gd egd Ze ed <eeZdede dededdf ddZddZdededefddZdS)zSSH Import ID: Import SSH idN)subputil)Cloud)Config) MetaSchema)ug_util) PER_INSTANCEz ssh-import-idcc_ssh_import_id)alpinecosdebianzraspberry-pi-osubuntu)iddistros frequencyactivate_by_schema_keysmetanamecfgcloudargsreturnc Cs|t|ds tddSttstddS|r3|d}g}t|dkr,|dd}t||dSt ||j \}}g}| D]q\}} g} | drTt |dg} nz| d} Wntyitd|YqBwzt | } dd | D} Wntytd |YqBwt| sqBzt| |WqBty} zt td || || WYd} ~ qBd} ~ wwt|r|ddS) N ssh_import_idzISkipping module named ssh_import_id, no 'ssh_import_id' directives found.zXssh-import-id is not installed, but module ssh_import_id is configured. Skipping module.rdefaultz+User %s is not configured for ssh_import_idcSsg|]}t|qS)str).0irrC/usr/lib/python3/dist-packages/cloudinit/config/cc_ssh_import_id.py Kszhandle..z5User %s is not correctly configured for ssh_import_idzssh-import-id failed for: %s %s)is_key_in_nested_dictLOGdebugrwhichSSH_IMPORT_ID_BINARYwarninglenimport_ssh_idsrnormalize_users_groupsdistroitemsrget_cfg_option_list Exception uniq_mergelogexcappend) rrrruseridsusers_groupselistuser_cfg import_idsexcrrrhandle"sh          r9c Cs|r|s td||dSzt|Wnty"}z|d}~wwtdr2ddd|tg|}ntdr@dd|tg|}ntddStd|z tj|d d WdStj ym}z t td ||d}~ww) Nz(empty user(%s) or ids(%s). not importingsudoz--preserve-env=https_proxyz-Hudoasz-uz:Neither sudo nor doas available! Unable to import SSH ids.zImporting SSH ids for user %s.F)capturez*Failed to run command to import %s SSH ids) r"r#pwdgetpwnamKeyErrorrr$r%errorProcessExecutionErrorrr/)r2r1r8cmdrrrr(asH    r(config search_keycCsz|D]6}||kr dSt||trt|||rdSt||tr:||D]}t|tr9t||r9dSq)qdS)zSearch for key nested in config. Note: A dict embedded in a list of lists will not be found walked - but in this case we don't need it. TF)keys isinstancedictr!list)rCrD config_keyitemrrrr!s    r!)__doc__loggingr= cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.distrosrcloudinit.settingsrr%r__annotations__ getLogger__name__r"rrHr9r(rGboolr!rrrrs&      ??