← Back
Editing: data_types.cpython-38.pyc
U ��Kj�7 � @ s d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z m Z m Z m Z m Z d dlmZmZmZ e�e�e��ZG dd� dej�ZG dd� de�ZG d d � d e�ZG d d � d e�ZG d d� de�ZG dd� d�Z e ddd�Z G dd� de e�Z G dd� de �Z G dd� de �Z G dd� de �Z!G dd � d e �Z"G d d � d e �Z#G d d � d e �Z$e e e e d!�d"d#�Z%e e e e d$�d%d&�Z&d5e e d(e'e(e)e*ef e*e'd)�d*d+�Z+d6ed, e*e,d)�d-d.�Z-G d/d0� d0�Z.e d1d(d�Z/G d2d(� d(e �Z0G d3d4� d4e0�Z1dS )7� N)�Enum)�Any�Dict�List�Optional�Type�TypeVar�Union)� exceptions�messages�utilc @ s e Zd ZejZdZdZdS )�IncorrectTypeErrorN)�__name__� __module__� __qualname__r Z E_INCORRECT_TYPE_ERROR_MESSAGE�_formatted_msg� expected_type�got_type� r r �5/usr/lib/python3/dist-packages/uaclient/data_types.pyr s r c s* e Zd ZejZeed�� fdd� Z� Z S )� IncorrectListElementTypeError��err�at_indexc s& t � j||jd� |j| _|j| _d S )N)�index� nested_msg��super�__init__�msgr r )�selfr r �� __class__r r r s z&IncorrectListElementTypeError.__init__) r r r r Z+E_INCORRECT_LIST_ELEMENT_TYPE_ERROR_MESSAGEr r �intr � __classcell__r r r! r r s r c s, e Zd ZejZeeed�� fdd� Z � Z S )� IncorrectDictElementTypeError�r �key�valuec s. t � j|t|�j|jd� |j| _|j| _d S )N)r' Z value_typer )r r �typer r r r )r r r' r( r! r r r s �z&IncorrectDictElementTypeError.__init__) r r r r Z+E_INCORRECT_DICT_ELEMENT_TYPE_ERROR_MESSAGEr r �strr r r$ r r r! r r% s r% c s. e Zd ZejZdZeed�� fdd� Z � Z S )�IncorrectFieldTypeErrorN�r r' c s& t � j||jd� |j| _|j| _d S )N)r' r r )r r r' r! r r r + s z IncorrectFieldTypeError.__init__) r r r r Z$E_INCORRECT_FIELD_TYPE_ERROR_MESSAGEr r' r r* r r$ r r r! r r+ ' s r+ c s6 e Zd ZejZeeee f e d�� fdd� Z � Z S )�IncorrectEnumValueError��values� enum_classc s8 t � j||jd� d�d�dd� |D ���| _d| _d S )Nr. z one of: {}z, c S s g | ] }t |��qS r )r* )�.0�vr r r � <listcomp>7 s z4IncorrectEnumValueError.__init__.<locals>.<listcomp>z<invalid_value>)r r r �format�joinr r )r r/ r0 r! r r r 4 s �z IncorrectEnumValueError.__init__) r r r r Z$E_INCORRECT_ENUM_VALUE_ERROR_MESSAGEr r r r* r# r r r$ r r r! r r- 1 s r- c @ s$ e Zd ZdZeeed�dd��ZdS )� DataValuez� Generic data value to be extended by more specific typed data values. This establishes the interface of a static/class method called `from_value` that returns the parsed value if appropriate. ��val�returnc C s | S �Nr �r8 r r r � from_valueC s zDataValue.from_valueN)r r r �__doc__� staticmethodr r< r r r r r6 < s r6 �E� EnumDataValue)Zboundc @ s* e Zd ZdZeee eed�dd��ZdS )r@ z� To be used for parsing enum values from_value raises an error if the value is not in the enum class values and returns the value if found. )�clsr8 r9 c C s> z | |�W S t k r8 dd� | D �}t|| d��Y nX d S )Nc S s g | ] }|j �qS r )r( )r1 �ir r r r3 W s z,EnumDataValue.from_value.<locals>.<listcomp>r. )� ValueErrorr- )rA r8 r/ r r r r< R s zEnumDataValue.from_valueN) r r r r= � classmethodr r? r r< r r r r r@ K s c @ s( e Zd ZdZdZeeed�dd��ZdS )�StringDataValuez� To be used for parsing string values from_value raises an error if the value is not a string and returns the string itself if it is a string. r* r7 c C s t | t�s tdt| �jd��| S )Nr* �r r )� isinstancer* r r) r r; r r r r< d s �zStringDataValue.from_valueN) r r r r= �python_type_namer> r r* r< r r r r rE [ s rE c @ s2 e Zd ZdZdZe�d�Zee e d�dd��Z dS )�StrictStringDataValuez� A string data value that rejects control characters (newline, carriage return, null byte) and spaces which could be used for injection attacks in apt sources files or auth config. r* z [\n\r\x00 ]r7 c C s8 t | t�s tdt| �jd��tj�| �r4tddd��| S )Nr* rF z str without control charactersz!str containing control characters)rG r* r r) r rI � _UNSAFE_CHARS�searchr; r r r r< w s � �z StrictStringDataValue.from_valueN) r r r r= rH �re�compilerJ r> r r* r< r r r r rI m s rI c @ s( e Zd ZdZdZeeed�dd��ZdS )� IntDataValuez� To be used for parsing int values from_value raises an error if the value is not a int and returns the int itself if it is a int. r# r7 c C s* t | t�rt | t�r&tdt| �jd��| S )Nr# rF )rG r# �boolr r) r r; r r r r< � s �zIntDataValue.from_valueN) r r r r= rH r> r r# r< r r r r rN � s rN c @ s( e Zd ZdZdZeeed�dd��ZdS )�FloatDataValuez� To be used for parsing float values from_value raises an error if the value is not a float and returns the float itself if it is a float. �floatr7 c C s. t | t�s&t | t�s&tdt| �jd��t| �S )NrQ rF )rG rQ r# r r) r r; r r r r< � s �zFloatDataValue.from_valueN) r r r r= rH r> r rQ r< r r r r rP � s rP c @ s( e Zd ZdZdZeeed�dd��ZdS )� BoolDataValuez� To be used for parsing bool values from_value raises an error if the value is not a bool and returns the bool itself if it is a bool. rO r7 c C s t | t�s tdt| �jd��| S )NrO rF )rG rO r r) r r; r r r r< � s �zBoolDataValue.from_valueN) r r r r= rH r> r rO r< r r r r rR � s rR c @ s* e Zd ZdZdZeeejd�dd��ZdS )�DatetimeDataValuez� This expects that value is a datetime. from_value raises an error if the value is not a datetime and returns the datetime itself if it is a datetime. �datetimer7 c C s" t | tj�s tdt| �jd��| S )NrT rF )rG rT r r) r r; r r r r< � s �z DatetimeDataValue.from_valueN) r r r r= rH r> r rT r< r r r r rS � s rS )�data_clsr9 c s G � fdd�dt �}|S )z� To be used for parsing lists of a certain DataValue type. Returns a class that extends DataValue and validates that each item in a list is the correct type in its from_value. c s( e Zd Z� Zeeed�� fdd� �ZdS )z data_list.<locals>._DataListr7 c sz t | t�s tdt| �jd��g }t| �D ]L\}}z|�� �|�� W q( tk rr } zt||d��W 5 d }~X Y q(X q(|S )N�listrF r ) rG rV r r) r � enumerate�appendr< r )r8 �new_valrB �item�e�rU r r r< � s � z'data_list.<locals>._DataList.from_valueN)r r r Zitem_clsr> r r r< r r\ r r � _DataList� s r] �r6 )rU r] r r\ r � data_list� s r_ )� value_clsr9 c s G � fdd�dt �}|S )z� To be used for parsing dicts of a certain DataValue type. Returns a class that extends DataValue and validates that each key, value is the correct type in its from_value. c s, e Zd ZdZ� Zeeed�� fdd� �ZdS )z data_dict.<locals>._DataDict�dictr7 c s� t | t�s tdt| �jd��i }| �� D ]R\}}z� �|�|t�|�< W q( tk rx } zt|||d��W 5 d }~X Y q(X q(|S )Nra rF r& ) rG ra r r) r �itemsr< rE r% )r8 rY r' r( r[ �r` r r r< � s" �� �z'data_dict.<locals>._DataDict.from_valueN) r r r rH Zdict_value_clsr> r r r< r rc r r � _DataDict� s rd r^ )r` rd r rc r � data_dict� s re T� DataObject)r8 � keep_noner9 c C s� g }| D ]z}t |t�r(|�|�|�� qt |t�rD|�t||�� qt |t�r`|�t||�� qt |t�rx|�|j � q|�|� q|S r: ) rG rf rX �to_dictrV �data_list_to_listra �data_dict_to_dictr r( )r8 rg rY rZ r r r ri s ri )r6 r6 c C s� i }| � � D ]t\}}t|t�r.|�|�||< q t|t�rHt||�||< q t|t�rbt||�||< q t|t�rx|j ||< q |||< q |S r: ) rb rG rf rh ra rj rV ri r r( )r8 rg rY r' r( r r r rj s rj c @ s4 e Zd ZdZdeee eee ee d�dd�Z dS )�FieldzB For defining the fields static property of a DataObject. TN�r' rU �required�dict_key�docc C s4 || _ || _|| _|d k r"|| _n| j | _|| _d S r: rl )r r' rU rm rn ro r r r r 3 s zField.__init__)TNN) r r r r= r* r r6 rO r r r r r r rk . s ��rk �Tc @ s� e Zd ZdZg Zdd� Zdd� Zdd� Zdee d �d d �Z dee d �d d�Z e dee e eed�dd��Ze ed�dd��ZdS )rf a� For defining a python object that can be parsed from a dict. Validates that a set of expected fields are present in the dict that is parsed and that the values of those fields are the correct DataValue by calling from_value on each. The fields are defined using the `fields` static property. DataObjects can be used in Fields of other DataObjects. To define a new DataObject: 1. Create a new class that extends DataObject. 2. Define the `fields` static property to be a list of Field objects 3. Define the constructor to take kwargs that match the list of Field objects. a. Example 1: Field("keyname", StringDataValue) -> keyname: str b. Example 2: Field("keyname", data_list(IntDataValue), required=False) -> keyname: Optional[List[int]] # noqa: E501 4. Use from_value or from_dict to parse a dict into the python object. c K s d S r: r )r Z_kwargsr r r r \ s zDataObject.__init__c C s: | j D ].}t| |jd �}t||jd �}||kr dS qdS )NFT)�fields�getattrr' )r �other�fieldZself_valZ other_valr r r �__eq___ s zDataObject.__eq__c C s d� | jj| �� �� �S )Nz{}{})r4 r" r rh �__repr__)r r r r rv g s �zDataObject.__repr__T)rg r9 c C s� i }| j D ]�}t| |jd �}d }t|t�r6|�|�}nBt|t�rLt||�}n,t|t�rbt ||�}nt|t �rt|j }n|}|d k s�|r |||j < q |S r: ) rq rr r' rG rf rh rV ri ra rj r r( rn )r rg �drt r8 rY r r r rh l s zDataObject.to_dict)� keep_nullr9 c C s t j| j|d�dtjd�S )N)rg T)Z sort_keysrA )�json�dumpsrh r ZDatetimeAwareJSONEncoder)r rx r r r �to_json� s �zDataObject.to_jsonF)rA rw � optional_type_errors_become_nullr9 c C s� i }| j D ]�}z||j }W n: tk rV |jrNtt|jjdd�|jd��nd }Y nX |d k r�z|j�|�}W nV tk r� } z8|js�|r�t � d|j |j |j � d }nt||jd��W 5 d }~X Y nX |||j < q | f |�S )NZnullrF r, zT%s is wrong type (expected %s but got %s) but considered optional - treating as null)rq rn �KeyErrorrm r+ r rU r r< �LOGZwarningr' r r )rA rw r| �kwargsrt r8 r[ r r r � from_dict� s@ �� � � zDataObject.from_dictr; c C s& t |t�s tdt|�jd��| �|�S )Nra rF )rG ra r r) r r� )rA r8 r r r r< � s �zDataObject.from_valueN)T)T)F)r r r r= rq r ru rv rO ra rh r* r{ rD r rp r� r r< r r r r rf H s" � �'c @ sD e Zd ZdZede�edee�dd�gZee e e d�dd�Z d S ) �AttachActionsConfigFilezj The format of the yaml file that can be passed with pro attach --attach-config /path/to/file �token�enable_servicesF)rm �r� r� c C s || _ || _d S r: r� )r r� r� r r r r � s z AttachActionsConfigFile.__init__N) r r r r= rk rE r_ rq r* r r r r r r r r� � s �r� )T)T)2rT ry ZloggingrL �enumr �typingr r r r r r r Zuaclientr r r Z getLoggerZ replace_top_level_logger_namer r~ ZUbuntuProErrorr r r% r+ r- r6 r? r@ rE rI rN rP rR rS r_ re rV r* r# rO ri ra rj rk rp rf r� r r r r �<module> sN $ " �� �� r
Save File
Cancel