
    쑛h                         d Z ddlZddlZddlZddlZddlZddlmZmZ ddl	m
Z
 ddlmZ ddlmZ ddlmZ ddlmZ dd	lmZmZmZ dd
lmZmZmZmZmZ  ej:                  e      ZdZ edddddgZ!dgZ"ddZ#ddZ$d Z%d Z&edk(  r e&        yy)zFDefine 'clean' utility and handler as part of cloud-init command line.    N)settingssources)	cc_mounts)uses_systemd)log_util)CLOUDINIT_NETPLAN_FILE)Init)ProcessExecutionErrorrunpartssubp)del_dirdel_fileget_config_logfilesis_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.confc           	         | st        j                  dd      } | j                  dddddd	
       | j                  dddd       | j                  ddddd       | j                  dddddd
       | j                  ddg dg dd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.
    cleanzJRemove logs, configs and artifacts so cloud-init re-runs on a clean system)progdescriptionz-lz--logs
store_trueFremove_logszRemove cloud-init logs.)actiondefaultdesthelpz--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.)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_confignetwork
datasourcefstab+remove_configzbRemove cloud-init generated config files of a certain type. Config types: all, ssh_config, network)choicesr   nargsr   r   )argparseArgumentParseradd_argumentparsers    5/usr/lib/python3/dist-packages/cloudinit/cmd/clean.py
get_parserr,   -   s     (($
 &   C  	 J   D   
 2    M    c           
      :   | j                          |r%t        | j                        D ]  }t        |        |rLt	        |      j                  ddg      r0t        D ]'  }t        j                  |      D ]  }t        |        ) |r2t	        |      j                  ddg      rt        D ]  }t        |        |r0t	        |      j                  ddg      rt        j                          |xr t	        |      j                  ddg      }t        j                  j                  | j                  j                        s+t!        j"                  dt$        t&        j(                         y|r	 | j+                         j-                          t        j                  j3                  | j                  j                  d
      }t        j                  d| j                  j                  z        D ]L  }||k(  r|s	 t        j                  j                  |      rt5        |      st7        |       nt        |       N 	 tA        tB        jD                         y# t.        j0                  $ r. t!        j"                  d	t$        t&        j(                         Y 
w xY w# t8        $ r9}	t!        j:                  dj=                  |t?        |	                   Y d}	~	 yd}	~	ww xY w# tF        $ r3}	t!        j:                  dtB        jD                   d|	        Y d}	~	yd}	~	ww xY w)a  Helper 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    zArtifacts already cleaned.)log	log_levelr   z%No datasource found, nothing cleaned.seedz%s/*zCould not remove {0}: {1}N   zFailure during run-parts of z: )$read_cfgr   cfgr   setintersectionGEN_NET_CONFIG_FILESglobGEN_SSH_CONFIG_FILESr   cleanup_fstabospathisdirpaths	cloud_dirr   	multi_logLOGloggingINFOfetchr   r   DataSourceNotFoundExceptionjoinr   r   OSErrorerrorformatstrr   r   CLEAN_RUNPARTS_DIR	Exception)
initr   r   r#   log_filer<   confclean_datasource	seed_pathes
             r+   remove_artifactsrS   r   s    	MMO+DHH5 	HX	]+88%9KL( 	D		$ 	 ]+88	 ) 	DTN	]+88%9IJ!$ ]);)H)H	* 77==--.(ll	

 	JJL  TZZ116:I		&4::#7#778 
9[	ww}}T"74=
,,- 5 22 	7!,, 	   	NN6==dCFKL	
  *8+F+F*Gr!M	
 	sC   -I 7AJ;K =JJ	K".KK	L')LLc           	         t        g       }t        ||j                  |j                  |j                        }|j
                  r,t               rt        t        dd       nt        t               |dk(  r:|j                  r.|j                  j                  ddd	      }	 t        |d
       |S |S # t        $ r;}t        j                   dj#                  |t%        |                   d}Y d}~|S d}~ww xY w)z3Handle calls to 'cloud-init clean' as a subcommand.)ds_depszuninitialized
i$  )moder   rebootnowN)rV   delaymessageF)capturez-Could not reboot this system using "{0}": {1}r2   )r	   rS   r   r   r#   
machine_idr   r   ETC_MACHINE_IDr   rW   distroshutdown_commandr   r
   r   rH   rI   rJ   )nameargsrM   	exit_codecmdrR   s         r+   handle_clean_argsrd      s    D d 0 0$2D2DI >~'8uE ^$A~$++kk** + 
	e$ 9 % 	NN?FFQ
 I	s   !B2 2	C6;0C11C6c                  r    t               } t        j                  t        d| j	                                      y)z4Tool to collect and tar all cloud-init related logs.r   N)r,   sysexitrd   
parse_argsr)   s    r+   mainri      s&    \FHHw(9(9(;<=r-   __main__)N)FN)'__doc__r&   r8   rB   r;   rf   	cloudinitr   r   cloudinit.configr   cloudinit.distrosr   cloudinit.logr   cloudinit.net.netplanr   cloudinit.stagesr	   cloudinit.subpr
   r   r   cloudinit.utilr   r   r   r   r   	getLogger__name__rA   r]   r7   r9   r,   rS   rd   ri    r-   r+   <module>rw      s    M    	 
 ' & * " 8 ! @ @  g!"3EF21  0 
BJEP:> zF r-   