Lh dZddlmZddlmZddlmZGddejZeejZ dZd 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<eZdZdZdZdZdZdZdZdZ dZ d S) SymbolDatabasez'A database of Python generated symbols.cR|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)selfmessagedescs g/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/google/protobuf/symbol_database.pyRegisterMessagezSymbolDatabase.RegisterMessageEs1  D!DM$""4((( Ncltjdkr|j|dSdS)zRegisters the given message descriptor in the local database. Args: message_descriptor (Descriptor): the message descriptor to add. pythonN)rTypepool_AddDescriptor)r message_descriptors rr z(SymbolDatabase.RegisterMessageDescriptorWs=   H,, i122222-,rchtjdkr|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)rrr_AddEnumDescriptor)r enum_descriptors rRegisterEnumDescriptorz%SymbolDatabase.RegisterEnumDescriptoras4  H,, i""?333 rcltjdkr|j|dSdS)zRegisters the given service descriptor in the local database. Args: service_descriptor (ServiceDescriptor): the service descriptor to register. rN)rrr_AddServiceDescriptor)r service_descriptors rRegisterServiceDescriptorz(SymbolDatabase.RegisterServiceDescriptoros=  H,, i%%&899999-,rcltjdkr|j|dSdS)zRegisters the given file descriptor in the local database. Args: file_descriptor (FileDescriptor): The file descriptor to register. rN)rrr_InternalAddFileDescriptor)r file_descriptors rRegisterFileDescriptorz%SymbolDatabase.RegisterFileDescriptorzs<   H,, i**?;;;;;-,rcL|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. )r rFindMessageTypeByName)r symbols r GetSymbolzSymbolDatabase.GetSymbols! =88@@ AArcfdi}|D]l}|j|}|jD]6}|D](} |j|||j<#t $rY%wxYw7m|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. c3LK|V|jD]}|D]}|VdS)zCWalk a message Descriptor and recursively yields all message names.N) nested_types)rmsg_desc nested_desc_GetAllMessagess rr-z3SymbolDatabase.GetMessages.._GetAllMessagessY jjj'(*?844  K     r)rFindFileByNamemessage_types_by_namevaluesr full_nameKeyError)r filesresult file_name file_descr+rr-s @r GetMessageszSymbolDatabase.GetMessagess&F )**955i5<<>>(#OH--  D %)]4%8F4> " "    D   MsA&& A3 2A3 N) __name__ __module__ __qualname____doc__rr rrr#r'r7rrrrBs//$333    : : :<<<BBB$$$$$$rr)rctS)z#Returns the default SymbolDatabase.)_DEFAULTr<rrDefaultr?s /rN) r;google.protobuf.internalrgoogle.protobufrrMessageFactoryrr?r>r<rrrCs><877777++++++++++++xxxxx_3xxxv >6688 9 9 9r