File manager - Edit - /usr/lib/python3/dist-packages/uaclient/files/__pycache__/files.cpython-310.pyc
Back
o KHvd�& � @ 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 e�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 �read2 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 r3 r5 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 ? s �zUserCacheFile.__init__)r6 r7 r8 r: r � __classcell__r r r@ r r= > 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 dS )-�MachineTokenFileNr �machine_token_overlay_pathc C sB t j}t||t j �| _t||d�| _|| _d | _d | _d | _ d S )NF) r �MACHINE_TOKEN_FILEr r) � private_file� public_filerD �_machine_token� _entitlements�_contract_expiry_datetime)r r rD � file_namer r r r F 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- U s$ � ��� zMachineTokenFile.writec C s: t �� r| j�� | j�� d| _d| _d| _dS t� � �)z!Delete both pub and private filesN) r rP rG r5 rF rH rI rJ r rV r r r r r5 m 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 r3 rQ �loads�DatetimeAwareJSONDecoder� Exception)r � file_handlerr$ r r r r3 y 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 r3 rD �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_pathrM )�errorrd )r r r" r �UserFacingErrorr �&INVALID_PATH_FOR_MACHINE_TOKEN_OVERLAYr2 r r. rQ rZ r r[ � ValueError�ERROR_JSON_DECODING_IN_FILEr: )r rD � machine_token_overlay_content�er r r r` � s, ��������z,MachineTokenFile.parse_machine_token_overlayc C s t | j�r | jd d S d S )N�machineTokenInfo� accountInfo�r; rc 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 rc �get_entitlements_from_tokenr r r r � entitlements� s �z MachineTokenFile.entitlementsrc 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_overridesrl � contractInfoc s s$ � | ] }|� d �|� d�fV qdS )�type�tokenN��get��.0rk r r r � <genexpr>� s � � �z?MachineTokenFile.get_entitlements_from_token.<locals>.<genexpr>�resourceTokensc s s � | ] }|� d �|fV qdS )rt Nrv rx r r r rz � s � � ��resourceEntitlements� entitlement� resourceToken)�uaclient.contractrr rw �dict�items) rc rr rq rs �tokens_by_name� ent_by_name�entitlement_name� ent_value�entitlement_cfgr r r rp � 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.rl rs � effectiveToN)rJ rc rw 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.rn 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�rc rw 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 )Nrl rs �idr� r r r r � contract_id" s �z MachineTokenFile.contract_id)r r� )r r� ) r6 r7 r8 r r9 r: r r r� r- r5 r3 r< r# rc r` ro rq � staticmethodr rp r r� r� �intr� r� r� r� r� r r r r rC E sH �� � # rC )rQ �loggingr r �typingr r r �uaclientr r r r r r � uaclient.contract_data_typesr �get_event_logger�event� getLoggerr6 r0 r r= rC r r r r �<module> s 0
| ver. 1.4 |
Github
|
.
| PHP 8.2.9 | Generation time: 0.1 |
proxy
|
phpinfo
|
Settings