U Xzf_ @sTdZddlmZddlmZddlmZGdddejZeedZ dd Zd S) aSA database of Python protocol buffer generated symbols. SymbolDatabase is the MessageFactory for messages generated at compile time, and makes it easy to create new instances of a registered type, given only the type's protocol buffer symbol name. Example usage:: db = symbol_database.SymbolDatabase() # Register symbols of interest, from one or multiple files. db.RegisterFileDescriptor(my_proto_pb2.DESCRIPTOR) db.RegisterMessage(my_proto_pb2.MyMessage) db.RegisterEnumDescriptor(my_proto_pb2.MyEnum.DESCRIPTOR) # The database can be used as a MessageFactory, to generate types based on # their name: types = db.GetMessages(['my_proto.proto']) my_message_instance = types['MyMessage']() # The database's underlying descriptor pool can be queried, so it's not # necessary to know a type's filename to be able to generate it: filename = db.pool.FindFileContainingSymbol('MyMessage') my_message_instance = db.GetMessages([filename])['MyMessage']() # This functionality is also provided directly via a convenience method: my_message_instance = db.GetSymbol('MyMessage')() )api_implementation)descriptor_pool)message_factoryc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)SymbolDatabasez'A database of Python generated symbols.cCs|j}||j|<|||S)a@Registers the given message type in the local database. Calls to GetSymbol() and GetMessages() will return messages registered here. Args: message: A :class:`google.protobuf.message.Message` subclass (or instance); its descriptor will be registered. Returns: The provided message. )Z DESCRIPTOR_classesRegisterMessageDescriptor)selfmessagedescr w/private/var/folders/n9/53xbvtmd7sjg1q1l55xmpg58n4mgng/T/pip-unpacked-wheel-3cr5c46n/google/protobuf/symbol_database.pyRegisterMessageEs   zSymbolDatabase.RegisterMessagecCstdkr|j|dS)zRegisters the given message descriptor in the local database. Args: message_descriptor (Descriptor): the message descriptor to add. pythonN)rTypepoolZ_AddDescriptor)rZmessage_descriptorr r r rWs z(SymbolDatabase.RegisterMessageDescriptorcCstdkr|j||S)zRegisters the given enum descriptor in the local database. Args: enum_descriptor (EnumDescriptor): The enum descriptor to register. Returns: EnumDescriptor: The provided descriptor. r)rrrZ_AddEnumDescriptor)rZenum_descriptorr r r RegisterEnumDescriptoras  z%SymbolDatabase.RegisterEnumDescriptorcCstdkr|j|dS)zRegisters the given service descriptor in the local database. Args: service_descriptor (ServiceDescriptor): the service descriptor to register. rN)rrrZ_AddServiceDescriptor)rZservice_descriptorr r r RegisterServiceDescriptoros z(SymbolDatabase.RegisterServiceDescriptorcCstdkr|j|dS)zRegisters the given file descriptor in the local database. Args: file_descriptor (FileDescriptor): The file descriptor to register. rN)rrrZ_InternalAddFileDescriptor)rZfile_descriptorr r r RegisterFileDescriptorzs z%SymbolDatabase.RegisterFileDescriptorcCs|j|j|S)awTries to find a symbol in the local database. Currently, this method only returns message.Message instances, however, if may be extended in future to support other symbol types. Args: symbol (str): a protocol buffer symbol. Returns: A Python class corresponding to the symbol. Raises: KeyError: if the symbol could not be found. )rrZFindMessageTypeByName)rsymbolr r r GetSymbolszSymbolDatabase.GetSymbolc srfddi}|D]X}|j|}|jD]<}|D].}z|j|||j<Wq:tk rfYq:Xq:q.q|S)aGets all registered messages from a specified file. Only messages already created and registered will be returned; (this is the case for imported _pb2 modules) But unlike MessageFactory, this version also returns already defined nested messages, but does not register any message extensions. Args: files (list[str]): The file names to extract messages from. Returns: A dictionary mapping proto names to the message classes. Raises: KeyError: if a file could not be found. c3s*|V|jD]}|D] }|Vqq dS)zCWalk a message Descriptor and recursively yields all message names.N)Z nested_types)r msg_descZ nested_desc_GetAllMessagesr r rs  z3SymbolDatabase.GetMessages.._GetAllMessages)rZFindFileByNameZmessage_types_by_namevaluesrZ full_nameKeyError)rfilesresult file_nameZ file_descrr r rr GetMessagess    zSymbolDatabase.GetMessagesN) __name__ __module__ __qualname____doc__r rrrrrrr r r r rBs   r)rcCstS)z#Returns the default SymbolDatabase.)_DEFAULTr r r r Defaultsr$N) r"Zgoogle.protobuf.internalrZgoogle.protobufrrZMessageFactoryrr$r#r r r r s    {