✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

3

���h��@sNdZd
Z
ddlmZ
ddlmZ
ddlmZ
Gdd�de�Ze�Zdd	�Z	d
S)ao
Provides a factory class for generating dynamic messages.

The easiest way to use this class is if you have access to the FileDescriptor
protos containing the messages you want to create you can just do the following:

message_classes = message_factory.GetMessages(iterable_of_file_descriptors)
my_proto_instance = message_classes['some.proto.package.MessageName']()
z"matthewtoia@google.com (Matt Toia)�)
�descriptor_pool)
�message)
�
reflectionc@s*eZ
dZd
Zd	dd�
Zdd�Zdd�ZdS)
�MessageFactoryz@Factory for creating Proto2 messages from descriptors in a pool.NcCs|
p
tj
�|_i|_d
S)zInitializes a new factory.N)rZDescriptorPool�pool�_classes)�selfr�r	�%/usr/lib/python3.6/message_factory.py�__init__2szMessageFactory.__init__cCs�|
|jkr�|
j
}ttkr&|
j
jd
d�}tj|tjf
|
dd��}||j|
<x |
j	D]}|j
rP|j|j
�

qPWx>|jj
D]2}|j|jkr�|j|j�

|j|j}|j|�

qtW|j|
S)aD
Builds a proto2 message class based on the passed in descriptor.

    Passing a descriptor with a fully qualified name matching a previous
    invocation will cause the same class to be returned.

    Args:
      descriptor: The descriptor to build from.

    Returns:
      A class describing the passed in descriptor.
    �ascii�ignoreN)�
DESCRIPTOR�
__module__)r�name�str�bytes�encoderZGeneratedProtocolMessageTyperZMessageZfieldsZmessage_type�GetPrototyper�
extensions�containing_type�RegisterExtension)rZ
descriptorZdescriptor_nameZresult_classZfield�	extension�extended_classr	r	r
r9s$

















zMessageFactory.GetPrototypecCs�i}x�|
D]~}|jj
|�
}x&t|jj��
D]}|j|�
||j<q*WxDt|jj��
D]2}|j|j	krn|j|j�

|j	|j}|j
|�

qRWq
W|S)
a�
Gets all the messages from a specified file.

    This will find and resolve dependencies, failing if the descriptor
    pool cannot satisfy them.

    Args:
      files: The file names to extract messages from.

    Returns:
      A dictionary mapping proto names to the message classes. This will include
      any dependent messages as well as any messages defined in the same file as
      a specified message.
    )rZFindFileByName�listZmessage_types_by_name�valuesrZ	full_nameZextensions_by_namerrr)r�files�result�	file_nameZ	file_descZdescrrr	r	r
�GetMessagesYs









zMessageFactory.GetMessages)
N)�__name__r�__qualname__�__doc__rrrr	r	r	r
r/s

 rc

sJd
d�|D�
�
��
fdd��x�
r4��
j�d�

qWt
jdd�|D�
�
S)a^
Builds a dictionary of all the messages available in a set of files.

  Args:
    file_protos: Iterable of FileDescriptorProto to build messages out of.

  Returns:
    A dictionary mapping proto names to the message classes. This will include
    any dependent messages as well as any messages defined in the same file as
    a specified message.
  c
Ssi|]}
|
|
j�qSr	)
r)�.0�
file_protor	r	r
�
<dictcomp>�szGetMessages.<locals>.<dictcomp>c
s6x$|jD]}
|
�
kr��
j
|
�
�

qWtjj|�

dS)
N)�
dependency�pop�_FACTORYrZAdd)r$r&)�_AddFile�file_by_namer	r
r)�s


zGetMessages.<locals>._AddFile�
c
Ssg|]
}
|
j�qSr	)
r)r#r$r	r	r
�
<listcomp>�szGetMessages.<locals>.<listcomp>)�popitemr(r)
Zfile_protosr	)r)r*r
r�s




rN)
r"�
__author__Zgoogle.protobufrrr�objectrr(rr	r	r	r
�<module>&s

O