File: //lib/python3/dist-packages/uaclient/daemon/__pycache__/retry_auto_attach.cpython-38.pyc
U
��JhU � @ s� d dl Z d dlZd dlZd dlmZmZmZmZmZ d dl mZ
d dl m Z m
Z
d dlmZ d dlmZ d dlmZ d dlmZmZ e�e�e��Zd d d
d d d
ddddddddddddgZdZ e e d�dd�Z ed�dd�Z edd�d d �Z!dS ) � N)�
exceptions�lock�messages�system�util)r )�FullAutoAttachOptions�full_auto_attach)� _is_attached)�UAConfig)� AUTO_ATTACH_STATUS_MOTD_FILE)�notices� state_filesi� 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�r tjj| jd�S t | tj�r.tjS t | tj �rJtj
j| j d�S t | tj �rftj
j| jd�S t | tj�r�tjj| jd�d�t| j�� S t | tj�r�d�| j�S tjd| d� t| �p�tjS d S ) N)Zdetail)�pid)� error_msg)�urlz: "{}"z"{}"zUnexpected exception)�exc_info)�
isinstance�api_exceptionsZInvalidProImager Z$RETRY_ERROR_DETAIL_INVALID_PRO_IMAGE�formatr ZNonAutoAttachImageErrorZ(RETRY_ERROR_DETAIL_NON_AUTO_ATTACH_IMAGE�
LockHeldErrorZ RETRY_ERROR_DETAIL_LOCK_HELDr ZContractAPIErrorZ%RETRY_ERROR_DETAIL_CONTRACT_API_ERRORZbodyZConnectivityErrorZ RETRY_ERROR_DETAIL_URL_ERROR_URLr �strZ cause_errorZUbuntuProError�msg�LOG�error�
UNKNOWN_ERROR)r � 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 �cleanupA s
��r) )r! r c C s� t | �jrd S tj�� }|d k r:|j}t|d � }|j}n d}t}d }t|�D �]\}}t j �
t j j �}|t j
|d� }|jddd�}tj�tj|| |d�� |} | d kr�tj} z |�� }W n tk
r� Y nX tjj|| d | |�� d�}
t�td|
d � z>tj d d
��( t j t j j || d | |�� d� W 5 Q R X W n t!j"k
�r\ Y nX t#�$|� t | �j�rz �qVz>tj%�� } t&� } | d k �r�| j'| _'| j(| _(t)| � W �qVW qN t*j+k
�r� t,�-d � Y �qVY qN t*j.k
�r }
z t,�/|
j0� W Y �
�qVW 5 d }
~
X Y qN tk
�rR }
zt1|
�}t,�2|
� W 5 d }
~
X Y qNX qNt3| � t | �j�s�tj�tjt4t�|d�� |} | d k�r�tj} tj5jt4t�d | d �}
t�td|
d � t j t j j6t4t�d | d � d S )
Nr )Zseconds)�secondZ microsecond)�interval_index�failure_reason� )� num_attempts�reasonZnext_run_datestring�
z
z+pro.daemon.retry_auto_attach.notice_updates)Z lock_holderz&already attached, ending retry service)r. r/ )7r Z 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 Z RetryLockr �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_attachM s�
��
�
��
�
�
��
��
�
�rA )"r4 Zloggingr: Zuaclientr r r r r Z uaclient.apir Z2uaclient.api.u.pro.attach.auto.full_auto_attach.v1r r Z(uaclient.api.u.pro.status.is_attached.v1r Zuaclient.configr
Zuaclient.daemonr Zuaclient.filesr r
Z getLoggerZ replace_top_level_logger_name�__name__r r2 ZFLAG_FILE_PATHr8 r r r) rA r r r r �<module> sB �