U
|
£ý°dó ã @ sz d Z dZddlmZ ddlmZ ddlmZ e ¡ dkrFddlm Z
|
n ddlm Z
|
e
|
j Z G d d
|
d
|
eZe Zd d Zd 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)é )Úapi_implementation)Údescriptor_pool)ÚmessageÚcpp)Ú cpp_message)Úpython_messagec @ s2 e Zd ZdZd ddZdd Zdd Zd d
|
ZdS ) ÚMessageFactoryz@Factory for creating Proto2 messages from descriptors in a pool.Nc C s |p
|
t ¡ | _i | _dS )zInitializes a new factory.N)r ZDescriptorPoolÚpoolÚ_classes)Úselfr © r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\google/protobuf/message_factory.pyÚ__init__; s zMessageFactory.__init__c C s, || j kr"| |¡}|| j |< |S | j | S )aE Obtains 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.
|
)r
|
ÚCreatePrototype)r Ú
|
descriptorÚ result_classr r r Ú GetPrototypeB s
|
|
|
|
zMessageFactory.GetPrototypec C s |j }t|tjf|dd}| |_|| j|< |jD ]}|jr2| |j¡ q2|j j
|
D ]2}|j | jkrn| |j ¡ | j|j }| |¡ qR|S )a¢ Builds a proto2 message class based on the passed in descriptor.
|
|
Don't call this function directly, it always creates a new class. Call
|
GetPrototype() instead. This method is meant to be overridden in subblasses
|
to perform additional operations on the newly constructed class.
|
|
Args:
|
descriptor: The descriptor to build from.
|
|
Returns:
|
A class describing the passed in descriptor.
|
N)Ú
|
DESCRIPTORÚ
|
__module__) ÚnameÚ _GENERATED_PROTOCOL_MESSAGE_TYPEr ÚMessageÚ_FACTORYr
|
ÚfieldsZ message_typer r Ú
|
extensionsÚcontaining_typeÚRegisterExtension)r r Zdescriptor_namer ÚfieldÚ extensionÚextended_classr r r r W s& ýý
|
|
z MessageFactory.CreatePrototypec C s| i }|D ]n}| j |¡}|j ¡ D ]}| |¡||j< q"|j ¡ D ]2}|j| jkr^| |j¡ | j|j }| |¡ qBq|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.
|
)
|
r ZFindFileByNameZmessage_types_by_nameÚvaluesr Z full_nameZextensions_by_namer r
|
r )r ÚfilesÚresultÚ file_nameZ file_descÚdescr r r r r Ú GetMessages{ s zMessageFactory.GetMessages)N)Ú__name__r Ú __qualname__Ú__doc__r r r r% r r r r r 8 s
|
|
$r c sF dd | D fdd r2 ¡ d q t dd | 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 S s i | ] }|j |qS r ©r ©Ú.0Ú
|
file_protor r r Ú
|
<dictcomp>° s z GetMessages.<locals>.<dictcomp>c s2 | j D ]}|kr |¡ qtj | ¡ d S )N)Ú
|
dependencyÚpopr r ZAdd)r, r. ©Ú_AddFileZ file_by_namer r r1 ± s
|
z GetMessages.<locals>._AddFileé c S s g | ]
|
}|j qS r r) r* r r r Ú
|
<listcomp>¹ s z GetMessages.<locals>.<listcomp>)Úpopitemr r% )Z file_protosr r0 r r% £ s
|
r% N)r( Ú
|
__author__Zgoogle.protobuf.internalr Zgoogle.protobufr r ÚTypeZgoogle.protobuf.pyextr Z message_implr Z GeneratedProtocolMessageTyper Úobjectr r r% r r r r Ú<module> s h
|
