o h@sHdZddlmZddlmZGdddejZeedZddZd S) aRA 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')() )descriptor_pool)message_factoryc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)SymbolDatabasez'A database of Python generated symbols.cCs|j}||j|<|||S)zRegisters the given message type in the local database. Calls to GetSymbol() and GetMessages() will return messages registered here. Args: message: a message.Message, to be registered. Returns: The provided message. ) DESCRIPTOR_classesRegisterMessageDescriptor)selfmessagedescr Q/var/www/html/env/lib/python3.10/site-packages/google/protobuf/symbol_database.pyRegisterMessageDs  zSymbolDatabase.RegisterMessagecC|j|dS)zRegisters the given message descriptor in the local database. Args: message_descriptor: a descriptor.MessageDescriptor. N)pool AddDescriptor)rmessage_descriptorr r r rUsz(SymbolDatabase.RegisterMessageDescriptorcCs|j||S)zRegisters the given enum descriptor in the local database. Args: enum_descriptor: a descriptor.EnumDescriptor. Returns: The provided descriptor. )rAddEnumDescriptor)renum_descriptorr r r RegisterEnumDescriptor]s z%SymbolDatabase.RegisterEnumDescriptorcCr)zRegisters the given service descriptor in the local database. Args: service_descriptor: a descriptor.ServiceDescriptor. Returns: The provided descriptor. N)rAddServiceDescriptor)rservice_descriptorr r r RegisterServiceDescriptori z(SymbolDatabase.RegisterServiceDescriptorcCr)zRegisters the given file descriptor in the local database. Args: file_descriptor: a descriptor.FileDescriptor. Returns: The provided descriptor. N)rAddFileDescriptor)rfile_descriptorr r r RegisterFileDescriptortrz%SymbolDatabase.RegisterFileDescriptorcCs|j|j|S)axTries 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: A str, a protocol buffer symbol. Returns: A Python class corresponding to the symbol. Raises: KeyError: if the symbol could not be found. )rrFindMessageTypeByName)rsymbolr r r GetSymbolszSymbolDatabase.GetSymbolc snfddi}|D]*}|j|}|jD]}|D]}z |j|||j<Wqty2Yqwqq |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: 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]}|Vq qdS)zCWalk a message Descriptor and recursively yields all message names.N) nested_types)r msg_desc nested_desc_GetAllMessagesr r r#s  z3SymbolDatabase.GetMessages.._GetAllMessages)rFindFileByNamemessage_types_by_namevaluesr full_nameKeyError)rfilesresult file_name file_descr r r r"r GetMessagess    zSymbolDatabase.GetMessagesN) __name__ __module__ __qualname____doc__r rrrrrr-r r r r rAs   r)rcCstS)z#Returns the default SymbolDatabase.)_DEFAULTr r r r Defaultsr3N)r1google.protobufrrMessageFactoryrr3r2r r r r s   w