File: //lib/python3/dist-packages/uaclient/daemon/__pycache__/retry_auto_attach.cpython-310.pyc
o
S��c� � @ s� d dl Z d dlZd dlZd dlmZmZmZmZ d dlmZ d dl
m Z m Z d dl
mZ d dlmZ d dlmZ e�d�Zg d �Zd
Zd ed efd
d�Zdefdd�Zded dfdd�ZdS )� N)�
exceptions�lock�messages�system)r )�FullAutoAttachOptions�full_auto_attach)�UAConfig)� AUTO_ATTACH_STATUS_MOTD_FILE)� state_filesz pro.daemon.retry_auto_attach)� r i i i i@8 �p r �Q r
� � r �� r � F r 逗 r z./run/ubuntu-advantage/flags/auto-attach-failed�e�returnc C s t | tj�r
tj�| j�S t | tj�rtjS t | tj �r$tj
j| j d�S t | tj �r1tj
�| j�S t | tj�r:tjS t | tj�rh| jrZ| jrQtjj| j| jd�}n tjj| jd�}ntj}|d�t| ��7 }|S t | tj�rtd�| j�S t�d�| �� t| �p�tj S )N)�pid)�code�url)r z: "{}"z"{}"zUnexpected exception: {}) �
isinstance�api_exceptions�InvalidProImager �$RETRY_ERROR_DETAIL_INVALID_PRO_IMAGE�format�contract_server_msg�NonAutoAttachImageError�(RETRY_ERROR_DETAIL_NON_AUTO_ATTACH_IMAGE�
LockHeldError� RETRY_ERROR_DETAIL_LOCK_HELDr �ContractAPIError�%RETRY_ERROR_DETAIL_CONTRACT_API_ERROR� api_error�ConnectivityError�%RETRY_ERROR_DETAIL_CONNECTIVITY_ERROR�UrlErrorr r �!RETRY_ERROR_DETAIL_URL_ERROR_CODE� RETRY_ERROR_DETAIL_URL_ERROR_URL�$RETRY_ERROR_DETAIL_URL_ERROR_GENERIC�str�UserFacingError�msg�LOG�error�RETRY_ERROR_DETAIL_UNKNOWN)r �failure_reason� r1 �C/usr/lib/python3/dist-packages/uaclient/daemon/retry_auto_attach.py�,full_auto_attach_exception_to_failure_reason( s< � � ��� r3 �cfgc C s2 t j�� t j�� t�t� | j�dt j
� d S )N� ) r
� retry_auto_attach_state_file�delete� retry_auto_attach_options_filer �ensure_file_absentr � notice_file�remover � AUTO_ATTACH_RETRY_NOTICE_PREFIX)r4 r1 r1 r2 �cleanupN s
r= c C s� | j rd S tj�� }|d ur|j}t|d � }|j}nd}t}d }t|�D ]�\}}tj� tj
j �}|tj |d� }|j
ddd�}tj�tj|| |d�� |} | d u rVtj} z|�� }W n tye Y nw tjj|| d | |�� d�}
t�t|
d � z(tj| dd �� | j � d
tj � | j � d
|
� W d � n1 s�w Y W n
t j!y� Y nw t"�#|� | j r� n`ztj$�� } t%� } | d ur�| j&| _&| j'| _'t(| � W nD t)j*y� t+�,d � Y n5 t)j-y� }
z
t+�.|
j/� W Y d }
~
n d }
~
w t�y }
zt0|
�}t+�1|
� W Y d }
~
q%d }
~
ww t2| � | j �sVtj�tjt3t�|d�� |} | d u �r9tj} tj4jt3t�d | d �}
t�t|
d � | j � d
|
� d S d S )
Nr )�seconds)�second� microsecond)�interval_indexr0 � )� num_attempts�reason�next_run_datestringz
z+pro.daemon.retry_auto_attach.notice_updates)r4 � lock_holderr5 z&already attached, ending retry service)rC rD )5� is_attachedr
r6 �readrA �RETRY_INTERVALSr0 � enumerate�datetime�now�timezone�utc� timedelta�replace�write�RetryAutoAttachStater r/ �
astimezone� Exception�AUTO_ATTACH_RETRY_NOTICEr � isoformatr �
write_filer r �SpinLockr: r; r< �addr r �time�sleepr8 r �enable� enable_betar r �AlreadyAttachedErrorr- �info�EntitlementsNotEnabledError�warningr, r3 r. r= �len�&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICE)r4 �persisted_state�offset� intervalsr0 �index�interval� last_attempt� next_attempt�
msg_reason�auto_attach_status_msg�persisted_options�optionsr r1 r1 r2 �retry_auto_attachU s�
�� �
���� ���
�
�����
����ro ) rK �loggingrZ �uaclientr r r r � uaclient.apir �2uaclient.api.u.pro.attach.auto.full_auto_attach.v1r r �uaclient.configr �uaclient.daemonr �uaclient.filesr
� getLoggerr- rI �FLAG_FILE_PATHrT r* r3 r= ro r1 r1 r1 r2 �<module> s
&