File: //lib/python3/dist-packages/uaclient/__pycache__/actions.cpython-310.pyc
o
nHJe*# � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZmZmZ d dlmZ d dlmZ d d lmZmZmZ d d
l m Z m Z m Z e� e�!e"��Z#d Z$d Z%d
e j&de'de(ddfdd�Z) d0d
e j&dejddfdd�Z*dddddd�d
e j&de'de(d e(d e(d e'd eee' fd d!�Z+ddd"�d
e j&d#ee' d$e(fd%d&�Z d1d'e'd(eee, ddfd)d*�Z-d
e j&fd+d,�Z.d
e j&d-e'fd.d/�Z/dS )2� N)�List�Optional)�clouds�config�contract� entitlements�
exceptions� livepatch)�log)�status)�system�timer�util)�AutoAttachCloudInstance)�identity)�CLOUD_BUILD_INFO�DEFAULT_CONFIG_FILE�DEFAULT_LOG_PREFIX)�AttachmentData�attachment_data_file�timer_jobs_state_file)zua-timer.servicezua-timer.timerzua-auto-attach.pathzua-auto-attach.servicezua-reboot-cmds.servicezubuntu-advantage.service�
�cfg�token� allow_enable�returnc
C sN ddl m} t�| �}tjjtjjd�}z |j||d�}W n t j
y5 } z t � t
|�� t �� �d}~ww | j�|� tj�� |�di ��dt�| ��}| �d|� z t�| i | jj|� W n( t j
t jfy� } zt�t|d �� tj | d
� || � |� � | �d} ~ ww t � � }
|
r�| �d |
� t�t|d �� || � t �!� 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)�tz)�contract_token�
attachment_dtN�machineTokenInfo� machineIdz
machine-id)� attached_at�r z instance-id)"� uaclient.timer.update_messagingr r �UAContractClient�datetime�now�timezone�utc�add_contract_machiner �UrlError�LOG� exception�str�ConnectivityError�machine_token_file�writer �get_machine_id� cache_clear�get� write_cache�process_entitlements_deltar �UbuntuProErrorr r � ua_statusr �update_activity_tokenr �get_instance_idr
�start) r r r r �contract_clientr" �new_machine_token�e�
machine_id�exc� current_iid� rB �2/usr/lib/python3/dist-packages/uaclient/actions.py�attach_with_token. sN
���
� � �� rD 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 r% �%get_contract_token_for_cloud_instancerD )r rE r r< �
tokenResponser rB rB rC � auto_attachh s
�rJ F� )�
assume_yes�
allow_beta� access_only�variant�
extra_args�namerL rM rN rO rP 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 rQ rO )rL rM � called_namerN rP )r �entitlement_factory�enable) r rQ rL rM rN rO rP �ent_cls� entitlementrB rB rC �enable_entitlement_by_name s ��rW )�simulate_with_token�show_allrX rY c C s: |rt j| ||d�\}}||fS t j| |d�}d}||fS )z6
Construct the current Pro status dictionary.
)r r rY )r rY r )r8 �simulate_statusr )r rX rY r �retrB rB rC 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�formatr. )�cmdr\ r] �out�_r> rB rB rC � _write_command_output_to_file� s $��rg c C s( | j pt| jtjjtgdd� tjD ��S )Nc s s. � | ]}t |tjj�r|jj|jd �V qdS ))rQ N)�
issubclassr �repo�RepoEntitlement�repo_list_file_tmplrc rQ )�.0rV rB rB rC � <genexpr>� s � �
�z#_get_state_files.<locals>.<genexpr>) �cfg_pathr �log_filer �ua_file�pathr r �ENTITLEMENT_CLASSESr# rB rB rC �_get_state_files� s ���rs �
output_dirc
C s� t dd�|�� t dd�|�� t d�tj�d�|�� t dd�|�� t d d
�|�� t d �d �d
d� tD ���d�|�� tD ]}t d�|�d�||�ddgd� qAt| �}t�� rdt �
� dt � nt � � g}t
|�D ]7\}}zt�t�|��}t�tj�|d�|��|� W qm ty� } zt�d|t|�� W Y d}~qmd}~ww |t�td � D ]I} tj�| �r�zt�| �}W n ty� } zt�d| t|�� W Y d}~q�d}~ww t�|�}t�� r�t�| |� t�tj�|tj�| ��|� 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.txtzujournalctl --boot=0 -o short-precise -u cloud-init-local.service -u cloud-init-config.service -u cloud-config.servicez{}/cloud-init-journal.txtz journalctl -o short-precise {}� c S s g | ] }d |v rd� |��qS )z.servicez-u {})rc )rl �srB rB rC �
<listcomp>� s z collect_logs.<locals>.<listcomp>z{}/pro-journal.txtzsystemctl status {}z {}/{}.txtr � )r] Nz
user{}.logz&Failed to collect user log file: %s
%s�*zFailed to load file: %s
%s) rg rc r �
LIVEPATCH_CMD�join� UA_SERVICESrs r �we_are_currently_root�pro_log�get_all_user_log_files�USER_LOG_COLLECTED_LIMIT�get_user_log_file� enumerate�redact_sensitive_logsr � load_filerb �osrq � Exceptionr, �warningr. �globr �isfile�basename)
r rt �service� state_files�user_log_files� log_file_idxro �contentr> �frB rB rC � collect_logs� s�
�
�
��� ���
����
��� ��
���r� )T)N)0r& r� �loggingr� �typingr r �uaclientr r r r r r r
r~ r r8 r r
r �uaclient.cloudsr r �uaclient.defaultsr r r �uaclient.files.state_filesr r r � getLogger� replace_top_level_logger_name�__name__r, r| r� �UAConfigr. �boolrD rJ rW �intrg rs r� rB rB rB rC �<module> s� ���
�=���
��������
�!���
���
�
�