File: //usr/lib/python3/dist-packages/uaclient/files/__pycache__/files.cpython-310.pyc
o
nHJe�% � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dlm Z m
Z
m Z m Z m
Z
d dlmZ e
�� Ze�e
�e��ZG dd� d�ZG dd � d e�ZG d
d � d �ZdS ) � N)�datetime)�Any�Dict�Optional)�defaults� event_logger�
exceptions�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 sp | j rtjntj}ztj�| j�tjkr tj | jdd� nt� | j� W n t
y- Y nw t � | j||� d S )Ni� )�mode)
r r �ROOT_READABLE_MODE�WORLD_READABLE_MODEr r �basenamer �PRIVATE_SUBDIR�makedirs�OSErrorr �
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 )Nz(Tried to load %s but file does not exist)r � load_filer �FileNotFoundError�LOG�debug)r r# r r r �read7 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 r2 r4 r r r r r s& ���
�
r c s" e Zd Zdef� fdd� Z� ZS )�
UserCacheFiler
c s t � j|t�� dd� d S )NF)r r )�superr r �get_user_cache_dir)r r
�� __class__r r r D s
�zUserCacheFile.__init__)r5 r6 r7 r9 r �
__classcell__r r r? r r< C s r< c @ s e Zd Zejdfdedee fdd�Zdefdd�Z d d
� Z
d ee fd d
�Z e dd� �Z
e dd� �Zdd� Ze d ee fdd��Ze dd� �Zedefdd��Ze d ee fdd ��Ze d d � �Ze d ee fd d ��Ze d-d"d#��Ze d-d$d%��Ze d.d'd(��Z e d)d*� �Z e d+d,� �Z dS )/�MachineTokenFileNr �machine_token_overlay_pathc C sF t j}t||d t j �| _t||d�| _|| _d | _d | _d | _ d S )N�/F)
r �MACHINE_TOKEN_FILEr r( � private_file� public_filerC �_machine_token�
_entitlements�_contract_expiry_datetime)r r rC � file_namer r r r K s �
zMachineTokenFile.__init__�private_contentc C sp t �� r4tj|t jd�}| j�|� t�|�j dd�}tj|t jd�}| j
�|� d| _ d| _ d| _
dS t�� �)z8Update the machine_token file for both pub/private files��clsF)� keep_noneN)r
�we_are_currently_root�json�dumps�DatetimeAwareJSONEncoderrF r- r � from_dict�to_dictrG rH rI rJ r �NonRootUserError)r rL �private_content_str�public_content�public_content_strr r r r- Z s$ � ���
zMachineTokenFile.writec C s: t �� r| j�� | j�� d| _d| _d| _dS t� � �)z!Delete both pub and private filesN)
r
rP rG r4 rF rH rI rJ r rV r r r r r4 r s
zMachineTokenFile.deleter c C sR t �� r| j}n| j}|�� }|sd S z tj|t jd�}W |S ty( Y |S w �NrM ) r
rP rF rG r2 rQ �loads�DatetimeAwareJSONDecoder� Exception)r � file_handlerr# r r r r2 ~ s ��zMachineTokenFile.readc C s t �� r | jjo | jjS | jjS r )r
rP rG r" rF 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)rH r2 rC �parse_machine_token_overlayr
� depth_first_merge_overlay_dict)r r# �machine_token_overlayr r r �
machine_token� s
��z MachineTokenFile.machine_tokenc C s t �|�}tj|tjd�S rZ )r r. rQ r[ r
r\ )r rC � machine_token_overlay_contentr r r ra � s ��z,MachineTokenFile.parse_machine_token_overlayc C s t | j�r | jd d S d S )N�machineTokenInfo� accountInfo�r: rd 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)rI rd �get_entitlements_from_tokenr r r r � entitlements� s �z MachineTokenFile.entitlementsrd 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_overridesrf � contractInfoc s s$ � | ]
}|� d �|� d�fV qdS )�type�tokenN��get��.0�er r r � <genexpr>� s
� �
�z?MachineTokenFile.get_entitlements_from_token.<locals>.<genexpr>�resourceTokensc s s � | ]
}|� d �|fV qdS )rn Nrp rr r r r ru � s
� �
��resourceEntitlements� entitlement�
resourceToken)�uaclient.contractrl rq �dict�items) rd rl rk rm �tokens_by_name� ent_by_name�entitlement_name� ent_value�entitlement_cfgr r r rj � s0 �
�
��
z,MachineTokenFile.get_entitlements_from_tokenc C s, | j s| j�di ��di ��dd�| _ | j S )z6Return a datetime of the attached contract expiration.rf rm � effectiveToN)rJ rd rq 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.rh r r r r � is_attached� s
z MachineTokenFile.is_attachedc C s* | j du rdS | 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.
N)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�rd rq 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 )Nrf rm �idr� r r r r � contract_id s �z MachineTokenFile.contract_idc C s | j r
| j �dg �S d S )Nrv r� r r r r �resource_tokens" s z MachineTokenFile.resource_tokens)r r� )r r� ) r5 r6 r7 r r8 r9 r r r{ r- r4 r2 r; r" rd ra ri rk � staticmethodr rj r r� r� �intr� r� r� r� r� r� r r r r rB J sL ��
�
#
rB )rQ �loggingr r �typingr r r �uaclientr r r r r
� uaclient.contract_data_typesr �get_event_logger�event� getLogger� replace_top_level_logger_namer5 r0 r r< rB r r r r �<module> s 5