OwlCyberSecurity - MANAGER
Edit File: machine_token.cpython-38.pyc
U ��g�# � @ s~ d dl Z d dlmZ d dlmZmZmZ d dlmZmZm Z m Z d dl m Z d dl mZ daG dd� d�Zd ed �d d �ZdS ) � N)�datetime)�Any�Dict�Optional)�defaults� exceptions�system�util)�PublicMachineTokenData)�UAFilec @ sh e Zd Zejdfeee d�dd�Zed�dd�Z dd � Z ee d �d d �Z e d d� �Z e dd� �Ze ee d �dd��Zdd� Ze eeef d �dd��Zd1ee d�dd�Zed2eeef ee d�dd ��Ze ee d �d d ��Ze d d � �Ze ee d �d!d"��Ze d#d �d$d%��Z e d#d �d&d'��Z e d(d �d)d*��Z e d+d,� �Z e d-d.� �Z e d/d0� �Z!dS )3�MachineTokenFileN�� directory�machine_token_overlay_pathc C sF t j}t||d t j �| _t||d�| _|| _d | _d | _d | _ d S )N�/F) r ZMACHINE_TOKEN_FILEr ZPRIVATE_SUBDIR� private_file� public_filer �_machine_token� _entitlements�_contract_expiry_datetime)�selfr r � file_name� r �>/usr/lib/python3/dist-packages/uaclient/files/machine_token.py�__init__ 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�dumpsZDatetimeAwareJSONEncoderr �writer Z from_dictZto_dictr r r r r �NonRootUserError)r r Zprivate_content_strZpublic_contentZpublic_content_strr r r r! s( � �� � zMachineTokenFile.writec C s< t �� r0| j�� | j�� d| _d| _d| _nt� � �dS )z!Delete both pub and private filesN) r r r �deleter r r r r r" �r r r r r# 4 s zMachineTokenFile.delete��returnc C sT t �� r| j}n| j}|�� }|s&d S ztj|t jd�}W n tk rN Y nX |S �Nr ) r r r r �readr �loads�DatetimeAwareJSONDecoder� Exception)r Z file_handler�contentr r r r( @ s zMachineTokenFile.readc C s$ t �� r| jjo| jjS | jjS d S )N)r r r � is_presentr r$ r r r r- N 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( r �parse_machine_token_overlayr Z depth_first_merge_overlay_dict)r r, �machine_token_overlayr r r � machine_tokenU s ��z MachineTokenFile.machine_tokenc C s& | j r"| j �di ��di ��d�S d S )N�machineTokenInfo� contractInfo�name�r0 �getr$ r r r � contract_nameg s ���z MachineTokenFile.contract_namec C s t �|�}tj|tjd�S r' )r Z load_filer r) r r* )r r Z machine_token_overlay_contentr r r r. r s ��z,MachineTokenFile.parse_machine_token_overlayc C s t | j�r| jd d S i S )Nr1 Z accountInfo��boolr0 r$ r r r �account{ s zMachineTokenFile.account��seriesc C s, | j r | j S | jsi S | �| j|�| _ | j S )z9Return configured entitlements keyed by entitlement named)r r0 �get_entitlements_from_token)r r; r r r � entitlements� s �z MachineTokenFile.entitlements)r0 r; c C s� ddl m} | si S i }| �di ��d�}|s2i S tdd� | �dg �D ��}tdd� |�d g �D ��}|�� D ]8\}}d |i} ||kr�|| | d < || |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_overridesr1 r2 c s s" | ]}|� d �|� d�fV qdS )�type�tokenN�r5 ��.0�er r r � <genexpr>� s �z?MachineTokenFile.get_entitlements_from_token.<locals>.<genexpr>�resourceTokensc s s | ]}|� d �|fV qdS )r? NrA rB r r r rE � s �ZresourceEntitlements� entitlementZ resourceTokenr: )Zuaclient.contractr> r5 �dict�items) r0 r; r> r= r2 Ztokens_by_nameZ ent_by_nameZentitlement_nameZ ent_valueZentitlement_cfgr r r r< � s0 � � �� z,MachineTokenFile.get_entitlements_from_tokenc C s2 | j s,| jr,| j�di ��di ��dd�| _ | j S )z6Return a datetime of the attached contract expiration.r1 r2 Z effectiveToN)r � is_attachedr0 r5 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.r7 r$ r r r rJ � 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)rK �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 activityTokenr4 r$ r r r �activity_token� s �z MachineTokenFile.activity_tokenc C s | j r| j �di ��d�S d S )NrN Z activityIDr4 r$ r r r � activity_id� s z MachineTokenFile.activity_idz Optional[int]c C s | j r| j �di ��d�S d S )NrN ZactivityPingIntervalr4 r$ r r r �activity_ping_interval� s �z'MachineTokenFile.activity_ping_intervalc C s& | j r"| j �di ��di ��d�S d S )Nr1 r2 �idr4 r$ r r r � contract_id� s ���z MachineTokenFile.contract_idc C s | j r| j �dg �S d S )NrF r4 r$ r r r �resource_tokens� s z MachineTokenFile.resource_tokensc C s, | � � �di ��di ��di ��dd �}|S )NZsupportrG Z affordancesZ onlySeries)r= r5 )r � only_seriesr r r rU � s � � � ��z MachineTokenFile.only_series)N)N)"�__name__� __module__� __qualname__r ZDEFAULT_DATA_DIR�strr r rH r! r# r( �propertyr- r0 r6 r. r r r9 r= � staticmethodr< r rK rJ �intrM rO rP rQ rS rT rU r r r r r sV �� � �% r r% c C s4 ddl m} ts0| s|� } t| j| j�d�d�atS )Nr )�UAConfigr/ r )Zuaclient.configr] �_machine_token_filer Zdata_dirZfeaturesr5 )Zcfgr] r r r �get_machine_token_file s ��r_ )N)r r �typingr r r Zuaclientr r r r Z uaclient.contract_data_typesr Zuaclient.files.filesr r^ r r_ r r r r �<module> s x