File: //proc/thread-self/root/lib/python3/dist-packages/uaclient/files/__pycache__/files.cpython-38.pyc
U
��Jh�
� @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z m
Z
m Z m Z e �
� Ze�e �e��ZG dd� d�ZG dd� d�ZG dd � d e�ZdS )
� N)�Any�Dict�Optional)�defaults� event_logger�
exceptions�system�utilc @ 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 sr | j r tjntj}z6tj�| j�tjkr:tj | jdd� n t� | j� W n t
k
r\ Y nX t � | j||� d S )Ni� )�mode)
r r ZROOT_READABLE_MODEZWORLD_READABLE_MODEr r �basenamer ZPRIVATE_SUBDIR�makedirs�OSErrorr Z
write_file)r r Z file_moder r r �write$ s ��z UAFile.writec C s< d }zt �| j�}W n" tk
r6 t�d| j� Y nX |S )Nz(Tried to load %s but file does not exist)r Z load_filer �FileNotFoundError�LOG�debug�r r r r r �read5 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 ZDEFAULT_DATA_DIR�str�boolr �propertyr r r r% r r* r+ r r r r r
s ��
r
c @ s^ e Zd Zed�dd�Zeeef d�dd�Ze eeef d�dd �Z
d
d � Z e d d
� �Z
dS )� ProJSONFile��pro_filec C s
|| _ d S r r3 )r r4 r r r r B s zProJSONFile.__init__r c C s | j jtj|tjd�d� d S )N��clsr )r4 r% �json�dumpsr ZDatetimeAwareJSONEncoderr) r r r r% H s �zProJSONFile.writer c
C sd | j �� }|r`ztj|tjd�W S tjk
r^ } z tj| j j dt
|� d��W 5 d }~X Y nX d S )Nr5 �
)�source�out) r4 r* r7 �loadsr ZDatetimeAwareJSONDecoderZJSONDecodeErrorr Z InvalidJsonr r/ )r r �er r r r* M s
�zProJSONFile.readc C s
| j �� S r )r4 r+ r r r r r+ Z s zProJSONFile.deletec C s | j jS r )r4 r r r r r r ] s zProJSONFile.is_presentN)r, r- r. r
r r r/ r r% r r* r+ r1 r r r r r r2 A s �
r2 c s" e Zd Zed�� fdd� Z� ZS )�
UserCacheFile)r c s t � j|t�� dd� d S )NF)r r
)�superr r Zget_user_cache_dir)r r �� __class__r r r c s
�zUserCacheFile.__init__)r, r- r. r/ r �
__classcell__r r r@ r r> b s r> )r7 Zloggingr �typingr r r Zuaclientr r r r r Zget_event_loggerZeventZ getLoggerZ replace_top_level_logger_namer, r' r
r2 r> r r r r �<module> s 5!