0x1949 Team - FAZEMRX - MANAGER
Edit File: retry_auto_attach.cpython-310.pyc
o �ϴfU � @ 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�rItjj| jd�d�t| j�� S t | tj�rUd�| j�S tjd| d� t| �pbtjS ) 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_URL_ERROR_URLr �str� cause_error�UbuntuProError�msg�LOG�error� UNKNOWN_ERROR)r � r1 �C/usr/lib/python3/dist-packages/uaclient/daemon/retry_auto_attach.py�,full_auto_attach_exception_to_failure_reason) s, � � �� r3 �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)r4 r1 r1 r2 �cleanupA 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�sitj�tjt4t�|d�� |} | d u �rCtj} 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)� lock_holderz&already attached, ending retry service)rD rE )7r � is_attachedr r5 �readrA �RETRY_INTERVALSrB � enumerate�datetime�now�timezone�utc� timedelta�replace�write�RetryAutoAttachStater r0 � astimezone� Exception�AUTO_ATTACH_RETRY_NOTICEr � isoformatr � write_filer r � RetryLockr �addr: r; r r# �time�sleepr7 r �enable� enable_betar r �AlreadyAttachedErrorr. �info�EntitlementsNotEnabledError�warningr- r3 r/ r= �len�&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICEr< )r4 �persisted_state�offset� intervalsrB �index�interval� last_attempt� next_attempt� msg_reason�auto_attach_status_msg�persisted_options�optionsr r1 r1 r2 �retry_auto_attachM s� �� � � �� � �� � � ��� �� �� � ��rq )"rM �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. rK �FLAG_FILE_PATHrV r* r3 r= rq r1 r1 r1 r2 �<module> s