"·T` ã@s×dZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZdd„Zddd „Zd d d „Zd d„Zdd„ZedkrÓeƒdS)zEDefine 'clean' utility and handler as part of cloud-init commandline.éN)ÚInit)ÚProcessExecutionErrorÚsubp)Údel_dirÚdel_fileÚget_config_logfilesÚis_linkcCstjjd|dƒdS)NzERROR: Ú )ÚsysÚstderrÚwrite)Úmsg©rú5/usr/lib/python3/dist-packages/cloudinit/cmd/clean.pyÚerrorsrc Cs”|stjddddƒ}|jddddd d d d d dƒ|jddddd d d dƒ|jddddd d d dd dƒ|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. ÚprogÚcleanZ descriptionzARemove logs and artifacts so cloud-init re-runs on a clean systemz-lz--logsÚactionÚ store_trueÚdefaultFÚdestÚ remove_logsÚhelpzRemove cloud-init logs.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.)ÚargparseÚArgumentParserÚ add_argument)ÚparserrrrÚ get_parsers   rFcCs:tdgƒ}|jƒ|rCx!t|jƒD]}t|ƒq/Wtjj|jj ƒs_dStjj |jj dƒ}x¹t j d|jj ƒD]ž}||kr°| r°q”y:tjj|ƒrßt |ƒ rßt |ƒn t|ƒWq”tk r1}z%tdj|t|ƒƒƒdSWYdd}~Xq”Xq”WdS)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}éN)rZread_cfgrZcfgrÚosÚpathÚisdirÚpathsZ cloud_dirÚjoinÚglobrrÚOSErrorrÚformatÚstr)rrZinitZlog_fileZ seed_pathr!ÚerrrÚremove_artifacts/s&    r*cCs›t|j|jƒ}|dkr—|jr—dddg}yt|ddƒWnGtk r–}z'tdj|t|ƒƒƒd}WYd d }~XnX|S) z3Handle calls to 'cloud-init clean' as a subcommand.rZshutdownz-rZnowZcaptureFz-Could not reboot this system using "{0}": {1}rN) r*rrZrebootrrrr'r()ÚnameÚargsZ exit_codeÚcmdr)rrrÚhandle_clean_argsOsr.cCs)tƒ}tjtd|jƒƒƒdS)z4Tool to collect and tar all cloud-init related logs.rN)rr Úexitr.Ú parse_args)rrrrÚmain^s r1Ú__main__)Ú__doc__rr%r r Zcloudinit.stagesrZcloudinit.subprrZcloudinit.utilrrrrrrr*r.r1Ú__name__rrrrÚs    "