Your IP : 216.73.217.13
U
8�-d5& � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dlm Z m
Z
m Z m Z m
Z
mZ d dlmZ e
�� Ze�e�ZG dd� d�ZG dd � d �ZdS )
� N)�datetime)�Any�Dict�Optional)�defaults� event_logger�
exceptions�messages�system�util)�PublicMachineTokenDatac @ s~ e Zd Zejdfeeed�dd�Zeed�dd��Z eed�dd ��Z
ed
d � �Z ed �d
d�Z e
e d�dd�Zdd� ZdS )�UAFileT)�name� directory�privatec C s* || _ || _|| _tj�| j | j�| _d S �N)�
_directoryZ
_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 s | j S r )r �r r r r r s z UAFile.pathc C s | j S r )r r r r r �
is_private s 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 r tjntj}tj�| j�sNt�| j� tj� | j�tj
krNt� | jd� t �
| j||� d S )Ni� )r r ZROOT_READABLE_MODEZWORLD_READABLE_MODEr r r r �makedirs�basename�PRIVATE_SUBDIR�chmodr
Z
write_file)r r" Z file_moder r r �write& s �� z UAFile.writec C s@ d }zt �| j�}W n& tk
r: t�d�| j�� Y nX |S )NzFile does not exist: {})r
� load_filer �FileNotFoundError�LOG�debug�format)r r" r r r �read2 s z UAFile.readc C s t �| j� d S r )r
Zensure_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 r- r. r r r r r
s ��
r
c @ s e Zd Zejdfeee d�dd�Zed�dd�Z dd � Z
ee d
�d d �Z e d
d� �Z
e dd� �Zdd� Ze ee d
�dd��Ze dd� �Zeed�dd��Ze ee d
�dd��Ze d d � �Ze ee d
�d d ��Ze d d
�d!d"��Ze d d
�d#d$��Ze d%d
�d&d'��Z e d(d)� �Z dS )*�MachineTokenFileN)r �machine_token_overlay_pathc C sB t j}t||t j �| _t||d�| _|| _d | _d | _d | _ d S )NF)
r ZMACHINE_TOKEN_FILEr
r% � private_file� public_filer7 �_machine_token�
_entitlements�_contract_expiry_datetime)r r r7 � file_namer r r r ? s �zMachineTokenFile.__init__)�private_contentc C sr t �� rftj|t jd�}| j�|� t�|�j dd�}tj|t jd�}| j
�|� d| _ d| _ d| _
nt�� �dS )z8Update the machine_token file for both pub/private files��clsF)Z keep_noneN)r �we_are_currently_root�json�dumpsZDatetimeAwareJSONEncoderr8 r' r Z from_dictZto_dictr9 r: r; r< r �NonRootUserError)r r> Zprivate_content_strZpublic_contentZpublic_content_strr r r r' N s( � �� � zMachineTokenFile.writec C s< t �� r0| j�� | j�� d| _d| _d| _nt� � �dS )z!Delete both pub and private filesN)
r rA r9 r. r8 r: r; r< r rD r r r r r. f s
zMachineTokenFile.deleter c C sT t �� r| j}n| j}|�� }|s&d S ztj|t jd�}W n tk
rN Y nX |S )Nr? ) r rA r8 r9 r- rB �loads�DatetimeAwareJSONDecoder� Exception)r Z file_handlerr" r r r r- r s zMachineTokenFile.readc C s$ t �� r| jjo| jjS | jjS d S r )r rA r9 r! r8 r r r r r! � s zMachineTokenFile.is_presentc C sB | j s<| �� }|r6| jr6| �| j�}|r6tj||d� || _ | j S )zAReturn the machine-token if cached in the machine token response.)Z base_dictZ overlay_dict)r: r- r7 �parse_machine_token_overlayr Z depth_first_merge_overlay_dict)r r" Zmachine_token_overlayr r r �
machine_token� s
��z MachineTokenFile.machine_tokenc
C s~ t j�|�s t�tjj|d���z t� |�}t
j |t j
d�W S tk
rx } z t�tjjt|�|d���W 5 d }~X Y nX d S )N)� file_pathr? )�errorrJ )r r r r ZUserFacingErrorr Z&INVALID_PATH_FOR_MACHINE_TOKEN_OVERLAYr, r
r( rB rE r rF �
ValueErrorZERROR_JSON_DECODING_IN_FILEr3 )r r7 Z machine_token_overlay_content�er r r rH � s* ���� ��z,MachineTokenFile.parse_machine_token_overlayc C s t | j�r| jd d S d S )N�machineTokenInfoZ accountInfo�r4 rI r r r r �account� s
zMachineTokenFile.accountc C s* | j r | j S | jsi S | �| j�| _ | j S )z9Return configured entitlements keyed by entitlement named)r; rI �get_entitlements_from_tokenr r r r � entitlements� s �z MachineTokenFile.entitlements)rI c C s� ddl m} | si S i }| �di ��d�}|s2i S tdd� | �dg �D ��}tdd� |�d g �D ��}|�� D ]4\}}d
|i}||kr�|| |d < ||� |||< qn|S ) z�Return a dictionary of entitlements keyed by entitlement name.
Return an empty dict if no entitlements are present.
r )�apply_contract_overridesrN � contractInfoc s s" | ]}|� d �|� d�fV qdS )�type�tokenN��get��.0rM r r r � <genexpr>� s �z?MachineTokenFile.get_entitlements_from_token.<locals>.<genexpr>ZresourceTokensc s s | ]}|� d �|fV qdS )rU NrW rY r r r r[ � s �ZresourceEntitlementsZ entitlementZ
resourceToken)Zuaclient.contractrS rX �dict�items) rI rS rR rT Ztokens_by_nameZ ent_by_nameZentitlement_nameZ ent_valueZentitlement_cfgr r r rQ � 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.rN rT Z effectiveToN)r< rI rX 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.rO r r r r � is_attached� s z MachineTokenFile.is_attachedc C s* | j dkrdS | 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 ZutcnowZdays)r Zdeltar r r �contract_remaining_days� s
z(MachineTokenFile.contract_remaining_daysz
Optional[str]c C s | j r| j �di ��d�S d S )N� activityInfoZ
activityToken�rI rX r r r r �activity_token s
�z MachineTokenFile.activity_tokenc C s | j r| j �di ��d�S d S )Nrb Z
activityIDrc r r r r � activity_id
s z MachineTokenFile.activity_idz
Optional[int]c C s | j r| j �di ��d�S d S )Nrb ZactivityPingIntervalrc r r r r �activity_ping_interval s
�z'MachineTokenFile.activity_ping_intervalc C s& | j r"| j �di ��di ��d�S d S )NrN rT �idrc r r r r � contract_id s ���z MachineTokenFile.contract_id) r/ r0 r1 r r2 r3 r r r\ r' r. r- r5 r! rI rH rP rR � staticmethodr rQ r r^ r_ �intra rd re rf rh r r r r r6 > sD ��
#
r6 )rB Zloggingr r �typingr r r Zuaclientr r r r r
r Z uaclient.contract_data_typesr Zget_event_loggerZeventZ getLoggerr/ r* r
r6 r r r r �<module> s
0