U ڲg;@sddlZddlZddlZddlZddlmZmZmZmZm Z ddl Z ddl Z ddl m Z ddl mZddlmZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZe j j!Z"eZ#Gddde$Z%Gddde%dZ&GdddZ'dZ(dS)N)AnyDictListOptionalType)descriptor_pb2)message) MessageToDict MessageToJsonParse) _file_info) _package_info)Field)MapField) RepeatedField)Marshal) ProtoType)has_upbcs(eZdZdZfddZeddZeddZdd Z d-d d e d d dZ fddZ e dddZe ddddZee d dddZee ee d dddZee ee e dddZd d d d d!d d d"edd#d$Zd d%ddd&d'Zd d d d d d(eeefdd)d*Zd+d,ZZS). MessageMetaz} | j1rd | j/| _#t%|| j&_'|| j#<| j/|| j/<q|rt2d|f|}t3j45|d |6}t3j47||}|D]"}||j&j8kr|j&j8"|qt9j:f|;di}t9j<|dd| Ddd|=D|d }fd!d|j>=D}|D]}|j?,@|j>;|qfd"d|jA=D}|D]}|jB,@|jA;|qt%dkr|j&jC,@|n ||j><tD| ||||d#|d<t||||}||j0_E|j0jFGD] } || _Eq`||jH|<|jI|d$r|jJ||d%|S)&N __qualname__.zz{pascal_key}Entryz_\wcSs|ddS)Nr)groupupper)mr1/tmp/pip-unpacked-wheel-zasgy6sz/proto/message.pyOz%MessageMeta.__new__..r)Z pascal_key __module__z{prefix}.{name})prefixnameZ map_entryT)rr _pb_optionsnumberkey)r$enumrvalue)r$r)r! parent_nameindexpackagepb_metaz_{}Z AttrsMetar"cSsg|] }|jqSr) descriptor.0irrr sz'MessageMeta.__new__..cSsg|]}tj|dqS)r!)rZOneofDescriptorProtor/rrrr2s)r!fieldZ oneof_decloptionscs g|]}|ddkr|qSNrr0p local_pathrrr2scs g|]}|ddkr|qSr6rr8r:rrr2s)fields full_namemarshalr5r+) new_class)r?Z fallback_salt)Ksuper__new__r compiletuplegetsplitr*joinlstripitems isinstancerformatresubreplacer collections OrderedDictrZ map_key_typeZ proto_typer'rrMessagerrZMESSAGEr$updatesetZ mcls_dataappendZoneof setdefaultlenr.Z oneof_indexstrhasattrcallabler,add DESCRIPTORfiler!r-optionaltyper Z _FileInfoZproto_file_namelowerZmaybe_add_descriptor dependencyrMessageOptionspopZDescriptorProtokeysnestedZ nested_type MergeFromZ nested_enumZ enum_typeZ message_type _MessageInfoparentr<valuesmessagesreadyZgenerate_file_pb)mclsr!basesattrsr+r>ixr=Z map_fieldsr%r4Zmsg_nameZ entry_attrsr<Z new_attrsZoneofsZ proto_importsr*Z field_msgZ field_enumZ opt_attrsfilename file_infoZ proto_importoptsdescZ child_pathsZ child_pathcls __class__r:rrA-s                     zMessageMeta.__new__cKstSN)rNrO)rjr!rkkwargsrrr __prepare__szMessageMeta.__prepare__cCs|jSru)r-)rrrrrmeta szMessageMeta.metacCstzRttt}|d|j}|dd|jD|dd|jD|WStk rnttYSXdS)N) rxr,wrap serialize deserializeto_json from_jsonto_dict copy_fromcss|] }|jVqdSrur3)r0trrr 5sz&MessageMeta.__dir__..css|] }|jVqdSrur3r0errrr6s) rRdirr]rQr,rZ nested_types enum_typesAttributeError)selfnamesrqrrr__dir__$s  zMessageMeta.__dir__NFcoercecCs@|dkr|jjSt||s:|r(||}ntd||jf|jS)aHReturn the underlying protobuf Message class or instance. Args: obj: If provided, and an instance of ``cls``, return the underlying protobuf instance. coerce (bool): If provided, will attempt to coerce ``obj`` to ``cls`` if it is not already an instance. Nz%r is not an instance of %s)rxr,rI TypeError__name___pb)rrobjrrrrr,<s   zMessageMeta.pbcs ||}t||d||S)zReturn a Message object that shallowly wraps the descriptor. Args: pb: A protocol buffer object, such as would be returned by :meth:`pb`. r)rAr@ __setattr__)rrr,instancersrrryTs zMessageMeta.wrapreturncCs|j|ddS)aReturn the serialized proto. Args: instance: An instance of this message type, or something compatible (accepted by the type's constructor). Returns: bytes: The serialized representation of the protocol buffer. Tr)r,SerializeToString)rrrrrrrz`s zMessageMeta.serializerP)payloadrcCs|||S)aGiven a serialized proto, deserialize it into a Message instance. Args: payload (bytes): The serialized proto. Returns: ~.Message: An instance of the message class against which this method was called. )ryr, FromString)rrrrrrr{ls zMessageMeta.deserialize)including_default_value_fieldsrcCs$tddkr |dk r tdtdS)z Warn Protobuf 5.x+ users that `including_default_value_fields` is deprecated if it is set. Args: including_default_value_fields (Optional(bool)): The value of `including_default_value_fields` set by the user. r34NzThe argument `including_default_value_fields` has been removed from Protobuf 5.x. Please use `always_print_fields_with_no_presence` instead. )PROTOBUF_VERSIONwarningswarnDeprecationWarning)rrrrrr:_warn_if_including_default_value_fields_is_used_protobuf_5xs zFMessageMeta._warn_if_including_default_value_fields_is_used_protobuf_5)$always_print_fields_with_no_presencerrcCs$|dk r |dk r ||kr tddS)a Raise Exception if both `always_print_fields_with_no_presence` and `including_default_value_fields` are set and the values differ. Args: always_print_fields_with_no_presence (Optional(bool)): The value of `always_print_fields_with_no_presence` set by the user. including_default_value_fields (Optional(bool)): The value of `including_default_value_fields` set by the user. Returns: None Raises: ValueError: if both `always_print_fields_with_no_presence` and `including_default_value_fields` are set and the values differ. Nz`Arguments `always_print_fields_with_no_presence` and `including_default_value_fields` must match) ValueErrorrrrrrrr0_raise_if_print_fields_values_are_set_and_differszrH_get_pb_type_from_keyrrJto_proto) rrrrvparamsr>r%r(pb_typepb_valuersrrrsL       zMessage.__init__cCs^d}z|jj|j}Wn>tk rT|d|jjkrP|d}|jj|j}YnX||fS)aYGiven a key, return the corresponding pb_type. Args: key(str): The name of the field. Returns: A tuple containing a key and pb_type. The pb_type will be the composite type of the field, or the primitive type if a primitive. If no corresponding field exists, return None. N_)r-r<rKeyError)rr%rrrrrs  zMessage._get_pb_type_from_keycCsnt|j}dd|jjD}|dd|jD|dd|jD|t t |d|S)NcSsh|]}|qSrr)r0f_namerrr sz"Message.__dir__..css|] }|jVqdSrur3)r0rrrrrsz"Message.__dir__..css|] }|jVqdSrur3rrrrrs)__bool__ __contains____dict__ __getattr__ __getstate__r __setstate__ __weakref__) r]r,rZr-r<rbrQrrrobject)rrqrrrrrs zMessage.__dir__cstfddjjDS)z4Return True if any field is truthy, False otherwise.c3s |]}|kot|VqdSru)getattr)r0krrrrsz#Message.__bool__..)anyr-r<rbrrrrrszMessage.__bool__cCs<t|j|}z|j|WStk r6t|YSXdS)aRReturn True if this field was set to something non-zero on the wire. In most cases, this method will return True when ``__getattr__`` would return a truthy value and False when it would return a falsy value, so explicitly calling this is not useful. The exception case is empty messages explicitly set on the wire, which are falsy from ``__getattr__``. This method allows to distinguish between an explicitly provided empty message and the absence of that message, which is useful in some edge cases. The most common edge case is the use of ``google.protobuf.BoolValue`` to get a boolean that distinguishes between ``False`` and ``None`` (or the same for a string, int, etc.). This library transparently handles that case for you, but this method remains available to accommodate cases not automatically covered. Args: key (str): The name of the field. Returns: bool: Whether the field's value corresponds to a non-empty wire serialization. N)rrZHasFieldrr)rr%rrrrrs  zMessage.__contains__cCs|j|dS)zmDelete the value on the given field. This is generally equivalent to setting a falsy value. N)r ClearField)rr%rrr __delattr__?szMessage.__delattr__cCs6t|t|r|j|jkSt||jjr2|j|kStS)z7Return True if the messages are equal, False otherwise.)rIr]rr-r,NotImplementedrrrrr__eq__Fs   zMessage.__eq__cCsR||\}}|dkr*td|jj|t|j|}|jj}|j ||||kdS)aRetrieve the given field's value. In protocol buffers, the presence of a field on a message is sufficient for it to always be "present". For primitives, a value of the correct type will always be returned (the "falsy" values in protocol buffers consistently match those in Python). For repeated fields, the falsy value is always an empty sequence. For messages, protocol buffers does distinguish between an empty message and absence, but this distinction is subtle and rarely relevant. Therefore, this method always returns an empty message (following the official implementation). To check for message presence, use ``key in self`` (in other words, ``__contains__``). .. note:: Some well-known protocol buffer types (e.g. ``google.protobuf.Timestamp``) will be converted to their Python equivalents. See the ``marshal`` module for more details. Nr)Zabsent) rrrJrtrrrr-r>Z to_python)rr%rrr>rrrrSs zMessage.__getattr__cCs ||k S)z9Return True if the messages are unequal, False otherwise.rrrrr__ne__tszMessage.__ne__cCs t|jSru)reprrrrrr__repr__xszMessage.__repr__cs|ddkrt||S|jj}||\}}|dkrLtd|jj|| ||}|j ||dk r|j |jj f||idS)zSet the value on the given field. For well-known protocol buffer types which are marshalled, either the protocol buffer object or the Python equivalent is accepted. rrNr)r@rr-r>rrrJrtrrrrrdr,)rr%r(r>rrrsrrr{s   zMessage.__setattr__cCs |jS)zSerialize for pickling.)rrrrrrrszMessage.__getstate__cs"|j|}td|dS)zDeserialization for pickling.rN)r-r,rr@r)rr(Znew_pbrsrrrszMessage.__setstate__)N)rrrrrrrrrrrrrrrrrrrrrsrrPs"J& ! rP) metaclassc@sFeZdZdZeeeeeej ddddZ e e e jdddZdS) rea%Metadata about a message. Args: fields (Tuple[~.fields.Field]): The fields declared on the message. package (str): The proto package. full_name (str): The full name of the message. file_info (~._FileInfo): The file descriptor and messages for the file containing this message. marshal (~.Marshal): The marshal instance to which this message was automatically registered. options (~.descriptor_pb2.MessageOptions): Any options that were set on the message. N)r<r+r=r>r5rcCsN||_||_||_tdd|D|_tdd|D|_||_d|_dS)Ncss|]}|j|fVqdSrur3r/rrrrsz(_MessageInfo.__init__..css|]}|j|fVqdSrur#r/rrrrs) r+r=r5rNrOr<Zfields_by_numberr>r)rr<r+r=r>r5rrrrs z_MessageInfo.__init__rcCs|jS)zReturn the protobuf message type for this descriptor. If a field on the message references another message which has not loaded, then this method returns None. )rrrrrr,sz_MessageInfo.pb)rrrrrrrVrrr`rrrrrPr,rrrrres re)rP))rNcollections.abcrrKtypingrrrrrrZgoogle.protobufZgooglerrZgoogle.protobuf.json_formatr r r protor r Z proto.fieldsrrrZ proto.marshalrZproto.primitivesrZ proto.utilsrZprotobuf __version__rZ_upbr]rrPre__all__rrrrs:          c*