a h=ic@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. )poolZFindMessageTypeByNameZ GetPrototype)factory full_nameZproto_descriptor proto_clsr n/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/google/protobuf/proto_builder.py_GetMessageFromFactory*s  r c Cstj|d}|dur8zt||}|WSty6Yn0|}t|tsRt|}t }|D],\}}| | d| t | dq^| d} |durd| }zt||}|WStyYn0|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_)rZMessageFactoryr KeyErroritems isinstancersortedhashlibsha1updateencodestr hexdigestrAdd_MakeFileDescriptorProto) fieldsrrrr field_itemsZ fields_hashf_namef_typeproto_file_namer r r MakeSimpleProtoClass:s8         rc Cs|dd\}}t}tj|dd||_||_|j }||_t |dD]V\}\}} |j } || _|t jjkr|t jjt jjd7}|| _tjj| _| | _qP|S)zAPopulate FileDescriptorProto for MessageFactory's DescriptorPool../)rsplitrZFileDescriptorProtoospathjoinreplacenamepackageZ message_typeadd enumeratefieldrZFieldDescriptorZFIRST_RESERVED_FIELD_NUMBERZLAST_RESERVED_FIELD_NUMBERnumberZFieldDescriptorProtoZLABEL_OPTIONALlabeltype) rrrr(r'Z file_protoZ desc_protoZf_numberrrZ field_protor r r rrs*    r)NN) __doc__ collectionsrrr#Zgoogle.protobufrrrr rrr r r r s     8