3 ��hX�@s|dZyddlmZWn ek r4ddlmZYnXddlZddlZddlmZddlm Z dd�Z ddd �Zd d�ZdS) zDynamic Protobuf class creator.�)�OrderedDictN)�descriptor_pb2)�message_factorycCs|jj|�}|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. )�poolZFindMessageTypeByNameZGetPrototype)�factory� full_nameZproto_descriptor� proto_cls�r �#/usr/lib/python3.6/proto_builder.py�_GetMessageFromFactory,s rc Cs�tj|d�}|dk r8yt||�}|Stk r6YnXt|j��}t|t�sVt|�}t j �}x4|D],\}}|j|jd��|jt |�jd��qdW|j�d} |dkr�d|j�}yt||�}|Stk r�YnX|jjt| ||��t||�S)a�Create 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�KeyError�list�items� isinstancer�sorted�hashlibZsha1�update�encode�strZ hexdigestrZAdd�_MakeFileDescriptorProto) Zfieldsrrrr�field_itemsZfields_hash�f_name�f_type�proto_file_namer r r �MakeSimpleProtoClass<s4 rcCs�|jdd�\}}tj�}tjj|jdd�|�|_||_|j j �}||_x@t|d�D]2\}\}} |jj �} || _|| _ tjj| _| | _qRW|S)zAPopulate FileDescriptorProto for MessageFactory's DescriptorPool.�.��/)�rsplitrZFileDescriptorProto�os�path�join�replace�name�packageZmessage_type�add� enumerateZfieldZnumberZFieldDescriptorProtoZLABEL_OPTIONALZlabel�type)rrrr$r#Z file_protoZ desc_protoZf_numberrrZfield_protor r r rts r)NN) �__doc__�collectionsr�ImportErrorZordereddictrrZgoogle.protobufrrrrrr r r r �s 8