Google
Edit File: structure.cpython-39.pyc
a �ic.2 � @ s� d dl Z d dlmZ d dlmZ d dlmZmZmZm Z m Z m Z g d�Z dZ G dd� de�ZG d d � d e�ZG d d � d e�ZG d d� de�ZG dd� ded�Zdd� ZG dd� de�Zddd�Zdd� ZdS )� N)�ABCMeta)�get_type_hints)� get_variant�get_type_arguments�unwrap_variant� is_base_type� Structure�List)�DBusStructureError�DBusData�generate_string_from_data� compare_dataZ__dbus_fields__c @ s e Zd ZdZdS )r z,General exception for DBus structure errors.N)�__name__� __module__� __qualname__�__doc__� r r �4/usr/lib/python3.9/site-packages/dasbus/structure.pyr ( s r c @ sj e Zd ZdZddd�Zedd� �Zedd� �Zed d � �Zed d � �Z d d� Z dd� Z dd� Z dd� Z dS )� DBusFieldz+Description of a field in a DBus structure.� c C s || _ || _|| _dS )z�Create a description of the field. :param name: a name of the field :param type_hint: a type hint :param description: a description N)�_name� _type_hint� _description)�self�name� type_hint� descriptionr r r �__init__0 s zDBusField.__init__c C s | j S )z4Name of the field. :return: a name )r �r r r r r ; s zDBusField.namec C s | j S )z>Type hint of the field. :return: a type hint )r r r r r r C s zDBusField.type_hintc C s | j S )zBDescription of the field. :return: a description )r r r r r r K s zDBusField.descriptionc C s | j �dd�S )zQReturn name of a data attribute. :return: a data attribute name �-�_)r �replacer r r r � data_nameS s zDBusField.data_namec C s t || j|� dS )zaSet the data attribute. :param obj: a data object :param value: a value N)�setattrr" �r �obj�valuer r r �set_data[ s zDBusField.set_datac C s | � |t|�� dS )ztSet the data attribute from a variant. :param obj: a data object :param variant: a variant N)r' r )r r% �variantr r r �set_data_variantc s zDBusField.set_data_variantc C s t || j�S )z\Get the data attribute. :param obj: a data object :return: a value )�getattrr" �r r% r r r �get_datak s zDBusField.get_datac C s t | j| �|��S )zkGet a variant of the data attribute. :param obj: a data object :return: a variant )r r r, r+ r r r �get_data_variants s zDBusField.get_data_variantN)r )r r r r r �propertyr r r r" r' r) r, r- r r r r r - s r c sB e Zd ZdZd � fdd� Zedd� �Z� fdd�Zd d � Z� Z S ) � DBusDataFieldz0Description of a data field in a DBus structure.r c s t � �|t|� || _dS �z�Create a description of the field. :param name: a name of the field :param data_type: a subclass of DBusData :param description: a description N)�superr r � _data_type�r r � data_typer �� __class__r r r s zDBusDataField.__init__c C s | j S �zMType of the data structure. :return: a subclass of DBusData �r2 r r r r r4 � s zDBusDataField.data_typec s | j �|�}t� �||� dS �zSet the data attribute.N)r2 �from_structurer1 r' r$ r5 r r r' � s zDBusDataField.set_datac C s | j �| �|��}t| j|�S �z$Get a variant of the data attribute.)r2 � to_structurer, r r r$ r r r r- � s z DBusDataField.get_data_variant)r � r r r r r r. r4 r' r- � __classcell__r r r5 r r/ | s r/ c sB e Zd ZdZd � fdd� Zedd� �Z� fdd�Zd d � Z� Z S ) �DBusDataListFieldz5Description of a data list field in a DBus structure.r c s t � �|tt |� || _dS r0 )r1 r r r r2 r3 r5 r r r � s zDBusDataListField.__init__c C s | j S r7 r8 r r r r r4 � s zDBusDataListField.data_typec s | j �|�}t� �||� dS r9 )r2 �from_structure_listr1 r' r$ r5 r r r' � s zDBusDataListField.set_datac C s | j �| �|��}t| j|�S r; )r2 �to_structure_listr, r r r$ r r r r- � s z"DBusDataListField.get_data_variant)r r= r r r5 r r? � s r? c sx e Zd ZdZ� fdd�Zeed�dd��Zeed�dd ��Zee e d �d d ��Z ee e d�d d��Z dd� Z � Z S )r z�Object representation of data in a DBus structure. Classes derived from this class should represent specific types of DBus structures. They will support a conversion from a DBus structure of this type to a Python object and back. c s( t � j|i |�� t| tt�| �� dS )zCreate a new data class.N)r1 �__init_subclass__r# �DBUS_FIELDS_ATTRIBUTE�DBusFieldFactory�generate_fields)�cls�args�kwargsr5 r r rB � s �zDBusData.__init_subclass__)� structurec C sl t |t�s td�t|�j���| � }t| �}|�� D ]2\}}|�|d�}|sZt d�|���|� ||� q4|S )z~Convert a DBus structure to a data object. :param structure: a DBus structure :return: a data object �Invalid type '{}'.NzField '{}' doesn't exist.) � isinstance�dict� TypeError�format�typer � get_fields�items�getr r) )rF rI �data�fieldsr r( �fieldr r r r: � s � �zDBusData.from_structure)�returnc C sN t || �s td�t|�j���i }t| �}|�� D ]\}}|�|�||< q2|S )zYConvert this data object to a DBus structure. :return: a DBus structure rJ )rK rM rN rO r rP rQ r- )rF rS rI rT r rU r r r r< � s �zDBusData.to_structure)� structuresc C s. t |t�s td�t|�j���tt| j|��S )z�Convert DBus structures to data objects. :param structures: a list of DBus structures :return: a list of data objects rJ )rK �listrM rN rO r �mapr: )rF rW r r r r@ � s �z DBusData.from_structure_listc C s t t| j|��S )z�Convert data objects to DBus structures. :param objects: a list of data objects :return: a list of DBus structures )rX rY r< )rF Zobjectsr r r rA s zDBusData.to_structure_listc C s t | �S )z%Convert this data object to a string.)r r r r r �__repr__ s zDBusData.__repr__)r r r r rB � classmethodr r: r< r r@ rA rZ r> r r r5 r r � s r )� metaclassc C s& t | td�}|du r"td�t���|S )zjReturn DBus fields of a data object. :param obj: a data object :return: a map of DBus fields Nz Fields are not defined at '{}'.)r* rC r rN )r% rT r r r rP s �rP c @ sL e Zd ZdZedd� �Zedd� �Zedd� �Zedd � �Zed d � �Z d S ) rD zA DBus field factory.c C s\ i }t �|�D ]<\}}| �||�s$q| �|�}| �||�}| �||�||< q|sXtd��|S )a� Generate DBus fields from properties of a class. Properties of the class will be used to generate a map of a DBus fields. The property should have a getter and a setter, otherwise an error is raised. The type hint of the getter is used to define the type of the DBus field. :param data_class: a data class :return: a map of DBus fields :raise DBusStructureError: if the DBus fields cannot be generated zNo fields found.)�inspectZ getmembers� _is_field�_get_field_name�_get_member_hint� _create_fieldr )rF Z data_classrT � member_name�memberr r r r r rE , s z DBusFieldFactory.generate_fieldsc C s |� d�rdS t|t�s dS dS )z�Is the member a representation of a DBus field? :param member_name: a name of the class member :param member: a class member :return: True or False r FT)� startswithrK r. )rF rb rc r r r r^ I s zDBusFieldFactory._is_fieldc C s |� dd�S )z�Get the name of the DBus field. :param member_name: a name of the class member :return: a name of the DBus field r r )r! )rF rb r r r r_ [ s z DBusFieldFactory._get_field_namec C sT |j std�|���|js(td�|���t|j�}|�dd�}|sPtd�|���|S )z�Get the type hint of the member. :param field_name: a name of the DBus field :param member: a class member :return: a type hint zField '{}' cannot be set.zField '{}' cannot be get.rV Nz Field '{}' has unknown type.)�fsetr rN �fgetr rR )rF � field_namerc Zgetter_type_hintsr r r r r` d s �� �z!DBusFieldFactory._get_member_hintc C sF t |t�rt||�S t |t�r<t|�\}t |t�r<t||�S t||�S )z�Create a representation of a DBus field. :param field_name: a name of the field :param member_hint: a type hint of the member :return: a new instance of DBus field )r r r/ r r r? r )rF rg Z member_hintZarg_hintr r r ra � s z DBusFieldFactory._create_fieldN) r r r r r[ rE r^ r_ r` ra r r r r rD ) s rD c C s� i }t | ��� D ]}|�| �||j< q|p,g D ]}|�|d� q.|pFi D ]}|| ||< qHtdd� |�� D ��}d�| jj d� |��S )a� Generate a string representation of a data object. Set the argument 'skip' to skip attributes with sensitive data. Set the argument 'add' to add other values to the string representation. The attributes in the string representation will be sorted alphabetically. :param obj: a data object :param skip: a list of names that should be skipped or None :param add: a dictionary of attributes to add or None :return: a string representation of the data object Nc S s g | ]\}}d � |t|���qS )z{}={})rN �repr)�.0r r& r r r � <listcomp>� s z-generate_string_from_data.<locals>.<listcomp>z{}({})z, ) rP �valuesr, r" �pop�sortedrQ rN r6 r �join)r% �skip�addZ dictionaryrU r Z attributesr r r r � s �r c C s( t | t�o&t |t�o&| �| �|�|�kS )z�Compare data of the given data objects. :param obj: a data object :param other: another data object :return: True if the data is equal, otherwise False )rK r r< )r% �otherr r r r � s ��r )NN)r] �abcr �typingr Z dasbus.typingr r r r r r �__all__rC � Exceptionr �objectr r/ r? r rP rD r r r r r r �<module> s O ]k