o }j6h.@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|dr t|dr dStS)N to_pythonto_protoT)hasattrNotImplemented)clsCrd/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/proto/marshal/marshal.py__subclasshook__*szRule.__subclasshook__N)__name__ __module__ __qualname____doc__ classmethodrrrrrr'src@sdeZdZdZddZddedefddZd d Zd d Z dd de fddZ ddde fddZ 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__Ds zBaseMarshal.__init__N proto_typerulecs<|rt|ts td|j<dSdtffdd }|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|ts td|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+r0rr/rregisterIs   zBaseMarshal.registercCsV|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|qdS)z(Reset the registry to its initial state.marshalN)1r$clearr3r Timestampr TimestampRulerDuration DurationRuler FieldMaskr FieldMaskRuler BoolValuer BoolValueRule BytesValueBytesValueRule DoubleValueDoubleValueRule FloatValueFloatValueRule Int32ValueInt32ValueRule Int64ValueInt64ValueRule StringValueStringValueRule UInt32ValueUInt32ValueRule UInt64ValueUInt64ValueRulerValuer ValueRule ListValue ListValueRuleStruct StructRulerBYTESpb_bytes BytesRuler STRINGY_NUMBER_RULES _proto_type)r(r,rrrr'{s(  zBaseMarshal.resetcCs\|j||j}||jkr,t|dr,|jD]\}}|j||j}||jkr+|Sq|S)N _instances)r$getr&rrZitems)r(r*r+_instancerrrget_rules zBaseMarshal.get_ruleabsentracCst|}|tjvrt||dS|tjvr&t|tr t|||dSt||dS|tjvs1|jtj vr7t ||dS|j |dj ||dS)Nr4)r5r*r*r`) r2rrepeated_composite_typesr repeated_scalar_typesr1r map_composite_typesrmap_composite_type_namesr r_r)r(r*valuera value_typerrrrs        zBaseMarshal.to_pythonF)strictricstjtjtjfvr)t|ttfr|jSt|tt fr)t |fdd|DSt|t rJj j rJj jrJt jfdd|DSjd|}|rft|sftdjj|jjd|S)Nc3s|] }|VqdSr#r).0ir/rr sz'BaseMarshal.to_proto..csi|] \}}||qSrrj)rkkv)recursive_typer(rr sz(BaseMarshal.to_proto..rbzLParameter must be instance of the same class; expected {expected}, got {got})expectedgot)rrOrQrSr1r r pblisttupler2dict DESCRIPTOR has_options GetOptions map_entryrgr\r_rr.formatr __class__)r(r*rgripb_valuer)r*rpr(rrs4   zBaseMarshal.to_protor#) rrrr r)r2rr3r'r_boolrrrrrrr"1s2'r"cs<eZdZdZiZdeffdd Zdeffdd 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|}|durt|}|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)rZr[super__new__)rrklassr}rrrs zMarshal.__new__cs"||_t|dstdSdS)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)_namerrr))r(rrrrr)s zMarshal.__init__) rrrr rZstrrr) __classcell__rrrrrs rc@s,eZdZdZdddefddZddZdS) r%z#A catch-all rule that does nothing.Nr`racC|Sr#r)r(r~rarrrr"zNoopRule.to_pythoncCrr#r)r(rgrrrr%rzNoopRule.to_proto)rrrr rrrrrrrr%s r%)r)abcenumgoogle.protobufrrrrrr proto.marshalrproto.marshal.collectionsr r r proto.marshal.rulesr rVr rrrrproto.primitivesrABCrr"rr%__all__rrrrs2                  J%