
    Uٵg	                     6   d dl Z d dlmZmZmZmZmZmZ d dlm	Z	 d dl
mZmZmZ d dlmZ d dlmZmZ  e j&                   ej(                  e            Zdej.                  fdZ e	j2                         dej.                  fd	       Zdej.                  fd
Ze	j8                   e	j:                  d      dej.                  fd              Z edej>                  ej@                  edejB                   e edejD                  dg dd      g      g      Z#y)    N)apt_newsconfigcontract
exceptionsmessagesutil)cli_util)ProArgumentProArgumentGroup
ProCommand)HelpCategory)refresh_motdupdate_motd_messagescfgc                     	 |j                          t        t        j                         y # t        $ r.}t        j                  |       t	        j
                         d }~ww xY wN)	process_configRuntimeErrorLOG	exceptionr   RefreshConfigFailureprintr   REFRESH_CONFIG_SUCCESS)argsr   excs      6/usr/lib/python3/dist-packages/uaclient/cli/refresh.py_action_refresh_configr      sP    0 
(
)
)*  0c--//0s   , 	A#)AA#c                     	 t        j                  |       t        t        j                         y # t        j                  $ r t        j                         w xY wr   )r   refreshr   ConnectivityErrorRefreshContractFailurer   r   REFRESH_CONTRACT_SUCCESS)_argsr   s     r   _action_refresh_contractr$      sK    2 
(
+
+, '' 2//112s	   1 (Ac                    	 t        |       t                |j                  rt        j                  |       t	        t
        j                         y # t        $ r.}t        j                  |       t        j                         d }~ww xY wr   )r   r   r   update_apt_newsr   r   REFRESH_MESSAGES_SUCCESS	Exceptionr   r   r   RefreshMessagesFailure)r#   r   r   s      r   _action_refresh_messagesr*      sf    	1S!<<$$S)
 	h//0	  2c//112s   6A 	B	)BB	zpro refreshc                    | j                   | j                   dk(  rt        | |       | j                   | j                   dk(  rt        | |       | j                   | j                   dk(  rt        | |       y)Nr   r   r   r   )targetr   r$   r*   )r   r   kwargss      r   action_refreshr.   .   sd     {{dkkX5tS){{dkkZ7 s+{{dkkZ7 s+    r   Tr,   ?)r   r   r   )helpnargschoicesdefault)	arguments)r1   descriptionactionpreserve_descriptionhelp_categoryargument_groups)$logginguaclientr   r   r   r   r   r   uaclient.clir	   uaclient.cli.commandsr
   r   r   uaclient.cli.parserr   uaclient.timer.update_messagingr   r   	getLoggerreplace_top_level_logger_name__name__r   UAConfigr   assert_attachedr$   r*   assert_rootassert_lock_filer.   CLI_ROOT_REFRESHCLI_REFRESH_DESCOTHERCLI_REFRESH_TARGETrefresh_command r/   r   <module>rN      s     K K ! K K , Ng:::8DE+foo + - - -1 1  
=)
 
 * 
 		"	"))$$!44> 
	
r/   