U #FZh.@s"ddlZddlZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l m Z dd l mZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZGdddejZGdddZGdddeZGdddZdZdS)N)message) duration_pb2) timestamp_pb2)field_mask_pb2) struct_pb2) wrappers_pb2)compat) MapComposite)Repeated)RepeatedComposite)bytes)stringy_numbers)dates)struct)wrappers) field_mask) ProtoTypec@seZdZdZeddZdS)Rulez,Abstract class definition for marshal rules.cCst|drt|drdStS)N to_pythonto_protoT)hasattrNotImplemented)clsCr\/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/site-packages/proto/marshal/marshal.py__subclasshook__*szRule.__subclasshook__N)__name__ __module__ __qualname____doc__ classmethodrrrrrr'src@sbeZdZdZddZdeedddZdd Zd d Z dd e d d dZ dde dddZ dS) BaseMarshalaThe base class to translate between protobuf and Python classes. Protocol buffers defines many common types (e.g. Timestamp, Duration) which also exist in the Python standard library. The marshal essentially translates between these: it keeps a registry of common protocol buffers and their Python representations, and translates back and forth. The protocol buffer class is always the "key" in this relationship; when presenting a message, the declared field types are used to determine whether a value should be transformed into another class. Similarly, when accepting a Python value (when setting a field, for example), the declared field type is still used. This means that, if appropriate, multiple protocol buffer types may use the same Python type. The primary implementation of this is :class:`Marshal`, which should usually be used instead of this class directly. cCsi|_t|_|dSN)_rulesNoopRule_noopreset)selfrrr__init__DszBaseMarshal.__init__N) proto_typerulecs<|r$t|tstd|j<dStdfdd }|S)aRegister a rule against the given ``proto_type``. This function expects a ``proto_type`` (the descriptor class) and a ``rule``; an object with a ``to_python`` and ``to_proto`` method. Each method should return the appropriate Python or protocol buffer type, and be idempotent (e.g. accept either type as input). This function can also be used as a decorator:: @marshal.register(timestamp_pb2.Timestamp) class TimestampRule: ... In this case, the class will be initialized for you with zero arguments. Args: proto_type (type): A protocol buffer message type. rule: A marshal object zIMarshal rule instances must implement `to_proto` and `to_python` methods.N rule_classcs"t|tstd|j<|S)NzJMarshal rule subclasses must implement `to_proto` and `to_python` methods.) issubclassr TypeErrorr$r,r*r(rrregister_rule_classms   z1BaseMarshal.register..register_rule_class) isinstancerr/r$type)r(r*r+r1rr0rregisterIs   zBaseMarshal.registercCsX|j|tjt|tjt |t j t |tjt|tjt|tjt|tjt|tjt|tjt|tjt|tjt|tj t!|t"j#t$j%|d|t"j&t$j'|d|t"j(t$j)|d|t*j+t,-t.j/D]}||j0|qsz'BaseMarshal.to_proto..csi|]\}}||qSrrH)rIkv)recursive_typer(rr sz(BaseMarshal.to_proto..rEzLParameter must be instance of the same class; expected {expected}, got {got})expectedgot)rr9r:r;r2r r Zpblisttupler3dictZ DESCRIPTORZ has_optionsZ GetOptionsZ map_entryrFr?rBrr/formatr __class__)r(r*rFrGpb_valuer)r*rNr(rrs2   zBaseMarshal.to_proto)N) rrrr r)r3rr4r'rBboolrrrrrrr"1s2'r"cs<eZdZdZiZedfdd Zedfdd ZZS)Marshala The translator between protocol buffer and Python instances. The bulk of the implementation is in :class:`BaseMarshal`. This class adds identity tracking: multiple instantiations of :class:`Marshal` with the same name will provide the same instance. )namecs.|j|}|dkr*t|}|j|<|S)zCreate a marshal instance. Args: name (str): The name of the marshal. Instantiating multiple marshals with the same ``name`` argument will provide the same marshal each time. N)r=r>super__new__)rrZklassrVrrr\s zMarshal.__new__cs||_t|dstdS)zInstantiate a marshal. Args: name (str): The name of the marshal. Instantiating multiple marshals with the same ``name`` argument will provide the same marshal each time. r$N)_namerr[r))r(rZr^rrr)s zMarshal.__init__) rrrr r=strr\r) __classcell__rrr^rrYsrYc@s,eZdZdZddedddZddZdS)r%z#A catch-all rule that does nothing.NrCcCs|Sr#r)r(rWrDrrrr"szNoopRule.to_pythoncCs|Sr#r)r(rFrrrr%szNoopRule.to_proto)rrrr rXrrrrrrr%sr%)rY)abcenumZgoogle.protobufrrrrrrZ proto.marshalrZproto.marshal.collectionsr r r Zproto.marshal.rulesr r<r rrrrZproto.primitivesrABCrr"rYr%__all__rrrrs0                  J%