o i@s dZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZdd lmZmZmZdd lmZmZmZmZmZeeZ d Z!ed d dddgZ"dgZ#de$de%fddZ&d ddZ'd!ddZ(ddZ)ddZ*edkre*dSdS)"zFDefine 'clean' utility and handler as part of cloud-init command line.N)settingssources) cc_mounts) uses_systemd)log_util)CLOUDINIT_NETPLAN_FILE)Init)ProcessExecutionErrorrunpartssubp)del_dirdel_fileget_config_logfilesis_link write_filez/etc/machine-idz-/etc/NetworkManager/conf.d/99-cloud-init.confz?/etc/NetworkManager/conf.d/30-cloud-init-ip6-addr-gen-mode.confz@/etc/NetworkManager/system-connections/cloud-init-*.nmconnectionz,/etc/systemd/network/10-cloud-init-*.networkz+/etc/network/interfaces.d/50-cloud-init.cfgz)/etc/ssh/sshd_config.d/50-cloud-init.conflog_filereturncCsDzt|}t|jst|jrWdSWdSty!YdSw)zCheck if a log file should be removed. Avoid tracebacks from attempting to remove device files. @param log_file: Path to the log file to check. @returns: True if the file should be removed, False otherwise. FT)osstatS_ISBLKst_modeS_ISCHROSError)r file_statr5/usr/lib/python3/dist-packages/cloudinit/cmd/clean.pyshould_remove_log_file.s  rc Cs|s tjddd}|jdddddd d |jd ddd d |jdddddd |jddddddd |jddgdgdddd|S)a#Build or extend an arg parser for clean utility. @param parser: Optional existing ArgumentParser instance representing the clean subcommand which will be extended to support the args of this utility. @returns: ArgumentParser with proper argument configuration. cleanzJRemove logs, configs and artifacts so cloud-init re-runs on a clean system)prog descriptionz-lz--logs store_trueF remove_logszRemove cloud-init logs.)actiondefaultdesthelpz --machine-idzSet /etc/machine-id to 'uninitialized\n' for golden image creation. On next boot, systemd generates a new machine-id. Remove /etc/machine-id on non-systemd environments.)r"r#r%z-rz--rebootz;Reboot system after logs are cleaned so cloud-init re-runs.z-sz--seed remove_seedz5Remove cloud-init seed directory /var/lib/cloud/seed.z-cz --configs)all ssh_confignetwork datasourcefstab+ remove_configzbRemove cloud-init generated config files of a certain type. Config types: all, ssh_config, network)choicesr#nargsr$r%)argparseArgumentParser add_argumentparserrrr get_parser?sX  r5Fc Cs||rt|jD] }t|rt|q |r2t|ddgr2tD]}t|D]}t|q*q#|rFt|ddgrFt D]}t|q?|rUt|ddgrUt |o_t|ddg}t j |jjsstjdttjddS|rz|Wntjytjd ttjdYnwt j |jjd }td |jjD]=}||kr|sqzt j |rt|st|nt|Wqty} ztd |t | WYd } ~ dSd } ~ wwzt!t"j#WdSt$y } ztdt"j#d| WYd } ~ dSd } ~ ww)aHelper which removes artifacts dir and optionally log files. @param: init: Init object to use @param: remove_logs: Boolean. Set True to delete the cloud_dir path. False preserves them. @param: remove_seed: Boolean. Set True to also delete seed subdir in paths.cloud_dir. @param: remove_config: List of strings. Can be any of: all, network, ssh_config, datasource, fstab. @returns: 0 on success, 1 otherwise. r'r)r(r+r*zArtifacts already cleaned.)log log_levelrz%No datasource found, nothing cleaned.seedz%s/*zCould not remove {0}: {1}NzFailure during run-parts of z: )%read_cfgrcfgrr set intersectionGEN_NET_CONFIG_FILESglobGEN_SSH_CONFIG_FILESr cleanup_fstabrpathisdirpaths cloud_dirr multi_logLOGloggingINFOfetchrrDataSourceNotFoundExceptionjoinrr rerrorformatstrr rCLEAN_RUNPARTS_DIR Exception) initr!r&r-rrBconfclean_datasource seed_patherrrremove_artifactss|         rWc Cstgd}t||j|j|j}|jr!trttdddnt t|dkr[|j r[|j j dddd }z t |d d W|StyZ}ztd |t|d }WYd}~|Sd}~ww|S)z3Handle calls to 'cloud-init clean' as a subcommand.)ds_depszuninitialized i$)moderrebootnowN)rYdelaymessageF)capturez-Could not reboot this system using "{0}": {1}r9)rrWr!r&r- machine_idrrETC_MACHINE_IDr rZdistroshutdown_commandr r rrMrNrO)nameargsrR exit_codecmdrVrrrhandle_clean_argss4 rgcCst}ttd|dS)z4Tool to collect and tar all cloud-init related logs.rN)r5sysexitrg parse_argsr3rrrmainsrk__main__)N)FN)+__doc__r0r?rHrrrh cloudinitrrcloudinit.configrcloudinit.distrosr cloudinit.logrcloudinit.net.netplanrcloudinit.stagesrcloudinit.subpr r r cloudinit.utilr r rrr getLogger__name__rGr`r>r@rOboolrr5rWrgrkrrrrsD         EI