
    x[h                         d Z ddlZddlZddlZddl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 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)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+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           
      x   | 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        |        t        j                  j                  | j                  j                        syt        j                  j                  | j                  j                  d      }t        j                  d| j                  j                  z        D ]L  }||k(  r|s	 t        j                  j                  |      rt        |      st!        |       nt        |       N 	 t-        t.        j0                         y# t"        $ r9}t%        j&                  dj)                  |t+        |                   Y d}~ y	d}~ww xY w# t2        $ r3}t%        j&                  d
t.        j0                   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.
    @returns: 0 on success, 1 otherwise.
    r   r   r   r   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_FILESospathisdirpaths	cloud_dirjoinr   r   OSErrorr   errorformatstrr	   r   CLEAN_RUNPARTS_DIR	Exception)	initr   r   r   log_filer5   conf	seed_pathes	            r'   remove_artifactsrE   m   s    	MMO+DHH5 	HX	]+88%9KL( 	D		$ 	 ]+88	 ) 	DTN	 77==--.TZZ116:I		&4::#7#778 
9[	ww}}T"74=
,,-   	NN6==dCFKL	
  *8+F+F*Gr!M	
 	s1   AF8G= 8	G:.G55G:=	H9)H44H9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)rH   delaymessageF)capturez-Could not reboot this system using "{0}": {1}r,   )r   rE   r   r   r   
machine_idr   r   ETC_MACHINE_IDr   rI   distroshutdown_commandr
   r   r   r;   r<   r=   )nameargsr@   	exit_codecmdrD   s         r'   handle_clean_argsrV      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exitrV   
parse_argsr%   s    r'   mainr[      s&    \FHHw(9(9(;<=r)   __main__)N)FN)!__doc__r"   r2   r4   rX   	cloudinitr   cloudinit.distrosr   cloudinit.logr   cloudinit.net.netplanr   cloudinit.stagesr   cloudinit.subpr   r	   r
   cloudinit.utilr   r   r   r   r   rO   r1   r3   r(   rE   rV   r[   __name__ r)   r'   <module>rg      s    M   	 
  * " 8 ! @ @  #3EF21  0 
@F/d:> zF r)   