⚝
One Hat Cyber Team
⚝
Your IP:
216.73.216.90
Server IP:
104.21.64.1
Server:
Linux vmi2315822.contaboserver.net 5.15.0-134-generic #145-Ubuntu SMP Wed Feb 12 20:08:39 UTC 2025 x86_64
Server Software:
LiteSpeed
PHP Version:
8.3.21
Buat File
|
Buat Folder
Eksekusi
Dir :
~
/
lib
/
python3
/
dist-packages
/
uaclient
/
__pycache__
/
View File Name :
data_types.cpython-310.pyc
o l�g�4 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m Z m Z m Z d dl mZmZmZ 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�Ze ddd�Z G dd� dee�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd � d e�Z!G d d � d e�Z"d e e d e e fd!d"�Z#d#e e d e e fd$d%�Z$ &d5d'ee d(e%e&e'e(ef d)e(d e%fd*d+�Z) &d5d'ed, d)e(d e*fd-d.�Z+G d/d0� d0�Z,e d1d(d�Z-G d2d(� d(e�Z.G d3d4� d4e.�Z/dS )6� N)�Enum)�Any�Dict�List�Optional�Type�TypeVar�Union)� exceptions�messages�utilc @ s e Zd ZejZdZdZdS )�IncorrectTypeErrorN)�__name__� __module__� __qualname__r � 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 ZejZdedef� fdd� Z� Z S )� IncorrectListElementTypeError�err�at_indexc �& t � j||jd� |j| _|j| _d S )N)�index� nested_msg��super�__init__�msgr r )�selfr r �� __class__r r r � z&IncorrectListElementTypeError.__init__) r r r r �+E_INCORRECT_LIST_ELEMENT_TYPE_ERROR_MESSAGEr r �intr � __classcell__r r r" r r s r c s0 e Zd ZejZdededef� fdd� Z � Z S )� IncorrectDictElementTypeErrorr �key�valuec s. t � j|t|�j|jd� |j| _|j| _d S )N)r) � value_typer )r r �typer r r r )r! r r) r* r" r r r s � z&IncorrectDictElementTypeError.__init__) r r r r �+E_INCORRECT_DICT_ELEMENT_TYPE_ERROR_MESSAGEr r �strr r r' r r r" r r( s "r( c s0 e Zd ZejZdZdedef� fdd� Z � Z S )�IncorrectFieldTypeErrorNr r) c r )N)r) r r )r! r r) r" r r r * r$ z IncorrectFieldTypeError.__init__) r r r r �$E_INCORRECT_FIELD_TYPE_ERROR_MESSAGEr r) r r. r r' r r r" r r/ & s r/ c s8 e Zd ZejZdeeee f de f� fdd� Z � Z S )�IncorrectEnumValueError�values� enum_classc s8 t � j||jd� d�d�dd� |D ���| _d| _d S )N�r2 r3 z one of: {}z, c S s g | ]}t |��qS r )r. )�.0�vr r r � <listcomp>6 s z4IncorrectEnumValueError.__init__.<locals>.<listcomp>z<invalid_value>)r r r �format�joinr r )r! r2 r3 r" r r r 3 s � z IncorrectEnumValueError.__init__) r r r r �$E_INCORRECT_ENUM_VALUE_ERROR_MESSAGEr r r r. r& r r r' r r r" r r1 0 s *r1 c @ s& e Zd ZdZededefdd��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 �r< r r r � from_valueB s zDataValue.from_valueN)r r r �__doc__� staticmethodr r@ r r r r r; ; s r; �E� EnumDataValue)�boundc @ s. e Zd ZdZedee dedefdd��ZdS )rD 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. �clsr< r= c C s4 z| |�W S t y dd� | D �}t|| d��w )Nc S s g | ]}|j �qS r )r* )r5 �ir r r r7 V s z,EnumDataValue.from_value.<locals>.<listcomp>r4 )� ValueErrorr1 )rF r< r2 r r r r@ Q s �zEnumDataValue.from_valueN) r r r rA � classmethodr rC r r@ r r r r rD J s c @ �* e Zd ZdZdZededefdd��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. r< r= c C � t | t�stdt| �jd��| S )Nr. �r r )� isinstancer. r r, r r? r r r r@ c � �zStringDataValue.from_valueN) r r r rA �python_type_namerB r r. r@ r r r r rK Z � rK c @ rJ )� 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& r< r= c C s* t | t�r t | t�rtdt| �jd��| S )Nr& rM )rN r& �boolr r, r r? r r r r@ u s �zIntDataValue.from_valueN) r r r rA rP rB r r& r@ r r r r rR l rQ rR c @ rJ )�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. �floatr< r= c C s. t | t�st | t�stdt| �jd��t| �S )NrU rM )rN rU r& r r, r r? r r r r@ � s �zFloatDataValue.from_valueN) r r r rA rP rB r rU r@ r r r r rT ~ rQ rT c @ rJ )� 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. rS r< r= c C rL )NrS rM )rN rS r r, r r? r r r r@ � rO zBoolDataValue.from_valueN) r r r rA rP rB r rS r@ r r r r rV � rQ rV c @ s, e Zd ZdZdZededejfdd��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. �datetimer< r= c C s" t | tj�stdt| �jd��| S )NrX rM )rN rX r r, r r? r r r r@ � s �z DatetimeDataValue.from_valueN) r r r rA rP rB r rX r@ r r r r rW � s rW �data_clsr= c � 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� Zededef� fdd� �ZdS )z data_list.<locals>._DataListr< r= c sn t | t�stdt| �jd��g }t| �D ] \}}z |�� �|�� W q ty4 } zt||d��d }~ww |S )N�listrM )r r ) rN r[ r r, r � enumerate�appendr@ r )r< �new_valrG �item�e�rY r r r@ � s � ��z'data_list.<locals>._DataList.from_valueN)r r r �item_clsrB r r r@ r ra r r � _DataList� s rc �r; )rY rc r ra r � data_list� s re � value_clsc rZ )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� Zededef� fdd� �ZdS )z data_dict.<locals>._DataDict�dictr< r= c st t | t�stdt| �jd��i }| �� D ]#\}}z � �|�|t�|�< W q ty7 } zt|||d��d }~ww |S )Nrg rM )r r) r* ) rN rg r r, r �itemsr@ rK r( )r< r^ r) r* r` �rf r r r@ � s �����z'data_dict.<locals>._DataDict.from_valueN) r r r rP �dict_value_clsrB r r r@ r ri r r � _DataDict� s rk rd )rf rk r ri r � data_dict� s rl Tr< � DataObject� keep_nonec C s� g }| D ]=}t |t�r|�|�|�� qt |t�r"|�t||�� qt |t�r0|�t||�� qt |t�r<|�|j � q|�|� q|S r> ) rN rm r] �to_dictr[ �data_list_to_listrg �data_dict_to_dictr r* )r< rn r^ r_ r r r rp � s rp )r; r; c C s� i }| � � D ]:\}}t|t�r|�|�||< qt|t�r$t||�||< qt|t�r1t||�||< qt|t�r<|j ||< q|||< q|S r> ) rh rN rm ro rg rq r[ rp r r* )r< rn r^ r) r* r r r rq s rq c @ sB e Zd ZdZ d dedee dedee dee f d d �Z dS ) �FieldzB For defining the fields static property of a DataObject. TNr) rY �required�dict_key�docc C s4 || _ || _|| _|d ur|| _n| j | _|| _d S r> )r) rY rs rt ru )r! r) rY rs rt ru r r r r s zField.__init__)TNN) r r r rA r. r r; rS r r r r r r rr s ������rr �Tc @ s� e Zd ZdZg Zdd� Zdd� Zdd� Zdd ed e fd d �Z dd ed e fdd�Z e ddee de ded efdd��Ze defdd��ZdS ) rm 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! �_kwargsr r r r C s zDataObject.__init__c C s: | j D ]}t| |jd �}t||jd �}||kr dS qdS )NFT)�fields�getattrr) )r! �other�field�self_val� other_valr r r �__eq__F s �zDataObject.__eq__c C s d� | jj| �� �� �S )Nz{}{})r8 r# r ro �__repr__)r! r r r r N s �zDataObject.__repr__Trn r= c C s� i }| j D ]B}t| |jd �}d }t|t�r|�|�}n!t|t�r&t||�}nt|t�r1t ||�}n t|t �r:|j }n|}|d usB|rG|||j < q|S r> ) rx ry r) rN rm ro r[ rp rg rq r r* rt )r! rn �dr{ r< r^ r r r ro S s" �zDataObject.to_dict� keep_nullc C s t j| j|d�dtjd�S )N)rn T)� sort_keysrF )�json�dumpsro r �DatetimeAwareJSONEncoder)r! r� r r r �to_jsoni s �zDataObject.to_jsonFrF r� � optional_type_errors_become_nullc C s� i }| j D ]c}z||j }W n ty) |jr%tt|jjdd�|jd��d }Y nw |d urcz|j�|�}W n, tyb } z |jsQ|rQt � d|j |j |j � d }nt||jd��W Y d }~nd }~ww |||j < q| di |��S )N�nullrM )r r) zT%s is wrong type (expected %s but got %s) but considered optional - treating as nullr )rx rt �KeyErrorrs r/ r rY r r@ �LOG�warningr) r r )rF r� r� �kwargsr{ r< r` r r r � from_dictp sF ��� �� ��� zDataObject.from_dictr< c C s& t |t�stdt|�jd��| �|�S )Nrg rM )rN rg r r, r r� )rF r< r r r r@ � s � zDataObject.from_valueN�T)F)r r r rA rx r r~ r rS rg ro r. r� rI r rv r� r r@ r r r r rm / s* ���� �'c @ sF e Zd ZdZede�edee�dd�gZdede e e fdd�Z dS ) �AttachActionsConfigFilezj The format of the yaml file that can be passed with pro attach --attach-config /path/to/file �token�enable_servicesF)rs c C s || _ || _d S r> )r� r� )r! r� r� r r r r � s z AttachActionsConfigFile.__init__N) r r r rA rr rK re rx r. r r r r r r r r� � s � r� r� )0rX r� �logging�enumr �typingr r r r r r r �uaclientr r r � getLogger� replace_top_level_logger_namer r� �UbuntuProErrorr r r( r/ r1 r; rC rD rK rR rT rV rW re rl r[ r. r& rS rp rg rq rr rv rm r� r r r r �<module> sT $ "��� ���� � r