File: //lib/python3/dist-packages/uaclient/files/__pycache__/files.cpython-310.pyc
o
S��cy/ � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZmZ d dl m
Z
m Z m Z m
Z
mZmZ d dlmZ e �� Ze�e�ZG dd� d�ZG dd � d �ZG d
d � d �ZdS ) � N)�datetime)�Any�Dict�Optional)�defaults� event_logger�
exceptions�messages�system�util)�PublicMachineTokenDatac @ s� e Zd Zejdfdededefdd�Zedefdd ��Z edefd
d ��Z
ed d
� �Z defdd�Z de
e fdd�Zdd� ZdS )�UAFileT�name� directory�privatec C s* || _ || _|| _tj�| j | j�| _d S �N)�
_directory�
_file_name� _is_private�os�path�join�_path)�selfr r r � r �6/usr/lib/python3/dist-packages/uaclient/files/files.py�__init__ s zUAFile.__init__�returnc C � | j S r )r �r r r r r � z UAFile.pathc C r r )r r r r r �
is_private r zUAFile.is_privatec C s t j�| j�S r )r r �existsr r r r �
is_present# s zUAFile.is_present�contentc C sb | j rtjntj}tj�| j�s't�| j� tj� | j�tj
kr't� | jd� t �
| j||� d S )Ni� )r! r �ROOT_READABLE_MODE�WORLD_READABLE_MODEr r r" r �makedirs�basename�PRIVATE_SUBDIR�chmodr
�
write_file)r r$ � file_moder r r �write' s �� z UAFile.writec C s> d }z t �| j�}W |S ty t�d�| j�� Y |S w )NzFile does not exist: {})r
� load_filer �FileNotFoundError�LOG�debug�format�r r$ r r r �read3 s ��z UAFile.readc C s t �| j� d S r )r
�ensure_file_absentr r r r r �delete; s z
UAFile.deleteN)�__name__�
__module__� __qualname__r �DEFAULT_DATA_DIR�str�boolr �propertyr r! r# r- r r4 r6 r r r r r
s& ���
�
r
c @ s e Zd Zejddfdededee fdd�Zde fd d
�Z
d d � Z d
ee fdd�Z e
dd� �Ze
dd� �Zdd� Ze
d
ee fdd��Ze
dd� �Zedefdd ��Ze
d
efd d ��Ze
d d � �Ze
d
efd!d"��Ze
d-d$d%��Ze
d-d&d'��Z e
d.d)d*��Z e
d+d,� �Z dS )/�MachineTokenFileTNr � root_mode�machine_token_overlay_pathc C sH t j}|| _t||t j �| _t||d�| _|| _d | _d | _ d | _
d S )NF) r �MACHINE_TOKEN_FILE�is_rootr
r) � private_file� public_filer@ �_machine_token�
_entitlements�_contract_expiry_datetime)r r r? r@ � file_namer r r r @ s
�
zMachineTokenFile.__init__r$ c C sv | j r7tj|tjd�}| j�|� t�|�}t� |��
d�}tj|tjd�}| j �|� d| _ d| _
d| _dS t�� �)z8Update the machine_token file for both pub/private files��clsFN)rB �json�dumpsr �DatetimeAwareJSONEncoderrC r- �loadsr � from_dict�to_dictrD rE rF rG r �NonRootUserError)r r$ � content_strr r r r- Q s �
�
zMachineTokenFile.writec C s8 | j r| j�� | j�� d| _d| _d| _dS t�� �)z!Delete both pub and private filesN) rB rD r6 rC rE rF rG r rQ r r r r r6 e s
zMachineTokenFile.deleter c C sP | j r| j}n| j}|�� }|sd S z tj|tjd�}W |S ty' Y |S w �NrI ) rB rC rD r4 rK rN r �DatetimeAwareJSONDecoder� Exception)r � file_handlerr$ r r r r4 q s ��zMachineTokenFile.readc C s | j r | jjo
| jjS | jjS r )rB rD r# rC r r r r r# s zMachineTokenFile.is_presentc C sB | j s | �� }|r| jr| �| j�}|rtj||d� || _ | j S )zAReturn the machine-token if cached in the machine token response.)� base_dict� overlay_dict)rE r4 r@ �parse_machine_token_overlayr � depth_first_merge_overlay_dict)r r$ �machine_token_overlayr r r �
machine_token� s
��z MachineTokenFile.machine_tokenc
C sp t j�|�st�tjj|d���zt� |�}t
j |t j
d�W S ty7 } z
t�tjjt|�|d���d }~ww )N)� file_pathrI )�errorr] )r r r" r �UserFacingErrorr �&INVALID_PATH_FOR_MACHINE_TOKEN_OVERLAYr2 r
r. rK rN r rT �
ValueError�ERROR_JSON_DECODING_IN_FILEr; )r r@ � machine_token_overlay_content�er r r rY � s, ��������z,MachineTokenFile.parse_machine_token_overlayc C s t | j�r | jd d S d S )N�machineTokenInfo� accountInfo�r< r\ r r r r �account� s
zMachineTokenFile.accountc C s* | j r| j S | js i S | �| j�| _ | j S )z9Return configured entitlements keyed by entitlement named)rF r\ �get_entitlements_from_tokenr r r r � entitlements� s �z MachineTokenFile.entitlementsr\ c C s� ddl m} | s
i S i }| �di ��d�}|si S tdd� | �dg �D ��}tdd� |�d g �D ��}|�� D ]\}}d
|i}||v rI|| |d < ||� |||< q7|S ) z�Return a dictionary of entitlements keyed by entitlement name.
Return an empty dict if no entitlements are present.
r )�apply_contract_overridesre � contractInfoc s s$ � | ]
}|� d �|� d�fV qdS )�type�tokenN��get��.0rd r r r � <genexpr>� s
� �
�z?MachineTokenFile.get_entitlements_from_token.<locals>.<genexpr>�resourceTokensc s s � | ]
}|� d �|fV qdS )rm Nro rq r r r rs � s
� �
��resourceEntitlements� entitlement�
resourceToken)�uaclient.contractrk rp �dict�items) r\ rk rj rl �tokens_by_name� ent_by_name�entitlement_name� ent_value�entitlement_cfgr r r ri � s0 �
�
��
z,MachineTokenFile.get_entitlements_from_tokenc C s | j s
| jd d d | _ | j S )z6Return a datetime of the attached contract expiration.re rl � effectiveTo)rG r\ r r r r �contract_expiry_datetime� s ���z)MachineTokenFile.contract_expiry_datetimec C s
t | j�S )z<Report whether this machine configuration is attached to UA.rg r r r r � is_attached� s
z MachineTokenFile.is_attachedc C s | j �� t�� �� }|jS )a Report num days until contract expiration based on effectiveTo
:return: A positive int representing the number of days the attached
contract remains in effect. Return a negative int for the number
of days beyond contract's effectiveTo date.
)r� �dater �utcnow�days)r �deltar r r �contract_remaining_days� s z(MachineTokenFile.contract_remaining_days�
Optional[str]c C � | j r
| j �di ��d�S d S )N� activityInfo�
activityToken�r\ rp r r r r �activity_token �
�z MachineTokenFile.activity_tokenc C r� )Nr� �
activityIDr� r r r r � activity_id s z MachineTokenFile.activity_id�
Optional[int]c C r� )Nr� �activityPingIntervalr� r r r r �activity_ping_interval r� z'MachineTokenFile.activity_ping_intervalc C s& | j r| j �di ��di ��d�S d S )Nre rl �idr� r r r r � contract_id s �z MachineTokenFile.contract_id)r r� )r r� ) r7 r8 r9 r r: r; r< r r ry r- r6 r4 r= r# r\ rY rh rj � staticmethodr ri r r� r� �intr� r� r� r� r� r r r r r> ? sN ���
�
#
r> c @ s� e Zd Zejdfdedefdd�Zdedefdd �Zdedefd
d �Z d ed
efdd�Z
d ed
efdd�Z dd� Z de
fdd�ZdS )�
NoticeFileTr r? c C s d}t ||d�| _|| _d S )Nz notices.jsonF)r
�filerB )r r r? rH r r r r " s
zNoticeFile.__init__�label� descriptionc C sD | j r | �� pg }||g}||vr |�|� | �|� dS dS tj�)zt
Adds a notice to the notices.json file.
Raises a NonRootUserError if the user is not root.
N)rB r4 �appendr- r rQ )r r� r� �notices�noticer r r �add+ s
�zNoticeFile.addc C �4 z | � ||� W dS tjy t�d� Y dS w )zh
Adds a notice to the notices.json file.
Logs a warning when adding as non-root
z%Trying to add notice as non-root userN)r� r rQ �event�warning)r r� r� r r r �try_add9 �
�zNoticeFile.try_add� label_regex� descr_regexc C s� | j rBg }| �� p
g }|r(|D ]\}}t�||�r t�||�r q|�||f� q|r1| �|� dS tj�| j j�r@| j �
� dS dS t j �)zw
Removes a notice to the notices.json file.
Raises a NonRootUserError if the user is not root.
N)
rB r4 �re�matchr� r- r r r" r� r6 r rQ )r r� r� r� �cached_notices� notice_label� notice_descrr r r �removeC s �zNoticeFile.removec C r� )zn
Removes a notice to the notices.json file.
Logs a warning when removing as non-root
z(Trying to remove notice as non-root userN)r� r rQ r� r� )r r� r� r r r �
try_removeX r� zNoticeFile.try_removec C s< | j �� }|s d S z tj|tjd�W S ty | Y S w rS )r� r4 rK rN r rT ra r3 r r r r4 b s
�zNoticeFile.readr$ c C s* t |t�s
tj|tjd�}| j�|� d S rS )�
isinstancer; rK rL r rM r� r- r3 r r r r- k s
zNoticeFile.writeN)r7 r8 r9 r r: r; r< r r� r� r� r� r4 r r- r r r r r� ! s ��
�
r� )rK �loggingr r� r �typingr r r �uaclientr r r r r
r � uaclient.contract_data_typesr �get_event_loggerr� � getLoggerr7 r0 r
r> r� r r r r �<module> s
0 c