File: //usr/lib/python3/dist-packages/uaclient/daemon/__pycache__/retry_auto_attach.cpython-310.pyc
o
nHJe� � @ 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��Zg d �Zd
Z d e d e fd
d�Z defdd�Z ded dfdd�Z!dS )� N)�
exceptions�lock�messages�system�util)r )�FullAutoAttachOptions�full_auto_attach)� _is_attached)�UAConfig)� AUTO_ATTACH_STATUS_MOTD_FILE)�notices� state_files)� 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�rtjj| jd�S t | tj�rtjS t | tj �r%tj
j| j d�S t | tj �r3tj
j| jd�S t | tj�r<tjS t | tj�rQtjj| jd�d�t| �� S t | tj�r]d�| j�S tjd| d� t| �pjtjS ) N)�detail)�pid)� error_msg)�urlz: "{}"z"{}"zUnexpected exception)�exc_info)�
isinstance�api_exceptions�InvalidProImager �$RETRY_ERROR_DETAIL_INVALID_PRO_IMAGE�formatr �NonAutoAttachImageError�(RETRY_ERROR_DETAIL_NON_AUTO_ATTACH_IMAGE�
LockHeldError� RETRY_ERROR_DETAIL_LOCK_HELDr �ContractAPIError�%RETRY_ERROR_DETAIL_CONTRACT_API_ERROR�body�ConnectivityError�%RETRY_ERROR_DETAIL_CONNECTIVITY_ERROR�UrlError� RETRY_ERROR_DETAIL_URL_ERROR_URLr �str�UbuntuProError�msg�LOG�error�RETRY_ERROR_DETAIL_UNKNOWN)r � r2 �C/usr/lib/python3/dist-packages/uaclient/daemon/retry_auto_attach.py�,full_auto_attach_exception_to_failure_reason) s0 � � � � r4 �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 �ensure_file_absentr r �remove�Notice� AUTO_ATTACH_RETRY_FULL_NOTICE� AUTO_ATTACH_RETRY_TOTAL_FAILURE)r5 r2 r2 r3 �cleanupC s
��r> c C s� t | �jrd S tj�� }|d ur |j}t|d � }|j}nd}t}d }t|�D ]�\}}t j �
t j j �}|t j
|d� }|jddd�}tj�tj|| |d�� |} | d u rXtj} z|�� }W n tyg Y nw tjj|| d | |�� d�}
t�td|
d � z*tj | d d
�� t j t j j || d | |�� d� W d � n1 s�w Y W n
t!j"y� Y nw t#�$|� t | �jr� naztj%�� } t&� } | d ur�| j'| _'| j(| _(t)| � W nE t*j+y� t,�-d � Y n6 t*j.�y }
z
t,�/|
j0� W Y d }
~
n d }
~
w t�y" }
zt1|
�}t,�2|
� W Y d }
~
q'd }
~
ww t3| � t | �j�sjtj�tjt4t�|d�� |} | d u �rDtj} tj5jt4t�d | d �}
t�td|
d � t j t j j6t4t�d | d � d S d S )
Nr )�seconds)�second� microsecond)�interval_index�failure_reason� )� num_attempts�reason�next_run_datestring�
z
z+pro.daemon.retry_auto_attach.notice_updates)r5 � lock_holderz&already attached, ending retry service)rE rF )7r � is_attachedr
r6 �readrB �RETRY_INTERVALSrC � enumerate�datetime�now�timezone�utc� timedelta�replace�write�RetryAutoAttachStater r1 �
astimezone� Exception�AUTO_ATTACH_RETRY_NOTICEr � isoformatr �
write_filer r �SpinLockr �addr; r< r r# �time�sleepr8 r �enable� enable_betar r �AlreadyAttachedErrorr/ �info�EntitlementsNotEnabledError�warningr. r4 r0 r> �len�&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICEr= )r5 �persisted_state�offset� intervalsrC �index�interval� last_attempt� next_attempt�
msg_reason�auto_attach_status_msg�persisted_options�optionsr r2 r2 r3 �retry_auto_attachO s�
�� �
�
��
� ��
�
�
��� ��
��
�
��rr )"rN �loggingr] �uaclientr r r r r � uaclient.apir �2uaclient.api.u.pro.attach.auto.full_auto_attach.v1r r �(uaclient.api.u.pro.status.is_attached.v1r �uaclient.configr
�uaclient.daemonr �uaclient.filesr r
� getLogger� replace_top_level_logger_name�__name__r/ rL �FLAG_FILE_PATHrW r, r4 r> rr r2 r2 r2 r3 �<module> s