o
    {j6h9                     @   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ur3t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_namecontaining_type
DESCRIPTOR)messageextension_handle r   v/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/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)selfextended_messager   r   r   __init__0   s   
z_ExtensionDict.__init__c                 C   s   t | j| | jj|}|dur|S |jtjkr || j}nF|jtj	krc|j
}t|ds9ddlm} || t|j
dsKddlm} ||j
 |j
 }z	|| jj W n tyb   Y nw |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_REPEATED_default_constructorcpp_typeCPPTYPE_MESSAGEmessage_typehasattrgoogle.protobufr   GetMessageClassr   _SetListener_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      g | ]}|j r|qS r   r   .0fieldr   r   r   
<listcomp>h       z)_ExtensionDict.__eq__.<locals>.<listcomp>c                 S   r(   r   r)   r*   r   r   r   r-   i   r.   )r   	__class__r   
ListFields)r   other	my_fieldsother_fieldsr   r   r   __eq__`   s   

z_ExtensionDict.__eq__c                 C   s
   | |k S Nr   )r   r1   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   r0   len)r   fieldsextension_fieldsr   r   r   __len__p   s   
z_ExtensionDict.__len__c                 C   s   t d)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rt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   GetTypeChecker
CheckValuer   	_Modified)r   r   valuetype_checkerr   r   r   __setitem__}   s   

z_ExtensionDict.__setitem__c                 C   s   | j | d S r5   )r   ClearExtension)r   r   r   r   r   __delitem__   s   z_ExtensionDict.__delitem__c                 C   "   | 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pool_extensions_by_namer   )r   name
descriptor
extensionsr   r   r   _FindExtensionByName      	z#_ExtensionDict._FindExtensionByNamec                 C   rF   )zTries to find a known extension with the field number.

    Args:
      number: Extension field number.

    Returns:
      Extension field descriptor.
    N)r   r	   rG   rH   _extensions_by_numberr   )r   numberrK   rL   r   r   r   _FindExtensionByNumber   rN   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   r0   r<   r   r   r   __iter__   s   z_ExtensionDict.__iter__c                 C   sf   t | j| || jjvrdS |jtjkrt| jj|S |jtj	kr1| jj|}|d uo0|j
S dS )NFT)r   r   r   r   r   r   boolr   r   r   _is_present_in_parent)r   r   rA   r   r   r   __contains__   s   z_ExtensionDict.__contains__N)__name__
__module____qualname____doc__r   r'   r4   r6   r:   r=   rC   rE   rM   rQ   rT   rW   r   r   r   r   r   (   s    )r   N)r[   google.protobuf.internalr   google.protobuf.descriptorr   r   objectr   r   r   r   r   <module>   s
   