Your IP : 216.73.216.14
U
8�-d� � @ sr d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZmZmZ d d lmZ e�d
�Z d Z ej e e dd �d
d�Z!d(ej ejdd�dd�Z"dddd�ej e e e e d�dd�Z#ddd�ej ee e d�dd�Zd)e eee$ dd �d d �Z%ej d �d d!�Z&ej e d"�d#d$�Z'ej ed%�d&d'�Z(dS )*� N)�List�Optional)�clouds�config�contract� entitlements�
exceptions� livepatch�messages)�status)�system�util)�AutoAttachCloudInstance)�identity)�CLOUD_BUILD_INFO�DEFAULT_CONFIG_FILE�DEFAULT_LOG_PREFIX)�timer_jobs_state_filez pro.actions)zua-timer.servicezua-timer.timerzua-auto-attach.pathzua-auto-attach.servicezua-reboot-cmds.servicezubuntu-advantage.service)�cfg�token� allow_enable�returnc
C s� ddl m} ztj| ||d� W nv tjk
r\ } z tj| d� || � |�W 5 d}~X Y n< tjk
r� } z tj| d� || � |�W 5 d}~X Y nX t �
� }|r�| � d|� || � dS )a:
Common functionality to take a token and attach via contract backend
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
r )�update_motd_messages)r �r Nz instance-id) Z uaclient.jobs.update_messagingr r Zrequest_updated_contractr ZUrlError� ua_statusr �UserFacingErrorr Zget_instance_idZ write_cache)r r r r �excZ current_iid� r �2/usr/lib/python3/dist-packages/uaclient/actions.py�attach_with_token' s&
�
r T)r �cloudr c C s0 t �| �}|j|d�}|d }t| ||d� dS )aS
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
:raise NonAutoAttachImageError: If this cloud type does not have
auto-attach support.
)�instanceZ
contractToken)r r N)r ZUAContractClientZ"request_auto_attach_contract_tokenr )r r r Zcontract_clientZ
tokenResponser r r r � auto_attachI s
�r" F)�
assume_yes�
allow_beta� access_only)r �namer# r$ r% c C s( t j| |d�}|| ||||d�}|�� S )z�
Constructs an entitlement based on the name provided. Passes kwargs onto
the entitlement constructor.
:raise EntitlementNotFoundError: If no entitlement with the given name is
found, then raises this error.
)r r& )r# r$ Z called_namer% )r Zentitlement_factory�enable)r r& r# r$ r% Zent_cls� entitlementr r r �enable_entitlement_by_name` s �r) )�simulate_with_token�show_all)r r* r+ c C s4 |rt j| ||d�\}}nt j| |d�}d}||fS )z6
Construct the current Pro status dictionary.
)r r r+ )r r+ r )r Zsimulate_statusr )r r* r+ r Zretr r r r y s � r )�filename� return_codesr c
C sf zt j| �� |d�\}}W n: tjk
rT } zt �d�|�t|�� W 5 d}~X Y nX t �||� dS )zCHelper which runs a command and writes output or error to filename.)Zrcsz{}-errorN)r Zsubp�splitr ZProcessExecutionError�
write_file�format�str)�cmdr, r- �out�_�er r r � _write_command_output_to_file� s
(r6 r c C s0 | j pt| j| j| jtjjtfdd� t j
D ��S )Nc s s, | ]$}t |tjj�r|jj|jd �V qdS ))r& N)�
issubclassr ZrepoZRepoEntitlementZrepo_list_file_tmplr0 r& )�.0r( r r r � <genexpr>� s �z#_get_state_files.<locals>.<genexpr>) Zcfg_pathr Zlog_fileZtimer_log_fileZdaemon_log_filer Zua_file�pathr r ZENTITLEMENT_CLASSESr r r r �_get_state_files� s ���r; )r �
output_dirc C sT t dd�|�� t dd�|�� t d�tj�d�|�� t dd�|�� t d �d
�d d � tD ���d
�|�� tD ]$}t d�|�d�||�ddgd� qrt| �}|t�td � D ]�}t j
� |�r�zt �
|�}W n> tk
�r } z t�d|t|�� W Y �q�W 5 d}~X Y nX t�|�}t�� �r0t �||� t �t j
�|t j
�|��|� q�dS )zG
Write all relevant Ubuntu Pro logs to the specified directory
zcloud-idz{}/cloud-id.txtzpro status --format jsonz{}/ua-status.jsonz {} statusz{}/livepatch-status.txtzsystemctl list-timers --allz{}/systemd-timers.txtzxjournalctl --boot=0 -o short-precise {} -u cloud-init-local.service -u cloud-init-config.service -u cloud-config.service� c S s g | ]}d |krd� |��qS )z.servicez-u {})r0 )r8 �sr r r �
<listcomp>� s z collect_logs.<locals>.<listcomp>z{}/journalctl.txtzsystemctl status {}z {}/{}.txtr � )r- �*zFailed to load file: %s
%sN)r6 r0 r Z
LIVEPATCH_CMD�join� UA_SERVICESr; �globr �osr: �isfiler Z load_file� Exception�loggingZwarningr1 r
Zredact_sensitive_logsZwe_are_currently_rootr/ �basename)r r< �serviceZ state_files�fZcontentr5 r r r � collect_logs� s\ � �
��� ��
�
�
�rL )r r c
C s� d }z t �� }W n� tjk
r� } zjt|tj�r@tjtjdd��t|tj �rXt�tj
��t|tj �r~tj tj
j|jd�dd��t�tj��W 5 d }~X Y nX |s�t�tj��|S )Nz auto-attach-cloud-type-error)Zmsg_code)�
cloud_typez(auto-attach-unsupported-cloud-type-error)r Zcloud_instance_factoryr ZCloudFactoryError�
isinstanceZCloudFactoryNoCloudErrorr r
Z UNABLE_TO_DETERMINE_CLOUD_TYPEZ CloudFactoryNonViableCloudErrorZUNSUPPORTED_AUTO_ATTACHZ!CloudFactoryUnsupportedCloudErrorZNonAutoAttachImageErrorZ"UNSUPPORTED_AUTO_ATTACH_CLOUD_TYPEr0 rM )r r! r5 r r r �get_cloud_instance� s4 � ����rO )T)N))rD rH rE �typingr r Zuaclientr r r r r r r
r r r r
Zuaclient.cloudsr r Zuaclient.defaultsr r r Zuaclient.files.state_filesr Z getLoggerZLOGrC ZUAConfigr1 �boolr r" r) �intr6 r; rL rO r r r r �<module> sf $
� % �� �� �� �
� >�