U S³dŠ#ã@s–dZdZddlZddlmZddlmZddlmZe ¡dkrNddlm Z n dd l m Z e j Zd d „Zd d „Zdd„ZGdd„deƒZddd„ZdS)aoProvides 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)éN)Úapi_implementation)Údescriptor_pool)ÚmessageÚpython)Úpython_message)Ú cpp_messagecCst|ddƒ}|r|St|ƒS)a7Obtains 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. Z_concrete_classN)ÚgetattrÚ_InternalCreateMessageClass)Ú descriptorZconcrete_class©r úC/tmp/pip-unpacked-wheel-1ori_g8k/google/protobuf/message_factory.pyÚGetMessageClass:s r cCsni}|D]`}| |¡}|j ¡D]}t|ƒ||j<q |j ¡D](}t|jƒ}| |¡|jr>t|jƒq>q|S)akGets all the messages from specified files. This will find and resolve dependencies, failing if the descriptor pool cannot satisfy them. Args: files: The file names to extract messages from. pool: The descriptor pool to find the files including the dependent files. Returns: A dictionary mapping proto names to the message classes. ) ZFindFileByNameZmessage_types_by_nameÚvaluesr Z full_nameZextensions_by_nameÚcontaining_typeÚRegisterExtensionÚ message_type)ÚfilesÚpoolÚresultÚ file_nameZ file_descÚdescÚ extensionÚextended_classr r r ÚGetMessageClassesForFilesLs    rcCsn|j}t|tjf|ddœƒ}|jD]}|jr"t|jƒq"|jjD](}t|j ƒ}|  |¡|jr@t|jƒq@|S)z·Builds a proto2 message class based on the passed in descriptor. Args: descriptor: The descriptor to build from. Returns: A class describing the passed in descriptor. N)Ú DESCRIPTORÚ __module__) ÚnameÚ _GENERATED_PROTOCOL_MESSAGE_TYPErÚMessageÚfieldsrr rÚ extensionsrr)r Zdescriptor_nameZ result_classÚfieldrrr r r r ts" ýý      r c@s2eZdZdZd dd„Zdd„Zdd„Zd d „ZdS) ÚMessageFactoryz@Factory for creating Proto2 messages from descriptors in a pool.NcCs|p t ¡|_dS)zInitializes a new factory.N)rÚDescriptorPoolr)Úselfrr r r Ú__init__–szMessageFactory.__init__cCst|ƒS)aEObtains 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 ©r$r r r r Ú GetPrototypešszMessageFactory.GetPrototypecCst|ƒ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 GetMessageClass() instead. Args: descriptor: The descriptor to build from. Returns: A class describing the passed in descriptor. )r r&r r r ÚCreatePrototype¬szMessageFactory.CreatePrototypecCs t||jƒ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. )rr)r$rr r r Ú GetMessages¾szMessageFactory.GetMessages)N)Ú__name__rÚ __qualname__Ú__doc__r%r'r(r)r r r r r"“s  r"csT|p t ¡‰dd„|Dƒ‰‡‡‡fdd„‰ˆr@ˆˆ ¡dƒq*tdd„|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. pool: The descriptor pool to add the file protos. 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. cSsi|] }|j|“qSr ©r©Ú.0Ú file_protor r r Ú ãszGetMessages..cs0|jD]}|ˆkrˆˆ |¡ƒqˆ |¡dS)N)Ú dependencyÚpopZAdd)r0r2©Ú_AddFileZdes_poolZ file_by_namer r r5äs zGetMessages.._AddFileécSsg|] }|j‘qSr r-r.r r r Ú íszGetMessages..)rr#Úpopitemr)Z file_protosrr r4r r)Ôs  ÿr))N)r,Ú __author__ÚwarningsZgoogle.protobuf.internalrZgoogle.protobufrrÚTyperZ message_implZgoogle.protobuf.pyextrZGeneratedProtocolMessageTyperr rr Úobjectr"r)r r r r Ús      (A