U Ld@sdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZdZdd d Zdd d ZddZddZedkredS)zEDefine 'clean' utility and handler as part of cloud-init commandline.N)settings) uses_systemd)Init)ProcessExecutionErrorrunpartssubp)del_dirdel_fileerrorget_config_logfilesis_link write_filez/etc/machine-idcCsh|stjddd}|jdddddd d |jd ddd d |jdddddd |jddddddd |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. cleanzARemove logs and artifacts so cloud-init re-runs on a clean system)progZ descriptionz-lz--logs store_trueF remove_logszRemove cloud-init logs.)actiondefaultdesthelpz --machine-idzSet /etc/machine-id to 'uninitialized ' for golden imagecreation. On next boot, systemd generates a new machine-id. Remove /etc/machine-id on non-systemd environments.)rrr-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.)argparseArgumentParser add_argumentparserr5/usr/lib/python3/dist-packages/cloudinit/cmd/clean.py get_parsersF  rFc Cs8tgd}||r.t|jD] }t|q tj|jj sBdStj |jj d}t d|jj D]|}||krx|sxqfz*tj|rt |st |nt|Wqftk r}z td|t|WYdSd}~XYqfXqfzttjWn@tk r2}z tdtjd |WYdSd}~XYnXdS) a4Helper which removes artifacts dir and optionally log files. @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. @returns: 0 on success, 1 otherwise. )Zds_depsrZseedz%s/*zCould not remove {0}: {1}NzFailure during run-parts of z: )rZread_cfgr Zcfgr ospathisdirpathsZ cloud_dirjoinglobr rOSErrorr formatstrrrZCLEAN_RUNPARTS_DIR Exception)rrZinitZlog_fileZ seed_pathr"errrremove_artifactsSs4     r,c Cst|j|j}|jr2tr*ttdddntt|dkr|jrdddg}zt |dd Wn:t k r}zt d |t |d }W5d }~XYnX|S) z3Handle calls to 'cloud-init clean' as a subcommand.zuninitialized i$)moderZshutdownrZnowF)Zcapturez-Could not reboot this system using "{0}": {1}r N)r,rrZ machine_idrr ETC_MACHINE_IDr Zrebootrrr r(r))nameargsZ exit_codecmdr+rrrhandle_clean_argszs$ r2cCst}ttd|dS)z4Tool to collect and tar all cloud-init related logs.rN)rsysexitr2 parse_argsrrrrmainsr6__main__)N)F)__doc__rr&r!r3Z cloudinitrZcloudinit.distrosrZcloudinit.stagesrZcloudinit.subprrrZcloudinit.utilrr r r r r r.rr,r2r6__name__rrrrs      5 '