Your IP : 216.73.216.52
U
8�-d� � @ s� d dl Z d dlZd dlZd dlmZmZmZmZ d dlmZ d dl
m Z m Z d dl
mZ d dlmZ d dlmZmZ e�d�Zd d d
d d d
ddddddddddddgZdZeed�dd�Zed�dd�Zedd�d d �Z dS ) � N)�
exceptions�lock�messages�system)r )�FullAutoAttachOptions�full_auto_attach)�UAConfig)� AUTO_ATTACH_STATUS_MOTD_FILE)�notices� state_filesz pro.daemon.retry_auto_attachi� i i i i@8 i�p i�Q i � i�� i F i�� z./run/ubuntu-advantage/flags/auto-attach-failed)�e�returnc C s t | tj�rtj�| j�S t | tj�r,tjS t | tj �rHtj
j| j d�S t | tj �rbtj
�| j�S t | tj�rttjS t | tj�r�| jr�| jr�tjj| j| jd�}q�tjj| jd�}ntj}|d�t| ��7 }|S t | tj�r�d�| j�S t�d�| �� t| ��ptj S d S )N)�pid)�code�url)r z: "{}"z"{}"zUnexpected exception: {}) �
isinstance�api_exceptionsZInvalidProImager Z$RETRY_ERROR_DETAIL_INVALID_PRO_IMAGE�formatZcontract_server_msgZNonAutoAttachImageErrorZ(RETRY_ERROR_DETAIL_NON_AUTO_ATTACH_IMAGE�
LockHeldErrorZ RETRY_ERROR_DETAIL_LOCK_HELDr ZContractAPIErrorZ%RETRY_ERROR_DETAIL_CONTRACT_API_ERRORZ api_errorZConnectivityErrorZ%RETRY_ERROR_DETAIL_CONNECTIVITY_ERRORZUrlErrorr r Z!RETRY_ERROR_DETAIL_URL_ERROR_CODEZ RETRY_ERROR_DETAIL_URL_ERROR_URLZ$RETRY_ERROR_DETAIL_URL_ERROR_GENERIC�strZUserFacingError�msg�LOG�error�RETRY_ERROR_DETAIL_UNKNOWN)r �failure_reason� r �C/usr/lib/python3/dist-packages/uaclient/daemon/retry_auto_attach.py�,full_auto_attach_exception_to_failure_reason( s> � � ��� r ��cfgc C s> t j�� t j�� t�t� t�tj j
� t�tj j � d S )N) r � retry_auto_attach_state_file�delete� retry_auto_attach_options_filer Zensure_file_absentr r
�remove�Notice� AUTO_ATTACH_RETRY_FULL_NOTICE� AUTO_ATTACH_RETRY_TOTAL_FAILUREr r r r �cleanupN s
��r' )r r
c C s� | j r
d S tj�� }|d k r6|j}t|d � }|j}n d}t}d }t|�D �]�\}}tj� tj
j �}|tj |d� }|j
ddd�}tj�tj|| |d�� |} | d kr�tj} z |�� }W n tk
r� Y nX tjj|| d | |�� d�}
t�t|
d � z@tj| dd ��( t j t j j || d | |�� d� W 5 Q R X W n t j!k
�rV Y nX t"�#|� | j �rp �qLz>tj$�� } t%� } | d k �r�| j&| _&| j'| _'t(| � W �qLW qJ t)j*k
�r� t+�,d
� Y �qLY qJ t)j-k
�r }
z t+�.|
j/� W Y �
�qLW 5 d }
~
X Y qJ tk
�rH }
zt0|
�}t+�1|
� W 5 d }
~
X Y qJX qJt2| � | j �s�tj�tjt3t�|d�� |} | d k�r�tj} tj4jt3t�d | d �}
t�t|
d � t j t j j5t3t�d | d � d S ) Nr )Zseconds)�secondZ microsecond)�interval_indexr � )� num_attempts�reasonZnext_run_datestringz
z+pro.daemon.retry_auto_attach.notice_updates)r Z lock_holderz&already attached, ending retry service)r+ r, )6Z is_attachedr r �readr) �RETRY_INTERVALSr � enumerate�datetimeZnow�timezoneZutcZ timedelta�replace�writeZRetryAutoAttachStater r Z
astimezone� ExceptionZAUTO_ATTACH_RETRY_NOTICEr Z isoformatr Z
write_filer r ZSpinLockr
�addr$ r% r r �time�sleepr" r �enableZ enable_betar r ZAlreadyAttachedErrorr �infoZEntitlementsNotEnabledErrorZwarningr r r r' �lenZ&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICEr& )r Zpersisted_state�offsetZ intervalsr �indexZintervalZ last_attemptZ next_attemptZ
msg_reasonZauto_attach_status_msgZpersisted_optionsZoptionsr r r r �retry_auto_attachZ s�
��
� ��
�
�
��
�� �
�r= ) r0 Zloggingr6 Zuaclientr r r r Z uaclient.apir Z2uaclient.api.u.pro.attach.auto.full_auto_attach.v1r r Zuaclient.configr Zuaclient.daemonr Zuaclient.filesr
r Z getLoggerr r. ZFLAG_FILE_PATHr4 r r r' r= r r r r �<module> s@
�&