U Sd@sndZddlmZddlZddlZddlmZddlmZddlmZddlm Z dd Z dd d Z d d Z dS)zDynamic Protobuf class creator.) OrderedDictN)descriptor_pb2) descriptor)descriptor_pool)message_factorycCs||}t|}|S)a-Get a proto class from the MessageFactory by name. Args: pool: a descriptor pool. 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. )ZFindMessageTypeByNamerZGetMessageClass)pool full_nameZproto_descriptor proto_clsr A/tmp/pip-unpacked-wheel-1ori_g8k/google/protobuf/proto_builder.py_GetMessageFromFactory+s  r c Cs|p t}|dk r:zt||}|WStk r8YnX|}t|tsTt|}t }|D],\}}| | d| t | dq`| d} |dkrd| }zt||}|WStk rYnX|t| ||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. Nzutf-8z.protoz6net.proto2.python.public.proto_builder.AnonymousProto_)rZDescriptorPoolr KeyErroritems isinstancersortedhashlibsha1updateencodestr hexdigestZAdd_MakeFileDescriptorProto) fieldsrrZ pool_instancer 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.protobufrrrrr rrr r r r s      7