o ˷e@sbdZddlmZddlZddlZddlmZddlmZddlmZddZ d d d Z d d Z dS)zDynamic Protobuf class creator.) OrderedDictN)descriptor_pb2) descriptor)message_factorycCs|j|}||}|S)a8Get a proto class from the MessageFactory by name. Args: factory: a MessageFactory instance. full_name: str, the fully qualified name of the proto type. Returns: A class, for the type identified by full_name. Raises: KeyError, if the proto is not found in the factory's descriptor pool. )poolFindMessageTypeByName GetPrototype)factory full_nameproto_descriptor proto_clsr T/var/www/ideatree/venv/lib/python3.10/site-packages/google/protobuf/proto_builder.py_GetMessageFromFactory*s rc Cstj|d}|durzt||}|WStyYnw|}t|ts)t|}t }|D]\}}| | d| t | dq/| d} |durhd| }zt||}|WStygYnw|jt| ||t||S)aCreate a Protobuf class whose fields are basic types. Note: this doesn't validate field names! Args: fields: dict of {name: field_type} mappings for each field in the proto. If this is an OrderedDict the order will be maintained, otherwise the fields will be sorted by name. full_name: optional str, the fully-qualified name of the proto type. pool: optional DescriptorPool instance. Returns: a class, the new protobuf class with a FileDescriptor. )rNzutf-8z.protoz6net.proto2.python.public.proto_builder.AnonymousProto_)rMessageFactoryrKeyErroritems isinstancersortedhashlibsha1updateencodestr hexdigestrAdd_MakeFileDescriptorProto) fieldsr rr r field_items fields_hashf_namef_typeproto_file_namer r rMakeSimpleProtoClass:s<          r#c Cs|dd\}}t}tj|dd||_||_|j }||_t |dD]+\}\}} |j } || _|t jjkrH|t jjt jjd7}|| _tjj| _| | _q(|S)zAPopulate FileDescriptorProto for MessageFactory's DescriptorPool../)rsplitrFileDescriptorProtoospathjoinreplacenamepackage message_typeadd enumeratefieldrFieldDescriptorFIRST_RESERVED_FIELD_NUMBERLAST_RESERVED_FIELD_NUMBERnumberFieldDescriptorProtoLABEL_OPTIONALlabeltype) r"r rr.r- file_proto desc_protof_numberr r! field_protor r rrrs*    r)NN) __doc__ collectionsrrr)google.protobufrrrrr#rr r r rs      8