23s2s

✏️ Editing: proto_builder.cpython-36.pyc

���hX������������������@���s|���d�Z�yddlmZ�W�n �ek
r4���ddlmZ�Y�nX�ddlZddlZddlmZ�ddlm	Z	�dd��Z
ddd	�Zd
d��ZdS�)
zDynamic Protobuf class creator.�����)�OrderedDictN)�descriptor_pb2)�message_factoryc�������������C���s���|�j�j|�}|�j|�}|S�)a8��Get 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����
r���c�������
������C���s����t�j|d�}|dk	r8yt||�}|S��tk
r6���Y�nX�t|�j���}t|�t�sVt|�}t	j
��}x4|D�],\}}|j|jd���|jt
|�jd���qdW�|j��d�}	|dkr�d|j���}yt||�}|S��tk
r����Y�nX�|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.
  )r���Nzutf-8z.protoz6net.proto2.python.public.proto_builder.AnonymousProto_)r���ZMessageFactoryr����KeyError�list�items�
isinstancer����sorted�hashlibZsha1�update�encode�strZ	hexdigestr���ZAdd�_MakeFileDescriptorProto)
Zfieldsr���r���r���r����field_itemsZfields_hash�f_name�f_type�proto_file_namer	���r	���r
����MakeSimpleProtoClass<���s4����

r���c�������������C���s����|j�dd�\}}tj��}tjj|jdd�|��|_||_|j	j
��}||_x@t|d�D�]2\}\}}	|jj
��}
||
_||
_
tjj|
_|	|
_qRW�|S�)zAPopulate FileDescriptorProto for MessageFactory's DescriptorPool.�.�����/)�rsplitr���ZFileDescriptorProto�os�path�join�replace�name�packageZmessage_type�add�	enumerateZfieldZnumberZFieldDescriptorProtoZLABEL_OPTIONALZlabel�type)r���r���r���r$���r#���Z
file_protoZ
desc_protoZf_numberr���r���Zfield_protor	���r	���r
���r���t���s����

r���)NN)
�__doc__�collectionsr����ImportErrorZordereddictr���r���Zgoogle.protobufr���r���r���r���r���r	���r	���r	���r
����<module>���s���
8

Sid Gifari From Gifari Industries - BD Cyber Security Team