
    }fU                         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mZmZ ddlmZ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)CLOUDINIT_NETPLAN_FILE)Init)ProcessExecutionErrorrunpartssubp)del_dirdel_fileerror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           
      P   | 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,        j.                         y# t"        $ r/}t%        dj'                  |t)        |                   Y d}~ y	d}~ww xY w# t0        $ r)}t%        d
t,        j.                   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   formatstrr   r   CLEAN_RUNPARTS_DIR	Exception)	initr   r   r   log_filer5   conf	seed_pathes	            r'   remove_artifactsrD   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=
,,-   	-44T3q6BC	
  *8+F+F*Gr!M	
 	s1   AF8G3 8	G0$G++G03	H%<H  H%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        $ r1}t        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)rG   delaymessageF)capturez-Could not reboot this system using "{0}": {1}r,   )r   rD   r   r   r   
machine_idr   r   ETC_MACHINE_IDr   rH   distroshutdown_commandr	   r   r   r;   r<   )nameargsr?   	exit_codecmdrC   s         r'   handle_clean_argsrU      s    D d 0 0$2D2DI >~'8uE ^$A~$++kk** + 
	e$ 9 % 	?FFQ
 I	s   !B2 2	C,;&C''C,c                  r    t               } t        j                  t        d| j	                                      y)z4Tool to collect and tar all cloud-init related logs.r   N)r(   sysexitrU   
parse_argsr%   s    r'   mainrZ      s&    \FHHw(9(9(;<=r)   __main__)N)FN) __doc__r"   r2   r4   rW   	cloudinitr   cloudinit.distrosr   cloudinit.net.netplanr   cloudinit.stagesr   cloudinit.subpr   r   r	   cloudinit.utilr
   r   r   r   r   r   rN   r1   r3   r(   rD   rU   rZ   __name__ r)   r'   <module>re      s    M   	 
  * 8 ! @ @  #3EF21  0 
@F/d:> zF r)   