File: //lib/python3.9/site-packages/google/protobuf/internal/__pycache__/python_message.cpython-39.pyc
a
x��g$� � @ s� d Z dZddlmZ ddlZddlZddlZddlZddlm Z ddl
m Z ddl
m Z ddl
m
Z
dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ ddlmZ ddlmZ ddlmZ ej Z dZ ej Z G dd� de �Z!dd� Z"dd� Z#dd� 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.d0d1� Z/d2d3� Z0d4d5� Z1G d6d7� d7e2�Z3d8d9� Z4d:d;� Z5d<d=� Z6d>d?� Z7d@dA� Z8dBdC� Z9dDdE� Z:dFZ;dGZ<dHdI� Z=dJdK� Z>dLdM� Z?dNdO� Z@dPdQ� ZAdRdS� ZBdTdU� ZCdVdW� ZDdXdY� ZEdZd[� ZFd\d]� ZGd^d_� ZHd`da� ZIdbdc� ZJddde� ZKdfdg� ZLdhdi� ZMdjdk� ZNdldm� ZOdndo� ZPdpdq� ZQdrds� ZRdtdu� ZSG dvdw� dweT�ZUG dxdy� dyeU�ZVdS )za
Contains a metaclass and helper functions used to create
protocol message classes from Descriptor objects at runtime.
Recall that a metaclass is the "type" of a class.
(A class is to a metaclass what an instance is to a class.)
In this case, we use the GeneratedProtocolMessageType metaclass
to inject all the useful functionality into the classes
output by the protocol compiler at compile-time.
The upshot of all this is that the real implementation
details for ALL pure-Python protocol buffers are *here in
this file*.
z#robinson@google.com (Will Robinson)� )�BytesION)�range)�api_implementation)�
containers)�decoder)�encoder)�enum_type_wrapper)�extension_dict)�message_listener)�
type_checkers)�well_known_types)� wire_format)�
descriptor��message)� text_formatzgoogle.protobuf.Anyc s0 e Zd ZdZdZ� fdd�Z� fdd�Z� ZS )� GeneratedProtocolMessageTypeaQ Metaclass for protocol message classes created at runtime from Descriptors.
We add implementations for all methods described in the Message class. We
also create properties to allow getting/setting all fields in the protocol
message. Finally, we create slots to prevent users from accidentally
"setting" nonexistent fields in the protocol message, which then wouldn't get
serialized / deserialized properly.
The protocol compiler currently uses this metaclass to create protocol
message classes at runtime. Clients can also manually create their own
classes at runtime, as in this example:
mydescriptor = Descriptor(.....)
factory = symbol_database.Default()
factory.pool.AddDescriptor(mydescriptor)
MyProtoClass = factory.GetPrototype(mydescriptor)
myproto_instance = MyProtoClass()
myproto.foo_field = 23
...
�
DESCRIPTORc s� |t j }t|t�r td��t|dd�}|r0|S |jtjv rN|tj|j f7 }t ||� t
||� t t | �}|� | |||�}|S )ak Custom allocation for runtime-generated class types.
We override __new__ because this is apparently the only place
where we can meaningfully set __slots__ on the class we're creating(?).
(The interplay between metaclasses and slots is not very well-documented).
Args:
name: Name of the class (ignored, but required by the
metaclass protocol).
bases: Base classes of the class we're constructing.
(Should be message.Message). We ignore this field, but
it's required by the metaclass protocol
dictionary: The class dictionary of the class we're
constructing. dictionary[_DESCRIPTOR_KEY] must contain
a Descriptor object describing this protocol message
type.
Returns:
Newly-allocated class.
Raises:
RuntimeError: Generated code only work with python cpp extension.
z\The generated code only work with python cpp extension, but it is using pure python runtime.�_concrete_classN)
r �_DESCRIPTOR_KEY�
isinstance�str� RuntimeError�getattr� full_namer ZWKTBASES�&_AddClassAttributesForNestedExtensions� _AddSlots�super�__new__)�cls�name�bases�
dictionaryr � new_class�
superclass�� __class__� �K/usr/lib/python3.9/site-packages/google/protobuf/internal/python_message.pyr k s
z$GeneratedProtocolMessageType.__new__c s� |t j }t|dd�}|r4|| u s0J d|j ��dS i | _|jr`|�� jr`t� |�df| jtj
< |j D ]}t | |� qf| |_
t|| � t|| � t|| � t|| � t| � t|| � t|| � tt | �}|�|||� dS )a� Here we perform the majority of our work on the class.
We add enum getters, an __init__ method, implementations
of all Message methods, and properties for all fields
in the protocol type.
Args:
name: Name of the class (ignored, but required by the
metaclass protocol).
bases: Base classes of the class we're constructing.
(Should be message.Message). We ignore this field, but
it's required by the metaclass protocol
dictionary: The class dictionary of the class we're
constructing. dictionary[_DESCRIPTOR_KEY] must contain
a Descriptor object describing this protocol message
type.
r NzBDuplicate `GeneratedProtocolMessageType` created for descriptor %r)r r r r �_decoders_by_tag� has_options�
GetOptions�message_set_wire_formatr ZMessageSetItemDecoderZMESSAGE_SET_ITEM_TAG�fields�_AttachFieldHelpersr �_AddEnumValues�_AddInitMethod�_AddPropertiesForFields�_AddPropertiesForExtensions�_AddStaticMethods�_AddMessageMethods�_AddPrivateHelperMethodsr �__init__)r r r! r" r Zexisting_class�fieldr$ r% r' r( r6 � s6
���
�
z%GeneratedProtocolMessageType.__init__)�__name__�
__module__� __qualname__�__doc__r r r6 �
__classcell__r' r' r% r( r P s 5r c C s | S )a Returns the name of the public property attribute which
clients can use to get and (in some cases) set the value
of a protocol message field.
Args:
proto_field_name: The protocol message field name, exactly
as it appears (or would appear) in a .proto file.
r' ��proto_field_namer' r' r( �
_PropertyName� s r? c C s g d�|d< dS )a Adds a __slots__ entry to dictionary, containing the names of all valid
attributes for this message type.
Args:
message_descriptor: A Descriptor instance describing this message type.
dictionary: Class dictionary to which we'll add a '__slots__' entry.
)
�_cached_byte_size�_cached_byte_size_dirty�_fields�_unknown_fields�_unknown_field_set�_is_present_in_parent� _listener�_listener_for_children� __weakref__�_oneofs� __slots__Nr' )�message_descriptorr" r' r' r( r � s r c C s2 | j o0| jjo0| j�� jo0| jtjko0| jtj kS �N)
� is_extension�containing_typer* r+ r, �type�_FieldDescriptor� TYPE_MESSAGE�labelZLABEL_OPTIONAL�r7 r' r' r( �_IsMessageSetExtension s �
�
�
�rT c C s | j tjko | jjo | j�� jS rL )rO rP rQ � message_typer* r+ � map_entryrS r' r' r( � _IsMapField s
�
�rW c C s | j jd }|jtjkS )N�value)rU �fields_by_name�cpp_typerP �CPPTYPE_MESSAGE)r7 Z
value_typer' r' r( �_IsMessageMapField s r\ c C s | j jdkrdS d}|S )N�proto3FT)rN �syntax)r7 Z enforce_utf8r' r' r( �_IsStrictUtf8Check s r_ c sN �j tjk��ot��j�}�jjdk�|s2d}nF�jjdkrP�joL�� � j
}n(�jop�� � � d�op�� � j
dk}| }t ����r�t
���}t
��t���}nNt��r�t
��j�}t
��j�}n,tj�j �j�|�}tj�j �j�|�}|�_|�_t���_� ����fdd�}|tj �j d� ��rJt��j��rJ|tj d� d S )Nr] FZproto2�packedc s t ��j| �}�j}|tjkr.t���r.tj}d }d}�j d urF�}n�r^�s^�j
tj kr^d}�r~t ��}t
��t��|�}nr|tjkr�t��}t
��j�|��j||�}nD�j
tj kr�tj| �j�|��j�}n tj| �j�|��j|�}||f� j|< d S �NFT)r ZTagBytes�numberrO rP Z TYPE_ENUMr ZSupportsOpenEnumsZ
TYPE_INT32�containing_oneofrZ r[ r\ r Z
MapDecoder�_GetInitializeDefaultForMapZ TYPE_STRINGr_ Z
StringDecoder�_default_constructorZTYPE_TO_DECODERr) ) Zwiretype� is_packed� tag_bytesZ decode_typeZoneof_descriptorZclear_if_defaultZis_message_map�
field_decoderZis_strict_utf8_check�r �field_descriptorZ is_map_entry� is_proto3Z is_repeatedr' r( �
AddDecoderI sL
�
��
� ��z'_AttachFieldHelpers.<locals>.AddDecoderT) rR rP �LABEL_REPEATEDr
ZIsTypePackablerO rN r^ r* r+ r` �HasFieldrW r Z
MapEncoderZMapSizerr\ rT ZMessageSetItemEncoderrb ZMessageSetItemSizerr ZTYPE_TO_ENCODERZ
TYPE_TO_SIZER�_encoder�_sizer� _DefaultValueConstructorForFieldre ZFIELD_TYPE_TO_WIRE_TYPEZWIRETYPE_LENGTH_DELIMITED)r rj Z is_packablerf Zhas_packed_falseZ
field_encoderZsizerrl r' ri r( r. & sV
� � � �
�
�
��( �r. c C s0 | j }|�� D ] \}}||vs"J �|||< qd S rL )�extensions_by_name�items)r r" �
extensions�extension_name�extension_fieldr' r' r( r z s r c C s@ | j D ]4}t||jt�|�� |jD ]}t||j|j� q$qdS )a Sets class-level attributes for all enum fields defined in this message.
Also exporting a class-level object that can name enum values.
Args:
descriptor: Descriptor object for this message type.
cls: Class we're constructing for this message type.
N)Z
enum_types�setattrr r ZEnumTypeWrapper�valuesrb )r r � enum_typeZ
enum_valuer' r' r( r/ � s
r/ c sv � j tjkrtd� j ��� jj}t�|d ��|d �t � �rT� ��fdd�}|S t����� ��fdd�}|S d S )Nz&map_entry set on non-repeated field %s�keyrX c s t �| j�j�� j�S rL )r Z
MessageMaprG rU r )r7 � key_checker� value_fieldr' r( �MakeMessageMapDefault� s
�z:_GetInitializeDefaultForMap.<locals>.MakeMessageMapDefaultc s t �| j��� j�S rL )r Z ScalarMaprG rU r )r7 r{ �
value_checkerr' r( �MakePrimitiveMapDefault� s �z<_GetInitializeDefaultForMap.<locals>.MakePrimitiveMapDefault)
rR rP rm �
ValueErrorr rU rY r �GetTypeCheckerr\ )r7 rY r} r r' )r7 r{ r~ r| r( rd � s �
rd c s� t � �rt� �S � jtjkrv� jr:� jg kr:td� j ��� jtj kr\� j
�� fdd�}|S t � � ���fdd�}|S � jtj kr�� j
�� �fdd�}|S � fdd �}|S )
a� Returns a function which returns a default value for a field.
Args:
field: FieldDescriptor object for this field.
The returned function has one argument:
message: Message instance containing this field, or a weakref proxy
of same.
That function in turn returns a default value for this field. The default
value may refer back to |message| via a weak reference.
z/Repeated field default value not empty list: %sc s t �| j� j�S rL )r Z RepeatedCompositeFieldContainerrG rU r rS r' r( �MakeRepeatedMessageDefault� s �zD_DefaultValueConstructorForField.<locals>.MakeRepeatedMessageDefaultc s t �| j� �S rL )r Z RepeatedScalarFieldContainerrG r )� type_checkerr' r( �MakeRepeatedScalarDefault� s �zC_DefaultValueConstructorForField.<locals>.MakeRepeatedScalarDefaultc sL t �dd �s J d� j�jf ����� }|�� jd ur@t| � �n| j� |S )Nr zAUninitialized concrete class found for field %r (message type %r))r r r � _SetListenerrc �_OneofListenerrG )r �result)r7 rU r' r( �MakeSubMessageDefault� s
��� �z?_DefaultValueConstructorForField.<locals>.MakeSubMessageDefaultc s � j S rL )�
default_valuer rS r' r( �MakeScalarDefault� s z;_DefaultValueConstructorForField.<locals>.MakeScalarDefault)
rW rd rR rP rm Zhas_default_valuer� r� rZ r[ rU r r� )r7 r� r� r� r� r' )r7 rU r� r( rq � s( �
rq c C sZ t �� d }t|j�dkr<t|�tu r<tdt|�| |f �}t�t|�|t �� d � dS )zCRe-raise the currently-handled TypeError with the field name added.� z%s for field %s.%s� N) �sys�exc_info�len�argsrO � TypeErrorr �sixZreraise)Z message_name�
field_name�excr' r' r( � _ReraiseTypeErrorWithFieldName� s r� c s, dd� � � �fdd�}d|_ d|_||_dS )z Adds an __init__ method to cls.c S sD t |tj�r@z| j| jW S ty> td| j|f ��Y n0 |S )a Convert a string or integer enum value to an integer.
If the value is a string, it is converted to the enum value in
enum_type with the same name. If the value is not a string, it's
returned as-is. (No conversion or bounds-checking is done.)
z Enum type %s: unknown label "%s")r r� Z string_typesZvalues_by_namerb �KeyErrorr� r )ry rX r' r' r( �_GetIntegerEnumValue� s �z,_AddInitMethod.<locals>._GetIntegerEnumValuec s d| _ t|�dk| _i | _i | _d| _d | _d| _t� � | _
t | �| _ |�
� D �]�\}}t�|�� � d u r|td�j|f ��|d u r�qN� jtjk�rV� �| �}� jtjk�r t� �r�t� �r�|D ]}|| �|| � q�n
|�|� n6|D ]0}t|t ��r |j f i |�� q�|� � �|� q�n,� jtj k�r@�� fdd�|D �}|� |� || j� < qN� jtjk�r̈ �| �}|}t|t ��r�� j j!f i |��}z|�|� W n t�y� t"�j|� Y n0 || j� < qN� jtj k�r�� j#|�}zt$| ||� W qN t�y t"�j|� Y qN0 qNd S )Nr r' Fz,%s() got an unexpected keyword argument "%s"c s g | ]}� �j |��qS r' )ry )�.0�val)r� r7 r' r( �
<listcomp>' s �z0_AddInitMethod.<locals>.init.<locals>.<listcomp>)%r@ r� rA rB rI rC rD rE �message_listener_mod�NullMessageListenerrF � _ListenerrG rs �_GetFieldByNamer� r rR rP rm re rZ r[ rW r\ � MergeFrom�updater �dict�addZ CPPTYPE_ENUM�extendrU r r� ry rw )�self�kwargsr� � field_value�copyrz r� �new_val�r� rK rS r( �init� sf
�
�
z _AddInitMethod.<locals>.initN)r9 r; r6 )rK r r� r' r� r( r0 � s
@r0 c C s6 z | j | W S ty0 td| j|f ��Y n0 dS )z�Returns a field descriptor by field name.
Args:
message_descriptor: A Descriptor describing all fields in message.
field_name: The name of the field to retrieve.
Returns:
The field descriptor associated with the field name.
�&Protocol message %s has no "%s" field.N)rY r� r� r )rK r� r' r' r( r� B s �r� c C s. | j D ]}t||� q| jr*tdd� �|_dS )�=Adds properties for all fields in this protocol message type.c S s t | �S rL )�_ExtensionDict�r� r' r' r( �<lambda>Z � z)_AddPropertiesForFields.<locals>.<lambda>N)r- �_AddPropertiesForField�
is_extendable�propertyZ
Extensions)r r r7 r' r' r( r1 R s
r1 c C sh t jdksJ �| j�� d }t||| j� | jt jkrBt| |� n"| j t j
krZt | |� n
t | |� dS )a# Adds a public property for a protocol message field.
Clients can use this property to get and (in the case
of non-repeated scalar fields) directly set the value
of a protocol message field.
Args:
field: A FieldDescriptor for this field.
cls: The class we're constructing.
�
�
_FIELD_NUMBERN)
rP Z MAX_CPPTYPEr �upperrw rb rR rm � _AddPropertiesForRepeatedFieldrZ r[ �*_AddPropertiesForNonRepeatedCompositeField�'_AddPropertiesForNonRepeatedScalarField)r7 r �
constant_namer' r' r( r� ] s r� c @ s e Zd ZdZdd� ZdS )�_FieldProperty)r c C s t j| |||d� || _d S )N��doc)r� r6 r )r� r �getter�setterr� r' r' r( r6 y s z_FieldProperty.__init__N)r8 r9 r: rJ r6 r' r' r' r( r� v s r� c sZ � j �t��}� fdd�}d|_d� |_�fdd�}d� }t||t� |||d�� dS ) a Adds a public property for a "repeated" protocol message field. Clients
can use this property to get the value of the field, which will be either a
RepeatedScalarFieldContainer or RepeatedCompositeFieldContainer (see
below).
Note that when clients add values to these containers, we perform
type-checking in the case of repeated scalar fields, and we also set any
necessary "has" bits as a side-effect.
Args:
field: A FieldDescriptor for this field.
cls: The class we're constructing.
c s0 | j �� �}|d u r,� �| �}| j �� |�}|S rL �rB �getre �
setdefault�r� r� rS r' r( r� � s
z._AddPropertiesForRepeatedField.<locals>.getterN�Getter for %s.c s t d� ��d S )NzIAssignment not allowed to repeated field "%s" in protocol message object.��AttributeError�r� � new_valuer= r' r( r� � s �z._AddPropertiesForRepeatedField.<locals>.setter�/Magic attribute generated for "%s" proto field.r� �r r? r9 r; rw r� �r7 r �
property_namer� r� r� r' �r7 r> r( r� ~ s
r� c s� �j }t|�}t�����j��jjdk}��fdd�}d|_d| |_|oR�j � � ��fdd���j rz��fdd �}n�}d|_d
| |_d | }t
||t �|||d �� dS )
a� Adds a public property for a nonrepeated, scalar protocol message field.
Clients can use this property to get and directly set the value of the field.
Note that when the client sets the value of a field by using this property,
all necessary "has" bits are set as a side-effect, and we also perform
type-checking.
Args:
field: A FieldDescriptor for this field.
cls: The class we're constructing.
r] c s | j ��� �S rL )rB r� r� )r� r7 r' r( r� � s z7_AddPropertiesForNonRepeatedScalarField.<locals>.getterNr� c
s| z�� |�}W n8 tyF } z td�j||f ��W Y d }~n
d }~0 0 � r`|s`| j��d � n
|| j�< | jsx| �� d S )Nz Cannot set %s to %.1024r: %s)Z
CheckValuer� r rB �poprA � _Modified)r� r� �e)�clear_when_set_to_defaultr7 r� r' r( � field_setter� s �
z=_AddPropertiesForNonRepeatedScalarField.<locals>.field_setterc s �| |� | � � � d S rL )�_UpdateOneofStater� )r7 r� r' r( r� � s
z7_AddPropertiesForNonRepeatedScalarField.<locals>.setterzSetter for %s.r� r� ) r r? r r� r� rN r^ r9 r; rc rw r� )r7 r r> r� rk r� r� r� r' )r� r� r7 r� r� r( r� � s"
r� c sZ � j �t��}� fdd�}d|_d� |_�fdd�}d� }t||t� |||d�� dS ) aN Adds a public property for a nonrepeated, composite protocol message field.
A composite field is a "group" or "message" field.
Clients can use this property to get the value of the field, but cannot
assign to the property directly.
Args:
field: A FieldDescriptor for this field.
cls: The class we're constructing.
c s0 | j �� �}|d u r,� �| �}| j �� |�}|S rL r� r� rS r' r( r� � s
z:_AddPropertiesForNonRepeatedCompositeField.<locals>.getterNr� c s t d� ��d S )NzJAssignment not allowed to composite field "%s" in protocol message object.r� r� r= r' r( r� s �z:_AddPropertiesForNonRepeatedCompositeField.<locals>.setterr� r� r� r� r' r� r( r� � s
r� c C s` | j }|�� D ]"\}}|�� d }t|||j� q| jdur\| jj}|j| |_|j| |_dS )r� r� N) rr rs r� rw rb �file�poolZ_extensions_by_numberZ_extensions_by_name)r r rt ru rv r� r� r' r' r( r2 s
r2 c s0 � fdd�}t |�� _� fdd�}t |�� _d S )Nc s& � j | _� j jj�| � t� | � d S rL )r rN r� r� Z_AddExtensionDescriptorr. )�extension_handle�r r' r( �RegisterExtension" s z,_AddStaticMethods.<locals>.RegisterExtensionc s � � }|� | � |S rL )�MergeFromString)�sr r� r' r( �
FromString* s
z%_AddStaticMethods.<locals>.FromString)� staticmethodr� r� )r r� r� r' r� r( r3 s
r3 c C s> | d j tjkr t| d �S | d jtjkr6| d jS dS dS )z�Given a (FieldDescriptor, value) tuple from _fields, return true if the
value should be included in the list returned by ListFields().r r� TN)rR rP rm �boolrZ r[ rE ��itemr' r' r( �
_IsPresent1 s
r� c C s dd� }||_ dS )� Helper for _AddMessageMethods().c S s( dd� | j �� D �}|jdd� d� |S )Nc S s g | ]}t |�r|�qS r' )r� )r� r� r' r' r( r� A r� z<_AddListFieldsMethod.<locals>.ListFields.<locals>.<listcomp>c S s
| d j S )Nr )rb r� r' r' r( r� B r� z:_AddListFieldsMethod.<locals>.ListFields.<locals>.<lambda>)rz )rB rs �sort)r� Z
all_fieldsr' r' r( �
ListFields@ s z(_AddListFieldsMethod.<locals>.ListFieldsN)r� )rK r r� r' r' r( �_AddListFieldsMethod= s r� zTProtocol message %s has no non-repeated submessage field "%s" nor marked as optionalz2Protocol message %s has no non-repeated field "%s"c s� �j dk}|rtnt�i ��jD ]4}|jtjkr2q |rJ|jtjkrJ|j sJq |�|j
< q �j D ]}|�|j
< q\� ���fdd�� � |_ dS )r� r] c s� z �| }W n$ t y0 t��j|f ��Y n0 t|tj�rjz� | | j| j�W S t yf Y dS 0 n0|jt j
kr�| j � |�}|d uo�|j
S || j v S d S )NF)r� r� r r �descriptor_modZOneofDescriptorrI r rZ rP r[ rB r� rE )r� r� r7 rX �rn Z error_msgZhassable_fieldsrK r' r( rn ` s
z$_AddHasFieldMethod.<locals>.HasFieldN)
r^ �_PROTO3_ERROR_TEMPLATE�_PROTO2_ERROR_TEMPLATEr- rR rP rm rZ r[ rc r Zoneofsrn )rK r rk r7 Zoneofr' r� r( �_AddHasFieldMethodL s
�
r� c s � fdd�}||_ dS )r� c s� z� j | }W nd tyr z,� j| }|| jv r>| j| }nW Y d S W n$ tyl td� j|f ��Y n0 Y n0 || jv r�t| j| d�r�| j| �� | j|= | j� |j
d �|u r�| j|j
= | � � d S )Nr� �InvalidateIterators) rY r� �oneofs_by_namerI r� r rB �hasattrr� r� rc r� )r� r� r7 �rK r' r( �
ClearFieldw s&
�
z(_AddClearFieldMethod.<locals>.ClearFieldN)r� )rK r r� r' r� r( �_AddClearFieldMethodu s !r� c C s dd� }|| _ dS )r� c S s* t �| |� || jv r | j|= | �� d S rL )r �_VerifyExtensionHandlerB r� )r� r� r' r' r( �ClearExtension� s
z0_AddClearExtensionMethod.<locals>.ClearExtensionN)r� )r r� r' r' r( �_AddClearExtensionMethod� s r� c C s dd� }|| _ dS )r� c S sZ t �| |� |jtjkr&td|j ��|jtjkrL| j �
|�}|d uoJ|j S || j v S d S )Nz"%s" is repeated.) r r� rR rP rm r� r rZ r[ rB r� rE )r� r� rX r' r' r( � HasExtension� s z,_AddHasExtensionMethod.<locals>.HasExtensionN)r� )r r� r' r' r( �_AddHasExtensionMethod� s
r� c C sh ddl m} |�� }| j}|s"dS |�d�d }|j�|�}|du rHdS |�|�}|� }|�| j � |S )at Unpacks Any message and returns the unpacked message.
This internal method is different from public Any Unpack method which takes
the target message as argument. _InternalUnpackAny method does not have
target message type and need to find the message type in descriptor pool.
Args:
msg: An Any message to be unpacked.
Returns:
The unpacked message.
r )�symbol_databaseN�/���)
�google.protobufr� ZDefault�type_url�splitr� ZFindMessageTypeByNameZ GetPrototypeZParseFromStringrX )�msgr� �factoryr� � type_namer Z
message_classr r' r' r( �_InternalUnpackAny� s
r� c C s dd� }||_ dS )r� c S s� t |tj�r|j| jkr dS | |u r(dS | jjtkrTt| �}t|�}|rT|rT||kS | �� |�� kshdS t| j �}|�
� t|j �}|�
� ||kS ra ) r � message_modZMessager r �_AnyFullTypeNamer� r� �listrC r� )r� �otherZany_aZany_bZunknown_fieldsZother_unknown_fieldsr' r' r( �__eq__� s$
�
z _AddEqualsMethod.<locals>.__eq__N)r )rK r r r' r' r( �_AddEqualsMethod� s r c C s dd� }||_ dS )r� c S s
t �| �S rL �r �MessageToStringr� r' r' r( �__str__� s z _AddStrMethod.<locals>.__str__N)r )rK r r r' r' r( �
_AddStrMethod� s r c C s dd� }||_ dS )r� c S s
t �| �S rL r r� r' r' r( �__repr__ s z _AddReprMethod.<locals>.__repr__N)r )rK r r r' r' r( �_AddReprMethod s r
c C s dd� }||_ dS )r� c S s t j| dd��d�S )NT)Zas_utf8zutf-8)r r �decoder� r' r' r( � __unicode__ s z&_AddUnicodeMethod.<locals>.__unicode__N)r )Zunused_message_descriptorr r r' r' r( �_AddUnicodeMethod s r
c C s< zt j| }||| �W S ty6 t�d| ��Y n0 dS )a Returns the number of bytes needed to serialize a non-repeated element.
The returned byte count includes space for tag information and any
other additional space associated with serializing value.
Args:
value: Value we're serializing.
field_number: Field number of this value. (Since the field number
is stored as part of a varint-encoded tag, this has an impact
on the total bytes required to serialize the value).
field_type: The type of the field. One of the TYPE_* constants
within FieldDescriptor.
zUnrecognized field type: %dN)r ZTYPE_TO_BYTE_SIZE_FNr� r� � EncodeError)rX � field_numberZ
field_type�fnr' r' r( �_BytesForNonRepeatedElement s
r c C s dd� }||_ dS )r� c S s� | j s | jS d}| j}|�� jrJ|jd �| j�}||jd �| j�7 }nD| � � D ]\}}||�|�7 }qR| j
D ] \}}|t |�t |� 7 }qp|| _d| _ d| j _
|S )Nr rz rX F)rA r@ r r+ rV rY rp rz rX r� rC r� rG �dirty)r� �sizer rj r� rg � value_bytesr' r' r( �ByteSize( s
z$_AddByteSizeMethod.<locals>.ByteSizeN)r )rK r r r' r' r( �_AddByteSizeMethod% s r c C s dd� }||_ dS )r� c [ s8 | � � s(t�d| jjd�| �� �f ��| jf i |��S )Nz)Message %s is missing required fields: %s�,)�
IsInitializedr� r r r �join�FindInitializationErrors�SerializePartialToString)r� r� r' r' r( �SerializeToStringC s ��z6_AddSerializeToStringMethod.<locals>.SerializeToStringN)r )rK r r r' r' r( �_AddSerializeToStringMethod@ s r c C s" dd� }||_ ddd�}||_dS )r� c [ s" t � }| j|jfi |�� |�� S rL )r �_InternalSerialize�write�getvalue)r� r� �outr' r' r( r P s zD_AddSerializePartialToStringMethod.<locals>.SerializePartialToStringNc S s� |d u rt �� }nt|�}| j}|�� jrX|jd �|| j|� |jd �|| j |� n@| �
� D ]\}}|�|||� q`| j D ]\}}||� ||� q~d S )Nrz rX ) r Z)IsPythonDefaultSerializationDeterministicr� r r+ rV rY ro rz rX r� rC )r� � write_bytesZ
deterministicr rj r� rg r r' r' r( �InternalSerializeV s" �
�
�z=_AddSerializePartialToStringMethod.<locals>.InternalSerialize)N)r r )rK r r r# r' r' r( �"_AddSerializePartialToStringMethodM s
r$ c s: dd� }||_ tj�tj�|j� � ��fdd�}||_dS )r� c
S s� t |t�rtjrtd��t|�}t|�}z | �|d|�|krFt�d��W nN t tfyh t�d��Y n0 t
j y� } zt�|��W Y d }~n
d }~0 0 |S )Nz�memoryview not supported in Python 2 with the pure Python proto implementation: this is to maintain compatibility with the C++ implementationr zUnexpected end-group tag.zTruncated message.) r �
memoryviewr� ZPY2r� r� �_InternalParser� � DecodeError�
IndexError�struct�error)r� Z
serialized�lengthr� r' r' r( r� o s � z2_AddMergeFromStringMethod.<locals>.MergeFromStringc s8 t |t�sJ �| �� | j}| j}||k�r4�||�\}}� �|d�\}} |du �r| js`g | _|du rxt�� | _| j}t �
|d�\}
} t � |
�\} }
| dkr�t
�d��|}t �|||
�\}}|dkr�|S |�| |
|� �||||�}|dkr�|S | j�||||� �� f� |}q"||||| |�}| r"| �| � q"|S )a6 Create a message from serialized bytes.
Args:
self: Message, instance of the proto message object.
buffer: memoryview of the serialized data.
pos: int, position to start in the serialized data.
end: int, end position of the serialized data.
Returns:
Message object.
)NNNr zField number 0 is illegal.r� )r r% r� rB rD r� rC r �UnknownFieldSetr Z
_DecodeVarintr
Z UnpackTagr� r' Z_DecodeUnknownFieldZ_add�append�tobytesr� )r� �buffer�pos�endZ
field_dictZunknown_field_setrg Znew_posrh Z
field_desc�tag�_r Z wire_typeZold_pos�data�Zdecoders_by_tagZ
local_ReadTagZlocal_SkipFieldr' r( �
InternalParse� sF
�� z0_AddMergeFromStringMethod.<locals>.InternalParseN)r� r ZReadTagZ SkipFieldr) r& )rK r r� r6 r' r5 r( �_AddMergeFromStringMethodm s 7r7 c s: dd� | j D �� d� fdd� }||_� fdd�}||_dS ) z[Adds the IsInitialized and FindInitializationError methods to the
protocol message class.c S s g | ]}|j tjkr|�qS r' )rR rP ZLABEL_REQUIRED)r� r7 r' r' r( r� � s �z+_AddIsInitializedMethod.<locals>.<listcomp>Nc s� � D ]B}|| j vs*|jtjkr| j | js|dur@|�| �� � dS qt| j �� �D ]�\}}|jtjkrV|j tj
kr�|j j r�|j �
� jr�qV|D ]*}|�� s�|dur�|�| �� � dS q�qV|jrV|�� sV|dur�|�| �� � dS qVdS )a Checks if all required fields of a message are set.
Args:
errors: A list which, if provided, will be populated with the field
paths of all missing required fields.
Returns:
True iff the specified message has all required fields set.
NFT)rB rZ rP r[ rE r� r r rs rR rm rU r* r+ rV r )r� �errorsr7 rX �element�Zrequired_fieldsr' r( r � s4
�
�
� z._AddIsInitializedMethod.<locals>.IsInitializedc s( g }�D ] }| � |j�s|�|j� q| �� D ]�\}}|jtjkr.|jrTd|j }n|j}t |�r�t
|�r�|D ]6}|| }d||f � |� � }|� fdd�|D �7 }qnn q.|j tj
kr�tt|��D ]6}|| }d||f � |� � }|� fdd�|D �7 }q�q.|d � |� � }|� fdd�|D �7 }q.|S ) z�Finds required fields which are not initialized.
Returns:
A list of strings. Each string is a path to an uninitialized field from
the top-level message, e.g. "foo.bar[5].baz".
z(%s)z%s[%s].c s g | ] }� | �qS r' r' �r� r* ��prefixr' r( r� r� zM_AddIsInitializedMethod.<locals>.FindInitializationErrors.<locals>.<listcomp>z%s[%d].c s g | ] }� | �qS r' r' r; r< r' r( r� r� �.c s g | ] }� | �qS r' r' r; r< r' r( r� r� )rn r r- r� rZ rP r[ rM r rW r\ r rR rm r r� ) r� r8 r7 rX r rz r9 Z
sub_errors�ir: r< r( r � s6 z9_AddIsInitializedMethod.<locals>.FindInitializationErrors)N)r- r r )rK r r r r' r: r( �_AddIsInitializedMethod� s
' ,r@ c s& t j�t j� � ��fdd�}|�_d S )Nc s, t |��s td�j|jjf ��|| us,J �| �� | j}|j�� D ]�\}}|j�kr�|�|�}|d u rz|� | �}|||< |�
|� qD|j � kr�|j r�|�|�}|d u r�|� | �}|||< |�
|� qD|| j|< |j
rD| �|� qD|j�r(| js�g | _| j�|j� | jd u �rt�� | _| j�|j� d S )NzLParameter to MergeFrom() must be instance of same class: expected %s got %s.)r r� r8 r& r� rB rs rR r� re r� rZ rE rc r� rC r� rD r r, Z_extend)r� r� r- r7 rX r� �r[ rm r r' r( r� & sB
��
z&_AddMergeFromMethod.<locals>.MergeFrom)rP rm r[ r� )r r� r' rA r( �_AddMergeFromMethod" s )rB c s � fdd�}||_ d S )Nc s^ z� j | }W n ty, td| ��Y n0 | j�|d�}|durV| �|j�rV|jS dS dS )zDReturns the name of the currently set field inside a oneof, or None.z)Protocol message has no oneof "%s" field.N)r� r� r� rI r� rn r )r� Z
oneof_namer7 Z nested_fieldr� r' r( �
WhichOneofS s �
z(_AddWhichOneofMethod.<locals>.WhichOneof)rC )rK r rC r' r� r( �_AddWhichOneofMethodR s rD c C s8 i | _ d| _| jd ur&| j�� d | _i | _| �� d S )Nr' )rB rC rD Z_clearrI r� r� r' r' r( �_Cleard s
rE c C s | j d u rt�� | _ | j S rL )rD r r, r� r' r' r( �_UnknownFieldsq s
rF c C s~ g | _ d | _| �� D ]d\}}|jtjkrt|�rPt|�rx|D ]}|| �� q<q|j tj
krp|D ] }|�� q`q|�� qd S rL ) rC rD r� rZ rP r[ rW r\ �DiscardUnknownFieldsrR rm )r� r7 rX rz Z sub_messager' r' r( �_DiscardUnknownFieldsx s rH c C s |d u rt �� | _n|| _d S rL )r� r� rF )r� Zlistenerr' r' r( r� � s r� c C s� t | |� t| |� t| |� | jr4t|� t|� t| |� t| |� t| |� t | |� t
| |� t | |� t | |� t
| |� t| |� t|� t| |� t|_t|_t|_t|_dS )z3Adds implementations of all Message methods to cls.N)r� r� r� r� r� r� r r r
r
r r r$ r7 r@ rB rD rE ZClearrF Z
UnknownFieldsrH rG r� )rK r r' r' r( r4 � s*
r4 c C s& dd� }dd� }||_ ||_||_dS )z5Adds implementation of private helper methods to cls.c S s( | j s$d| _ d| j_d| _| j�� dS )zwSets the _cached_byte_size_dirty bit to true,
and propagates this to our listener iff this was a state change.
TN)rA rG r rE rF �Modifiedr� r' r' r( rI � s
z*_AddPrivateHelperMethods.<locals>.Modifiedc S s0 | j �|j|�}||ur,| j|= || j |j< dS )z�Sets field as the active field in its containing oneof.
Will also delete currently active field in the oneof, if it is different
from the argument. Does not mark the message as modified.
N)rI r� rc rB )r� r7 Z other_fieldr' r' r( r� � s z3_AddPrivateHelperMethods.<locals>._UpdateOneofStateN)r� Z SetInParentr� )rK r rI r� r' r' r( r5 � s
r5 c @ s e Zd ZdZdd� Zdd� ZdS )r� a0 MessageListener implementation that a parent message registers with its
child message.
In order to support semantics like:
foo.bar.baz.qux = 23
assert foo.HasField('bar')
...child objects must have back references to their parents.
This helper class is at the heart of this support.
c C s* t |tj�r|| _n t�|�| _d| _dS )z�Args:
parent_message: The message whose _Modified() method we should call when
we receive Modified() messages.
FN)r �weakref� ProxyType�_parent_message_weakref�proxyr )r� �parent_messager' r' r( r6 � s z_Listener.__init__c C s0 | j r
d S z| j�� W n ty* Y n0 d S rL )r rL r� �ReferenceErrorr� r' r' r( rI � s z_Listener.ModifiedN)r8 r9 r: r; r6 rI r' r' r' r( r� � s r� c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )r� zCSpecial listener implementation for setting composite oneof fields.c s t t| ��|� || _dS )z�Args:
parent_message: The message whose _Modified() method we should call when
we receive Modified() messages.
field: The descriptor of the field being set in the parent message.
N)r r� r6 �_field)r� rN r7 r% r' r( r6 � s z_OneofListener.__init__c s8 z | j �| j� tt| ��� W n ty2 Y n0 dS )zEAlso updates the state of the containing oneof in the parent message.N)rL r� rP r r� rI rO r� r% r' r( rI s
z_OneofListener.Modified)r8 r9 r: r; r6 rI r< r' r' r% r( r� � s r� )Wr; �
__author__�ior r) r� rJ r� Z six.movesr Zgoogle.protobuf.internalr r r r r r r
r� r r r
r� r r� r r� r ZFieldDescriptorrP r r� rO r r? r rT rW r\ r_ r. r r/ rd rq r� r0 r� r1 r� r� r� r� r� r� r2 r3 r� r� r� r� r� r� r� r� r� r r r
r
r r r r$ r7 r@ rB rD rE rF rH r� r4 r5 �objectr� r� r' r' r' r( �<module>$ s� T: W ,;, �)& (
V_0
".