3
SLj @ s d Z dZddlmZ ddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddlm Z dd lm
Z
dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ejZdZG dd de Z dd 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 Z-d4d5 Z.d6d7 Z/d8d9 Z0d:d; Z1dd? Z3d@dA Z4dBZ5dCZ6dDdE Z7dFdG Z8dHdI Z9dJdK Z:dLdM Z;dNdO ZdTdU Z?dVdW Z@dXdY ZAdZd[ ZBd\d] ZCd^d_ ZDd`da ZEdbdc ZFddde ZGdfdg ZHdhdi ZIdjdk ZJdldm ZKdndo ZLdpdq ZMG drds dseNZOG dtdu dueOZPG dvdw dweNZQdS )xa
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)api_implementation)
containers)decoder)encoder)enum_type_wrapper)message_listener)
type_checkers)well_known_types) wire_format)
descriptor)message) text_formatzgoogle.protobuf.Anyc s0 e Zd ZdZdZ fddZ fddZ 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 sZ |t j }|jtjkr(|tj|j f7 }t|| t|| tt | }|j| |||}|S )a 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.
) r _DESCRIPTOR_KEY full_namer
ZWKTBASES&_AddClassAttributesForNestedExtensions _AddSlotssuper__new__)clsnamebases
dictionaryr
superclass new_class) __class__ $/usr/lib/python3.6/python_message.pyr i s
z$GeneratedProtocolMessageType.__new__c s |t j }i | _|jr6|j jr6tj|df| jtj< x|j D ]}t
| | q>W | |_ t || t
|| t|| t|| t| t|| t|| tt | }|j||| 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.
N)r r _decoders_by_tag has_options
GetOptionsmessage_set_wire_formatr ZMessageSetItemDecoderZMESSAGE_SET_ITEM_TAGfields_AttachFieldHelpers_concrete_class_AddEnumValues_AddInitMethod_AddPropertiesForFields_AddPropertiesForExtensions_AddStaticMethods_AddMessageMethods_AddPrivateHelperMethodsr __init__)r r r r r fieldr )r r r r. s"
z%GeneratedProtocolMessageType.__init__)__name__
__module__ __qualname____doc__r r r.
__classcell__r r )r r r N s r 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 r6 c C sj t |tstd| |js*td|j |js>td|j |j| jk rftd|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)
isinstance_FieldDescriptorKeyError is_extensionr containing_typer )r
extension_handler r r _VerifyExtensionHandle s
r= c C s ddddddddd g |d
< d S ) 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_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| jj jo0| jtjko0| jtj kS )N)
r: r; r! r" r# typer8 TYPE_MESSAGElabelZLABEL_OPTIONAL)r/ r r r _IsMessageSetExtension s
rL c C s | j tjko | jjo | jj jS )N)rI r8 rJ message_typer! r" map_entry)r/ r r r _IsMapField s rO c C s | j jd }|jtjkS )Nvalue)rM fields_by_namecpp_typer8 CPPTYPE_MESSAGE)r/ Z
value_typer r r _IsMessageMapField
s rT c s@ j tjkotjj}|s&d}nFjjdkrDjo@j j
}n(jodj j dodj j
dk}| }t rt
j}t
jt}nNtrt
jj}t
jj}n,tjj j|}tjj j|}|_|_t_ fdd}|tj j d r.AddDecoderT) rK r8 LABEL_REPEATEDr ZIsTypePackablerI r; syntaxr! r" rU HasFieldrO r Z
MapEncoderZMapSizerrT rL ZMessageSetItemEncoderrV ZMessageSetItemSizerr ZTYPE_TO_ENCODERZ
TYPE_TO_SIZER_encoder_sizer _DefaultValueConstructorForFieldrY ZFIELD_TYPE_TO_WIRE_TYPEZWIRETYPE_LENGTH_DELIMITED)r r] Z is_packablerZ Zhas_packed_falseZ
field_encoderZsizerr` r )r r] r^ r_ r r% sB
r% c C s( | j }x |j D ]\}}|||< qW d S )N)extensions_by_nameitems)r r extension_dictextension_nameextension_fieldr r r r U s r c C sH xB| j D ]8}t||jtj| x |jD ]}t||j|j q(W qW dS )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_typessetattrr r ZEnumTypeWrappervaluesrV )r r enum_typeZ
enum_valuer r r r' \ s r' c sv j tjkrtd j jj}tj|d |d t rT fdd}|S tj fdd}|S d S )Nz&map_entry set on non-repeated field %skeyrP c s t j| jj jS )N)r Z
MessageMaprD rM )r
)r/ key_checker value_fieldr r MakeMessageMapDefaultt s
z:_GetInitializeDefaultForMap..MakeMessageMapDefaultc s t j| j jS )N)r Z ScalarMaprD rM )r
)r/ rp
value_checkerr r MakePrimitiveMapDefault{ s z<_GetInitializeDefaultForMap..MakePrimitiveMapDefault)
rK r8 ra
ValueErrorr rM rQ r GetTypeCheckerrT )r/ rQ rr rt r )r/ rp rs rq r rX k s
rX c s t rt S jtjkrv jr: jg kr:td j jtj kr\ j
fdd}|S t j 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 jS )N)r Z RepeatedCompositeFieldContainerrD rM )r
)r/ r r MakeRepeatedMessageDefault s zD_DefaultValueConstructorForField..MakeRepeatedMessageDefaultc s t j| j S )N)r Z RepeatedScalarFieldContainerrD )r
) type_checkerr r MakeRepeatedScalarDefault s zC_DefaultValueConstructorForField..MakeRepeatedScalarDefaultc s, j }|j jd k r t| n| j |S )N)r& _SetListenerrW _OneofListenerrD )r
result)r/ rM r r MakeSubMessageDefault s
z?_DefaultValueConstructorForField..MakeSubMessageDefaultc s j S )N)
default_value)r
)r/ r r MakeScalarDefault s z;_DefaultValueConstructorForField..MakeScalarDefault)
rO rX rK r8 ra Zhas_default_valuer~ ru rR rS rM r rv )r/ rw ry r} r r )r/ rM rx r rf s&
rf c C sZ t j d }t|jdkr td| j|f Y nX |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")r7 r Z string_typesZvalues_by_namerV r9 ru r )rn rP r r r _GetIntegerEnumValue s z,_AddInitMethod.._GetIntegerEnumValuec s" d| _ t|dk| _i | _i | _f | _d| _tj | _ t
| | _ x|j D ]\}}t
| d krztdj|f |d krqL jtjkrX j| } jtjkr t rt rx*|D ]}|| j|| qW n
|j| n6xb|D ],}t|tr |j f | q|j j| qW n, jtj krB fdd|D }|j | || j < qL jtjkr̈ j| }|}t|tr j j f |}y|j| W n" tk
r t!j| Y nX || j < qL jtj kr j"|}yt#| || W qL tk
r t!j| Y qLX qLW d S )Nr Fz,%s() got an unexpected keyword argument '%s'c s g | ]} j |qS r )rn ).0val)r r/ r r
s z0_AddInitMethod..init..)$r> r r? r@ rF rA rB message_listener_modNullMessageListenerrC _ListenerrD rh _GetFieldByNamer r rK r8 ra rY rR rS rO rT MergeFromupdater7 dictaddZ CPPTYPE_ENUMextendrM r& r rn rl )selfkwargsr field_valuecopyro r Znew_val)r rH )r/ r init s`
z _AddInitMethod..initN)r1 r3 r. )rH r r r )r rH r r( s
=r( c C s6 y
| j | S tk
r0 td| j|f Y nX dS )zReturns 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.
z&Protocol message %s has no "%s" field.N)rQ r9 ru r )rH r r r r r s
r c C s2 x| j D ]}t|| qW | jr.tdd |_dS )z=Adds properties for all fields in this protocol message type.c S s t | S )N)_ExtensionDict)r r r r / s z)_AddPropertiesForFields..N)r$ _AddPropertiesForField
is_extendablepropertyZ
Extensions)r r r/ r r r r) ' s r) c C sZ | j j d }t||| j | jtjkr4t| | n"| jtj krLt
| | 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) r upperrl rV rK r8 ra _AddPropertiesForRepeatedFieldrR rS *_AddPropertiesForNonRepeatedCompositeField'_AddPropertiesForNonRepeatedScalarField)r/ r
constant_namer r r r 2 s r c sX 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 j }|d kr, j| }| j j |}|S )N)r@ getrY
setdefault)r r )r/ r r getter\ s
z._AddPropertiesForRepeatedField..getterNzGetter for %s.c s t d d S )NzIAssignment not allowed to repeated field "%s" in protocol message object.)AttributeError)r new_value)r5 r r settero s z._AddPropertiesForRepeatedField..setterz/Magic attribute generated for "%s" proto field.)doc)r r6 r1 r3 rl r )r/ r
property_namer r r r )r/ r5 r r K s
r c s j }t|}tjjt }jjdk}fdd}d|_d| |_ |oXj
fddj
rfdd }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.
proto3c s | j j S )N)r@ r )r )r~ r/ r r r s z7_AddPropertiesForNonRepeatedScalarField..getterNzGetter for %s.c s@ j |} r$| r$| jjd n
|| j< | js<| j d S )N)
CheckValuer@ popr? _Modified)r r )clear_when_set_to_defaultr/ rx r r field_setter s
z=_AddPropertiesForNonRepeatedScalarField..field_setterc s | | | j d S )N)_UpdateOneofState)r r )r/ r r r r s
z7_AddPropertiesForNonRepeatedScalarField..setterzSetter for %s.z/Magic attribute generated for "%s" proto field.)r )
r r6 r rv r~ setr; rb r1 r3 rW rl r ) r/ r r5 r Z valid_values is_proto3r r r r )r r~ r/ r rx r r w s$
r c sX 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 j }|d kr, j| }| j j |}|S )N)r@ r rY r )r r )r/ r r r s
z:_AddPropertiesForNonRepeatedCompositeField..getterNzGetter for %s.c s t d d S )NzJAssignment not allowed to composite field "%s" in protocol message object.)r )r r )r5 r r r s z:_AddPropertiesForNonRepeatedCompositeField..setterz/Magic attribute generated for "%s" proto field.)r )r r6 r1 r3 rl r )r/ r r r r r r )r/ r5 r r s
r c C sd | j }x.|j D ]"\}}|j d }t|||j qW | jdk r`| jj}|j| |_|j| |_dS )z=Adds properties for all fields in this protocol message type.r N) rg rh r rl rV filepool_extensions_by_number_extensions_by_name)r r ri rj rk r r r r r r* s
r* c s0 fdd}t | _ fdd}t | _d S )Nc s& j | _ j jjj| t | d S )N)r r; r r ZAddExtensionDescriptorr% )r< )r r r RegisterExtension s z,_AddStaticMethods..RegisterExtensionc s }|j | |S )N)MergeFromString)sr
)r r r
FromString s
z%_AddStaticMethods..FromString) staticmethodr r )r r r r )r r r+ s
r+ c C s> | d j tjkr t| d S | d jtjkr6| d jS dS dS )zGiven a (FieldDescriptor, value) tuple from _fields, return true if the
value should be included in the list returned by ListFields().r r TN)rK r8 ra boolrR rS rB )itemr r r
_IsPresent s
r c C s dd }||_ dS )z Helper for _AddMessageMethods().c S s( dd | j j D }|jdd d |S )Nc S s g | ]}t |r|qS r )r )r r r r r r
s z<_AddListFieldsMethod..ListFields..c S s
| d j S )Nr )rV )r r r r r s z:_AddListFieldsMethod..ListFields..)ro )r@ rh sort)r Z
all_fieldsr r r
ListFields s z(_AddListFieldsMethod..ListFieldsN)r )rH r r r r r _AddListFieldsMethod s r z:Protocol message has no non-repeated submessage field "%s"z/Protocol message has no non-repeated field "%s"c s | j dk}|rtnti x@| jD ]6}|jtjkr4q"|rN|jtjkrN|j rNq"||j
< q"W |szx| j D ]}||j
< qhW fdd |_ dS )z Helper for _AddMessageMethods().r c s y | }W n t k
r, t| Y nX t|tjrdy | | j| jS t k
r` dS X n0|jtj kr| j
j |}|d k o|j S || j
kS d S )NF)
r9 ru r7 descriptor_modZOneofDescriptorrF r rR r8 rS r@ r rB )r r r/ rP )rc error_msghassable_fieldsr r rc ( s z$_AddHasFieldMethod..HasFieldN)
rb _Proto3HasError_Proto2HasErrorr$ rK r8 ra rR rS rW r Zoneofsrc )rH r r r/ Zoneofr )rc r r r _AddHasFieldMethod s
r c s fdd}||_ dS )z Helper for _AddMessageMethods().c s y j | }W nd tk
rr y( j| }|| jkr@| j| }nd S W n& tk
rl td j|f Y nX Y nX || jkrt| j| dr| j| j | j|= | jj |j
d |kr| j|j
= | j d S )Nz(Protocol message %s() has no "%s" field.InvalidateIterators) rQ r9 oneofs_by_namerF ru r r@ hasattrr r rW r )r r r/ )rH r r
ClearField? s$
z(_AddClearFieldMethod..ClearFieldN)r )rH r r r )rH r _AddClearFieldMethod= s !r c C s dd }|| _ dS )z Helper for _AddMessageMethods().c S s( t | | || jkr | j|= | j d S )N)r= r@ r )r r< r r r ClearExtensione s
z0_AddClearExtensionMethod..ClearExtensionN)r )r r r r r _AddClearExtensionMethodc s r c C s dd }|| _ dS )z Helper for _AddMessageMethods().c S sX t | | |jtjkr$td|j |jtjkrJ| jj |}|d k oH|j
S || jkS d S )Nz"%s" is repeated.) r= rK r8 ra r9 r rR rS r@ r rB )r r< rP r r r HasExtensionq s
z,_AddHasExtensionMethod..HasExtensionN)r )r r r r r _AddHasExtensionMethodo s
r c C sh ddl m} |j }| j}|s"dS |jdd }|jj|}|dkrHdS |j|}| }|j| 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/r )
google.protobufr ZDefaulttype_urlsplitr ZFindMessageTypeByNameZ GetPrototypeZParseFromStringrP )msgr factoryr Z type_namer Z
message_classr
r r r _InternalUnpackAny} s
r c C s dd }||_ dS )z Helper for _AddMessageMethods().c S s t |tj s|j| jkr dS | |kr*dS | jjtkrVt| }t|}|rV|rV||kS | j |j ksjdS t| j }|j
t|j }|j
||kS )NFT) r7 message_modZMessager r _AnyFullTypeNamer r listrA r )r otherZany_aZany_bZunknown_fieldsZother_unknown_fieldsr r r __eq__ s"
z _AddEqualsMethod..__eq__N)r )rH r r r r r _AddEqualsMethod s r c C s dd }||_ dS )z Helper for _AddMessageMethods().c S s
t j| S )N)r MessageToString)r r r r __str__ s z _AddStrMethod..__str__N)r )rH r r r r r
_AddStrMethod s r c C s dd }||_ dS )z Helper for _AddMessageMethods().c S s
t j| S )N)r r )r r r r __repr__ s z _AddReprMethod..__repr__N)r )rH r r r r r _AddReprMethod s r c C s dd }||_ dS )z Helper for _AddMessageMethods().c S s t j| ddjdS )NT)Zas_utf8zutf-8)r r decode)r r r r __unicode__ s z&_AddUnicodeMethod..__unicode__N)r )Zunused_message_descriptorr r r r r _AddUnicodeMethod s r c C s< yt j| }||| S tk
r6 tjd| Y nX 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_FNr9 r EncodeError)rP Z field_numberZ
field_typefnr r r _BytesForNonRepeatedElement s
r c C s dd }||_ dS )z Helper for _AddMessageMethods().c S s | j s | jS d}| j}|j jrJ|jd j| j}||jd j| j7 }nLx"| j D ]\}}||j|7 }qTW x&| j
D ] \}}|t |t | 7 }qvW || _d| _ d| j _
|S )Nr ro rP F)r? r> r r" rN rQ re ro rP r rA r rD dirty)r sizer r] r r[ value_bytesr r r ByteSize s
z$_AddByteSizeMethod..ByteSizeN)r )rH r r r r r _AddByteSizeMethod s r c C s dd }||_ dS )z Helper for _AddMessageMethods().c [ s8 g }| j s,tjd| jjdj| j f | jf |S )Nz)Message %s is missing required fields: %s,)
IsInitializedr r r r joinFindInitializationErrorsSerializePartialToString)r r errorsr r r SerializeToString s z6_AddSerializeToStringMethod..SerializeToStringN)r )rH r r r r r _AddSerializeToStringMethod s r c C s" dd }||_ ddd}||_dS )z Helper for _AddMessageMethods().c [ s t }| j|jf| |j S )N)r _InternalSerializewritegetvalue)r r outr r r r s zD_AddSerializePartialToStringMethod..SerializePartialToStringNc S s |d krt j }nt|}| j}|j jrX|jd j|| j| |jd j|| j | nHx"| j
D ]\}}|j||| qbW x"| j D ]\}}|| || qW d S )Nro rP ) r Z)IsPythonDefaultSerializationDeterministicr r r" rN rQ rd ro rP r rA )r Z write_bytesZ
deterministicr r] r r[ r r r r InternalSerialize s
z=_AddSerializePartialToStringMethod..InternalSerialize)N)r r )rH r r r r r r "_AddSerializePartialToStringMethod s
r c sF dd }||_ tjtj|j | jdk fdd}||_dS )z Helper for _AddMessageMethods().c S s| t |}y | j|d||kr&tjdW nN ttfk
rJ tjdY n. tjk
rv } ztj|W Y d d }~X nX |S )Nr zUnexpected end-group tag.zTruncated message.)r _InternalParser Z DecodeError
IndexErrorr structerror)r Z
serializedZlengther r r r 8 s z2_AddMergeFromStringMethod..MergeFromStringr c s | j | j}| j}x||kr||\}} j|d\}} |d kr|}
||||}|dkrb|S sptj r|s~g }| _|j|||
| f |}q||||| |}| r| j| qW |S )Nr )NNr )r r@ rA r r Z&GetPythonProto3PreserveUnknownsDefaultappendr ) r bufferposendZ
field_dictZunknown_field_listr[ Znew_posr\ Z
field_descZvalue_start_pos)decoders_by_tagr
local_ReadTaglocal_SkipFieldr r
InternalParseL s,
z0_AddMergeFromStringMethod..InternalParseN)r r ZReadTagZ SkipFieldr rb r )rH r r r r )r r r r r _AddMergeFromStringMethod6 s
r 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 )rK r8 ZLABEL_REQUIRED)r r/ r r r r k s z+_AddIsInitializedMethod..Nc s xJ D ]B}|| j ks.|jtjkr| j | j r|dk rD|j| j dS qW xt| j j D ]\}}|jtjkr\|j tj
kr|j j r|j j
jrq\xZ|D ]&}|j s|dk r|j| j dS qW q\|jr\|j r\|dk r|j| j dS q\W dS )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)r@ rR r8 rS rB r r r rh rK ra rM r! r" rN r )r r r/ rP element)required_fieldsr r r n s.
z._AddIsInitializedMethod..IsInitializedc s> g }x$D ] }| j |js
|j|j q
W x | j D ]\}}|jtjkr6|jr^d|j }n|j}t |rt
|rx@|D ]6}|| }d||f |j }| fdd|D 7 }qzW n q6|j tj
krxntt|D ]6}|| }d||f |j }| fdd|D 7 }qW q6|d |j }| fdd|D 7 }q6W |S ) zFinds 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 s zM_AddIsInitializedMethod..FindInitializationErrors..z%s[%d].c s g | ] } | qS r r )r r )r r r r s .c s g | ] } | qS r r )r r )r r r r s )rc r r
r rR r8 rS r: r rO rT r rK ra ranger ) r r r/ rP r ro r Z
sub_errorsi)r )r r r s6
z9_AddIsInitializedMethod..FindInitializationErrors)N)r$ r r )rH r r r r )r r _AddIsInitializedMethodg s
' ,r c s& t jt j fdd}|_d S )Nc s t |s tdj|jjf | j | j}x|jj D ]\}}|jkr||j|}|d krp|j | }|||< |j
| q:|j kr|j r|j|}|d kr|j | }|||< |j
| q:|| j|< |j
r:| j| q:W |jr| jsg | _| jj|j d S )NzLParameter to MergeFrom() must be instance of same class: expected %s got %s.)r7 r r0 r r r@ rh rK r rY r rR rB rW r rA r )r r r$ r/ rP r )rS ra r r r r s6
z&_AddMergeFromMethod..MergeFrom)r8 ra rS r )r r r )rS ra r r _AddMergeFromMethod s %r c s fdd}||_ d S )Nc s` y j | }W n tk
r. td| Y nX | jj|d}|dk rX| j|jrX|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 r9 ru rF r rc r )r Z
oneof_namer/ Z nested_field)rH r r
WhichOneof s z(_AddWhichOneofMethod..WhichOneof)r )rH r r r )rH r _AddWhichOneofMethod s r c C s dd }|| _ d S )Nc S s t | f | j fS )N)rI __getstate__)r r r r
__reduce__ s z$_AddReduceMethod..__reduce__)r )r r r r r _AddReduceMethod s r c C s i | _ f | _i | _| j d S )N)r@ rA rF r )r r r r _Clear
s r c C sX g | _ xL| j D ]@\}}|jtjkr|jtjkrHx |D ] }|j q6W q|j qW d S )N)rA r rR r8 rS rK ra DiscardUnknownFields)r r/ rP Z sub_messager r r _DiscardUnknownFields s
r" c C s |d krt j | _n|| _d S )N)r r rC )r Zlistenerr r r rz s rz 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 r r r r r r ZClearr" r! rz )rH r r r r r, $ s*
r, c C s& dd }dd }||_ ||_||_dS )z5Adds implementation of private helper methods to cls.c S s( | j s$d| _ d| j_d| _| jj dS )zwSets the _cached_byte_size_dirty bit to true,
and propagates this to our listener iff this was a state change.
TN)r? rD r rB rC Modified)r r r r r# A s
z*_AddPrivateHelperMethods..Modifiedc S s0 | j j|j|}||k r,| j|= || j |j< dS )zSets 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)rF r rW r@ )r r/ Z other_fieldr r r r P s z3_AddPrivateHelperMethods.._UpdateOneofStateN)r Z SetInParentr )rH r r# r r r r r- > s
r- 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 |tjr|| _n tj|| _d| _dS )zArgs:
parent_message: The message whose _Modified() method we should call when
we receive Modified() messages.
FN)r7 weakref ProxyType_parent_message_weakrefproxyr )r parent_messager r r r. n s z_Listener.__init__c C s2 | j r
d S y| jj W n tk
r, Y nX d S )N)r r& r ReferenceError)r r r r r# s z_Listener.ModifiedN)r0 r1 r2 r3 r. r# r r r r r ` s r c s, e Zd ZdZ fddZ fddZ ZS )r{ zCSpecial listener implementation for setting composite oneof fields.c s t t| j| || _dS )zArgs:
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{ r. _field)r r( r/ )r r r r. s z_OneofListener.__init__c s: y | j j| j tt| j W n tk
r4 Y nX dS )zEAlso updates the state of the containing oneof in the parent message.N)r& r r* r r{ r# r) )r )r r r r# s
z_OneofListener.Modified)r0 r1 r2 r3 r. r# r4 r r )r r r{ s r{ c @ sP e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dS )r zDict-like container for supporting an indexable "Extensions"
field on proto instances.
Note that in all cases we expect extension handles to be
FieldDescriptors.
c C s
|| _ dS )zMextended_message: Message instance for which we are the Extensions dict.
N)_extended_message)r Zextended_messager r r r. s z_ExtensionDict.__init__c C s t | j| | jjj|}|dk r&|S |jtjkr@|j| j}nF|jtj kr|j
j }y|j | jj
W q tk
r| Y qX n|jS | jjj||}|S )z8Returns the current value of the given extension handle.N)r= r+ r@ r rK r8 ra rY rR rS rM r& rz rD r) r~ r )r r< r| r r r __getitem__ s
z_ExtensionDict.__getitem__c C sH t || jsdS | jj }|jj }dd |D }dd |D }||kS )NFc S s g | ]}|j r|qS r )r: )r r/ r r r r s z)_ExtensionDict.__eq__..c S s g | ]}|j r|qS r )r: )r r/ r r r r s )r7 r r+ r )r r Z my_fieldsZ other_fieldsr r r r s
z_ExtensionDict.__eq__c C s
| |k S )Nr )r r r r r __ne__ s z_ExtensionDict.__ne__c C s t dd S )Nzunhashable object)r )r r r r __hash__ s z_ExtensionDict.__hash__c C s\ t | j| |jtjks$|jtjkr2td|j t j
|}|j || jj |< | jj
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+ rK r8 ra rR rS r r r rv r r@ r )r r< rP rx r r r __setitem__ s
z_ExtensionDict.__setitem__c C s | j jj|dS )zTries to find a known extension with the specified name.
Args:
name: Extension full name.
Returns:
Extension field descriptor.
N)r+ r r )r r r r r _FindExtensionByName s z#_ExtensionDict._FindExtensionByNamec C s | j jj|dS )zTries to find a known extension with the field number.
Args:
number: Extension field number.
Returns:
Extension field descriptor.
N)r+ r r )r rV r r r _FindExtensionByNumber s z%_ExtensionDict._FindExtensionByNumberN) r0 r1 r2 r3 r. r, r r- r. r/ r0 r1 r r r r r s !
r )Rr3
__author__ior r r r$ r Zgoogle.protobuf.internalr r r r r r r r r
r r r r r
r r ZFieldDescriptorr8 r rI r r6 r= r rL rO rT r% r r' rX rf r r( r r) r r r r r* r+ r r r r r r r r r r r r r r r r r r r r r r r r" rz r, r- objectr r{ r r r r r 1 s k C7 T ,8,
*& ( 1_, ".