← Back
Editing: data_types.cpython-38.pyc
U Uٵg� � @ s� d dl Z d dlmZ d dlmZmZmZmZmZm Z d dl m Z d dl m Z d dlmZ d dlmZ d dlmZ d d lmZmZ G d d � d e�Ze d e d �ZG dd� dee �ZdS )� N)�Enum)�Callable�Dict�Generic�Optional�Type�TypeVar)� exceptions)� DataObject)�UAFile)�DatetimeAwareJSONDecoder)�parser)� safe_dump� safe_loadc @ s e Zd ZdZdZdS )�DataObjectFileFormat�json�yamlN)�__name__� __module__� __qualname__�JSON�YAML� r r �;/usr/lib/python3/dist-packages/uaclient/files/data_types.pyr s r �DOFType)Zboundc @ sr e Zd Zejddfee eeee e ge f e d�dd�Z ee d�dd�Z ed �d d �Zd d � Zedd� �ZdS )�DataObjectFileNF��data_object_cls�ua_file� file_format�preprocess_data� optional_type_errors_become_nullc C s" || _ || _|| _|| _|| _d S �Nr )�selfr r r r r! r r r �__init__ s �zDataObjectFile.__init__)�returnc C s� | j �� }|d krd S d }| jtjkrdztj|td�}W q� tjk r` t j | j j dd��Y q�X nB| jtj kr�z t |�}W n( tjk r� t j | j j dd��Y nX |d kr�d S | jr�| �|�}| jj|| jd�S )N)�clsr )� file_namer r )r! )r �readr r r r �loadsr ZJSONDecodeErrorr ZInvalidFileFormatError�pathr r � yaml_parserZ ParserErrorr r Z from_dictr! )r# �raw_dataZ parsed_datar r r r( & s> � � � �zDataObjectFile.read)�contentc C sF | j tjkr|�� }n | j tjkr6|�� }t|dd�}| j�|� d S )NF)Zdefault_flow_style) r r r Zto_jsonr Zto_dictr r �write)r# r- Z str_content�datar r r r. H s zDataObjectFile.writec C s | j �� d S r" )r �delete�r# r r r r0 Q s zDataObjectFile.deletec C s | j jS r" )r r* r1 r r r r* T s zDataObjectFile.path)r r r r r r r r r r r �boolr$ r( r. r0 �propertyr* r r r r r s �� " r )r �enumr �typingr r r r r r Zuaclientr Zuaclient.data_typesr Zuaclient.files.filesr Z uaclient.utilr Z uaclient.yamlr r+ r r r r r r r r r �<module> s
Save File
Cancel