File: //lib/python3.9/site-packages/google/protobuf/internal/__pycache__/extension_dict.cpython-39.pyc
a
x��g� � @ s8 d Z ddlmZ ddlmZ dd� ZG dd� de�ZdS ) z7Contains _ExtensionDict class to represent extensions.
� )�
type_checkers)�FieldDescriptorc C sj t |t�std| ��|js*td|j ��|js>td|j ��|j| jurftd|j|jj| jjf ��dS )z0Verify that the given extension handle is valid.z3HasExtension() expects an extension handle, got: %sz"%s" is not an extension.z""%s" is missing a containing_type.zKExtension "%s" extends message type "%s", but this message is of type "%s".N)�
isinstancer �KeyError� is_extension� full_nameZcontaining_typeZ
DESCRIPTOR)�message�extension_handle� r
�K/usr/lib/python3.9/site-packages/google/protobuf/internal/extension_dict.py�_VerifyExtensionHandle& s"
�� ��r c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�_ExtensionDictz�Dict-like container for Extension fields on proto instances.
Note that in all cases we expect extension handles to be
FieldDescriptors.
c C s
|| _ dS )z^
Args:
extended_message: Message instance for which we are the Extensions dict.
N)�_extended_message)�selfZextended_messager
r
r �__init__G s z_ExtensionDict.__init__c C s� t | j|� | jj�|�}|dur&|S |jtjkr@|�| j�}n�|jtj kr�|j
}t |d�sj| jj �
|� t|j
dd�s�J d|j|j
jf ��|j
�� }z|�| jj� W q� ty� Y q�0 n|jS | jj�||�}|S )z8Returns the current value of the given extension handle.N�_concrete_classzAUninitialized concrete class found for field %r (message type %r))r r �_fields�get�labelr �LABEL_REPEATEDZ_default_constructor�cpp_type�CPPTYPE_MESSAGE� message_type�hasattrZ_FACTORYZ GetPrototype�getattrr r Z _SetListenerZ_listener_for_children�ReferenceErrorZ
default_value�
setdefault)r r �resultr r
r
r � __getitem__N s6
���
�z_ExtensionDict.__getitem__c C sH t || j�sdS | j�� }|j�� }dd� |D �}dd� |D �}||kS )NFc S s g | ]}|j r|�qS r
�r ��.0Zfieldr
r
r �
<listcomp> � z)_ExtensionDict.__eq__.<locals>.<listcomp>c S s g | ]}|j r|�qS r
r r r
r
r r" � r# )r � __class__r �
ListFields)r �otherZ my_fieldsZ other_fieldsr
r
r �__eq__w s
z_ExtensionDict.__eq__c C s
| |k S �Nr
)r r& r
r
r �__ne__� s z_ExtensionDict.__ne__c C s | j �� }dd� |D �}t|�S )Nc S s g | ]}|d j r|�qS )r r r r
r
r r" � r# z*_ExtensionDict.__len__.<locals>.<listcomp>)r r% �len)r �fieldsZextension_fieldsr
r
r �__len__� s
z_ExtensionDict.__len__c C s t d��d S )Nzunhashable object)� TypeError�r r
r
r �__hash__� s z_ExtensionDict.__hash__c C s\ t | j|� |jtjks$|jtjkr2td|j ��t �
|�}|� |�| jj |< | j�
� dS )zlIf extension_handle specifies a non-repeated, scalar extension
field, sets the value of that field.
zKCannot assign to extension "%s" because it is a repeated or composite type.N)r r r r r r r r- r r ZGetTypeCheckerZ
CheckValuer Z _Modified)r r �valueZ type_checkerr
r
r � __setitem__� s
���
�
z_ExtensionDict.__setitem__c C s | j �|� d S r( )r ZClearExtension)r r r
r
r � __delitem__� s z_ExtensionDict.__delitem__c C s | j j�|d�S )z�Tries to find a known extension with the specified name.
Args:
name: Extension full name.
Returns:
Extension field descriptor.
N)r Z_extensions_by_namer )r �namer
r
r �_FindExtensionByName� s z#_ExtensionDict._FindExtensionByNamec C s | j j�|d�S )z�Tries to find a known extension with the field number.
Args:
number: Extension field number.
Returns:
Extension field descriptor.
N)r Z_extensions_by_numberr )r �numberr
r
r �_FindExtensionByNumber� s z%_ExtensionDict._FindExtensionByNumberc C s dd� | j �� D �S )Nc s s | ]}|d j r|d V qdS )r Nr )r! �fr
r
r � <genexpr>� s
�z*_ExtensionDict.__iter__.<locals>.<genexpr>)r r% r. r
r
r �__iter__� s z_ExtensionDict.__iter__c C sf t | j|� || jjvr dS |jtjkr:t| jj�|��S |jtj krb| jj�|�}|d uo`|j
S dS )NFT) r r r r r r �boolr r r Z_is_present_in_parent)r r r0 r
r
r � __contains__� s z_ExtensionDict.__contains__N)�__name__�
__module__� __qualname__�__doc__r r r' r) r, r/ r1 r2 r4 r6 r9 r; r
r
r
r r
? s )
r
N)r? Zgoogle.protobuf.internalr Zgoogle.protobuf.descriptorr r �objectr
r
r
r
r �<module> s