OwlCyberSecurity - MANAGER
Edit File: retry_auto_attach.cpython-38.pyc
U UٵgU � @ 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 �