File: //usr/lib/python3/dist-packages/uaclient/__pycache__/actions.cpython-310.pyc
o
S��c6! � @ s� 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 d d
lm Z e� d �Z e � � Z d Z!d
ej"de#de$ddfdd�Z% d/d
ej"dejddfdd�Z&dddd�d
ej"de#de$de$d e$f
d d �Z'ddd �d
ej"d ee# d!e$fd"d#�Z d0d$e#d%eee( ddfd&d'�Z)d
ej"fd(d)�Z*d
ej"d*e#fd+d,�Z+d
ej"defd-d.�Z,dS )1� N)�List�Optional)�clouds�config�contract� entitlements� event_logger�
exceptions�messages)�status)�system�util)�AutoAttachCloudInstance)�identity)�CLOUD_BUILD_INFO�DEFAULT_CONFIG_FILE�DEFAULT_LOG_PREFIX)�
LIVEPATCH_CMD)�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} z
tj| ||d� W n1 tjy) } z tj| d� || � |�d}~w tjyA } z tj| d� || � |�d}~ww t �
� }|rN| � 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_apt_and_motd_messages)r �r Nz instance-id) � uaclient.jobs.update_messagingr r �request_updated_contractr �UrlError� ua_statusr �UserFacingErrorr �get_instance_id� write_cache)r r r r �exc� current_iid� r$ �2/usr/lib/python3/dist-packages/uaclient/actions.py�attach_with_token) s(
� � �� r& T�cloudc 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.
)�instance�
contractToken)r r N)r �UAContractClient�"request_auto_attach_contract_tokenr& )r r' r �contract_client�
tokenResponser r$ r$ r% � auto_attachK s
�r. F)�
assume_yes�
allow_beta� access_only�namer/ r0 r1 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 r2 )r/ r0 � called_namer1 )r �entitlement_factory�enable)r r2 r/ r0 r1 �ent_cls� entitlementr$ r$ r% �enable_entitlement_by_nameb s �r8 )�simulate_with_token�show_allr9 r: c C s: |rt j| ||d�\}}||fS t j| |d�}d}||fS )z6
Construct the current Pro status dictionary.
)r r r: )r r: r )r �simulate_statusr )r r9 r: r �retr$ r$ r% r { s
� �r �filename� return_codesc
C sj z
t j| �� |d�\}}W n tjy, } zt �d�|�t|�� W Y d}~dS d}~ww t �||� dS )zCHelper which runs a command and writes output or error to filename.)�rcsz{}-errorN)r �subp�splitr �ProcessExecutionError�
write_file�format�str)�cmdr= r> �out�_�er$ r$ r% � _write_command_output_to_file� s $��rJ c C s0 | j pt| j| j| jtjjtgdd� t j
D ��S )Nc s s. � | ]}t |tjj�r|jj|jd �V qdS ))r2 N)�
issubclassr �repo�RepoEntitlement�repo_list_file_tmplrD r2 )�.0r7 r$ r$ r% � <genexpr>� s � �
�z#_get_state_files.<locals>.<genexpr>) �cfg_pathr �log_file�timer_log_file�daemon_log_filer �ua_file�pathr r �ENTITLEMENT_CLASSESr r$ r$ r% �_get_state_files� s ���rX �
output_dirc C sL t dd�|�� t dd�|�� t d�t�d�|�� t dd�|�� t d �d
�d d � tD ���d
�|�� tD ]}t d�|�d�||�ddgd� q8t| �}|t�td � D ]K}tj �
|�r�zt � |�}W n t
y� } zt�d|t|�� W Y d}~qXd}~ww t�|�}t�� dkr�t �||� t �tj �|tj �|��|� qXdS )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 |v rd� |��qS )z.servicez-u {})rD )rO �sr$ r$ r% �
<listcomp>� s z collect_logs.<locals>.<listcomp>z{}/journalctl.txtzsystemctl status {}z {}/{}.txtr � )r> �*zFailed to load file: %s
%sN)rJ rD r �join� UA_SERVICESrX �globr �osrV �isfiler � load_file� Exception�logging�warningrE r
�redact_sensitive_logs�getuidrC �basename)r rY �service� state_files�f�contentrI r$ r$ r% � collect_logs� s^
�
��� ���
� ��
���ro c
C s� d }zt �� }W n@ tjyH } z3t|tj�r tjtjdd��t|tj �r+t�tj
��t|tj �r>tj tj
j|jd�dd��t�tj��d }~ww |sQt�tj��|S )Nz auto-attach-cloud-type-error)�msg_code)�
cloud_typez(auto-attach-unsupported-cloud-type-error)r �cloud_instance_factoryr �CloudFactoryError�
isinstance�CloudFactoryNoCloudErrorr r
� UNABLE_TO_DETERMINE_CLOUD_TYPE� CloudFactoryNonViableCloudError�UNSUPPORTED_AUTO_ATTACH�!CloudFactoryUnsupportedCloudError�NonAutoAttachImageError�"UNSUPPORTED_AUTO_ATTACH_CLOUD_TYPErD rq )r r( rI r$ r$ r% �get_cloud_instance� s8 � ������r| )T)N)-ra rf rb �typingr r �uaclientr r r r r r r
r r r r
�uaclient.cloudsr r �uaclient.defaultsr r r � uaclient.entitlements.livepatchr �uaclient.files.state_filesr � getLogger�LOG�get_event_logger�eventr` �UAConfigrE �boolr& r. r8 �intrJ rX ro r| r$ r$ r$ r% �<module> s� $
���
�%���
������
� ���
���
�
� =��