g5~dZddlZddlmZddlmZddlmZGddZeejZ d Zy) 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')() N)api_implementation)descriptor_pool)message_factorycXeZdZdZiZddZdZdZdZdZ dZ d Z d Z d Z d Zd Zy)SymbolDatabasez'A database of Python generated symbols.Nc>|xstj|_y)z!Initializes a new SymbolDatabase.N)rDescriptorPoolpool)selfr s T/var/www/openai/venv/lib/python3.12/site-packages/google/protobuf/symbol_database.py__init__zSymbolDatabase.__init__2s8668DIcVtjdtj|S)NzSymbolDatabase.GetPrototype() is deprecated. Please use message_factory.GetMessageClass() instead. SymbolDatabase.GetPrototype() will be removed soon.)warningswarnrGetMessageClassr descriptors r GetPrototypezSymbolDatabase.GetPrototype6s( MMHI  * *: 66rcVtjdtj|S)NzDirectly call CreatePrototype() is wrong. Please use message_factory.GetMessageClass() instead. SymbolDatabase.CreatePrototype() will be removed soon.)rrr_InternalCreateMessageClassrs r CreatePrototypezSymbolDatabase.CreatePrototype<s( MMKL  6 6z BBrcltjdtj||jS)NzSymbolDatabase.GetMessages() is deprecated. Please use message_factory.GetMessageClassedForFiles() instead. SymbolDatabase.GetMessages() will be removed soon.)rrrGetMessageClassedForFilesr )r filess r GetMessageszSymbolDatabase.GetMessagesBs. MMGH  4 4UDII FFrc^|j}||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. ) DESCRIPTOR_classesRegisterMessageDescriptor)r messagedescs r RegisterMessagezSymbolDatabase.RegisterMessageHs0   D!DMM$""4( Nrcjtjdk(r|jj|yy)zRegisters the given message descriptor in the local database. Args: message_descriptor (Descriptor): the message descriptor to add. pythonN)rTyper _AddDescriptor)r message_descriptors r r z(SymbolDatabase.RegisterMessageDescriptorZs-  H, ii12-rcjtjdk(r|jj||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%)rr&r _AddEnumDescriptor)r enum_descriptors r RegisterEnumDescriptorz%SymbolDatabase.RegisterEnumDescriptords- H, ii""?3 rcjtjdk(r|jj|yy)zRegisters the given service descriptor in the local database. Args: service_descriptor (ServiceDescriptor): the service descriptor to register. r%N)rr&r _AddServiceDescriptor)r service_descriptors r RegisterServiceDescriptorz(SymbolDatabase.RegisterServiceDescriptorrs- H, ii%%&89-rcjtjdk(r|jj|yy)zRegisters the given file descriptor in the local database. Args: file_descriptor (FileDescriptor): The file descriptor to register. r%N)rr&r _InternalAddFileDescriptor)r file_descriptors r RegisterFileDescriptorz%SymbolDatabase.RegisterFileDescriptor}s,  H, ii**?;-rcR|j|jj|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. )rr FindMessageTypeByName)r symbols r GetSymbolzSymbolDatabase.GetSymbols" ==88@ AArc fdi}|D]f}|jj|}|jjD],}|D]} |j|||j <!.h|S#t $rY3wxYw)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. c3ZK||jD]}|D]}|yw)zCWalk a message Descriptor and recursively yields all message names.N) nested_types)r"msg_desc nested_desc_GetAllMessagess r r>z3SymbolDatabase.GetMessages.._GetAllMessagess1 j''(*84K 5(s(+)r FindFileByNamemessage_types_by_namevaluesr full_nameKeyError)r rresult file_name file_descr<r"r>s @r rzSymbolDatabase.GetMessagess&F ))**95i55<<>(#H-D %)]]4%8F4>> ".? M   sA66 B B )N)__name__ __module__ __qualname____doc__rr rrrr#r r,r0r4r8rr rr,sF/(97 C G $3  :<B$$rr)r ctS)z#Returns the default SymbolDatabase.)_DEFAULTrKrr DefaultrNs /r) rJrgoogle.protobuf.internalrgoogle.protobufrrrrNrMrKrr rQsB:7++QQh 6668 9r