U
    "FZh9                     @   s8   d Z ddlmZ ddlmZ dd ZG dd deZdS )	z7Contains _ExtensionDict class to represent extensions.
    )type_checkers)FieldDescriptorc                 C   sj   t |tstd| |js*td|j |js>td|j |j| jk	rftd|j|jj| jjf dS )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_nameZcontaining_type
DESCRIPTOR)messageextension_handle r   n/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/site-packages/google/protobuf/internal/extension_dict.py_VerifyExtensionHandle   s"    
r   c                   @   sp   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd ZdS )_ExtensionDictzDict-like container for Extension fields on proto instances.

  Note that in all cases we expect extension handles to be
  FieldDescriptors.
  c                 C   s
   || _ dS )z^
    Args:
      extended_message: Message instance for which we are the Extensions dict.
    N)_extended_message)selfZextended_messager   r   r   __init__0   s    z_ExtensionDict.__init__c                 C   s   t | j| | jj|}|dk	r&|S |jtjkr@|| j}n|jtj	kr|j
}t|dsrddlm} || t|j
dsddlm} ||j
 |j
 }z|| jj W q tk
r   Y qX n|jS | jj||}|S )z8Returns the current value of the given extension handle.N_concrete_classr   )message_factory)r   r   _fieldsgetlabelr   LABEL_REPEATEDZ_default_constructorcpp_typeCPPTYPE_MESSAGEmessage_typehasattrZgoogle.protobufr   ZGetMessageClassr   Z_SetListenerZ_listener_for_childrenReferenceErrordefault_value
setdefault)r   r
   resultr   r   r   r   r   __getitem__7   s2    


 z_ExtensionDict.__getitem__c                 C   sH   t || jsdS | j }|j }dd |D }dd |D }||kS )NFc                 S   s   g | ]}|j r|qS r   r   .0fieldr   r   r   
<listcomp>h   s      z)_ExtensionDict.__eq__.<locals>.<listcomp>c                 S   s   g | ]}|j r|qS r   r!   r"   r   r   r   r%   i   s      )r   	__class__r   
ListFields)r   otherZ	my_fieldsZother_fieldsr   r   r   __eq__`   s    

z_ExtensionDict.__eq__c                 C   s
   | |k S Nr   )r   r(   r   r   r   __ne__m   s    z_ExtensionDict.__ne__c                 C   s    | j  }dd |D }t|S )Nc                 S   s   g | ]}|d  j r|qS )r   r!   r"   r   r   r   r%   s   s     
 z*_ExtensionDict.__len__.<locals>.<listcomp>)r   r'   len)r   fieldsZextension_fieldsr   r   r   __len__p   s    
z_ExtensionDict.__len__c                 C   s   t dd S )Nzunhashable object)	TypeErrorr   r   r   r   __hash__v   s    z_ExtensionDict.__hash__c                 C   s\   t | j| |jtjks$|jtjkr2td|j t	
|}||| jj|< | j  dS )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   ZGetTypeCheckerZ
CheckValuer   Z	_Modified)r   r
   valueZtype_checkerr   r   r   __setitem__}   s    


z_ExtensionDict.__setitem__c                 C   s   | j | d S r*   )r   ZClearExtension)r   r
   r   r   r   __delitem__   s    z_ExtensionDict.__delitem__c                 C   s"   | j j}|jjj| }||dS )zTries to find a known extension with the specified name.

    Args:
      name: Extension full name.

    Returns:
      Extension field descriptor.
    N)r   r   filepoolZ_extensions_by_namer   )r   name
descriptor
extensionsr   r   r   _FindExtensionByName   s    	z#_ExtensionDict._FindExtensionByNamec                 C   s"   | j j}|jjj| }||dS )zTries to find a known extension with the field number.

    Args:
      number: Extension field number.

    Returns:
      Extension field descriptor.
    N)r   r   r5   r6   Z_extensions_by_numberr   )r   numberr8   r9   r   r   r   _FindExtensionByNumber   s    	z%_ExtensionDict._FindExtensionByNumberc                 C   s   dd | j  D S )Nc                 s   s    | ]}|d  j r|d  V  qdS )r   Nr!   )r#   fr   r   r   	<genexpr>   s    
z*_ExtensionDict.__iter__.<locals>.<genexpr>)r   r'   r0   r   r   r   __iter__   s    z_ExtensionDict.__iter__c                 C   sf   t | j| || jjkrdS |jtjkr:t| jj|S |jtj	krb| jj|}|d k	o`|j
S dS )NFT)r   r   r   r   r   r   boolr   r   r   Z_is_present_in_parent)r   r
   r2   r   r   r   __contains__   s    z_ExtensionDict.__contains__N)__name__
__module____qualname____doc__r   r    r)   r+   r.   r1   r3   r4   r:   r<   r?   rA   r   r   r   r   r   (   s   )r   N)rE   Zgoogle.protobuf.internalr   Zgoogle.protobuf.descriptorr   r   objectr   r   r   r   r   <module>   s   