OwlCyberSecurity - MANAGER
Edit File: files.cpython-38.pyc
U Uٵg� � @ 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!