Nb? d Z ddlZddlmZ ddlmZ ddlmZ G d d Z e ej Z d ZdS )
aS A 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_factoryc Z e Zd ZdZi ZddZd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
ZdS )SymbolDatabasez'A database of Python generated symbols.Nc : |pt j | _ dS )z!Initializes a new SymbolDatabase.N)r DescriptorPoolpool)selfr
s /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/google/protobuf/symbol_database.py__init__zSymbolDatabase.__init__I s 8688DIII c R t j d t j | S )NzSymbolDatabase.GetPrototype() is deprecated. Please use message_factory.GetMessageClass() instead. SymbolDatabase.GetPrototype() will be removed soon.)warningswarnr GetMessageClassr
descriptors r GetPrototypezSymbolDatabase.GetPrototypeM s2 M H I I I *: 6 66r c R t j d t j | S )NzDirectly call CreatePrototype() is wrong. Please use message_factory.GetMessageClass() instead. SymbolDatabase.CreatePrototype() will be removed soon.)r r r _InternalCreateMessageClassr s r CreatePrototypez SymbolDatabase.CreatePrototypeS s2 M K L L L 6z B BBr c ^ t j d t j || j S )NzSymbolDatabase.GetMessages() is deprecated. Please use message_factory.GetMessageClassedForFiles() instead. SymbolDatabase.GetMessages() will be removed soon.)r r r GetMessageClassedForFilesr
)r filess r GetMessageszSymbolDatabase.GetMessagesY s6 M G H H H 4UDI F FFr c R |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 RegisterMessagez SymbolDatabase.RegisterMessage_ s1 D!DM$""4((( Nr c l t j dk r | j | dS dS )zRegisters the given message descriptor in the local database.
Args:
message_descriptor (Descriptor): the message descriptor to add.
pythonN)r Typer
_AddDescriptor)r message_descriptors r r z(SymbolDatabase.RegisterMessageDescriptorq s= H,,
i 122222 -,r c h t j dk r| 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% )r r& r
_AddEnumDescriptor)r enum_descriptors r RegisterEnumDescriptorz%SymbolDatabase.RegisterEnumDescriptor{ s4 H,,
i""?333 r c l t j dk r | j | dS dS )zRegisters the given service descriptor in the local database.
Args:
service_descriptor (ServiceDescriptor): the service descriptor to
register.
r% N)r r& r
_AddServiceDescriptor)r service_descriptors r RegisterServiceDescriptorz(SymbolDatabase.RegisterServiceDescriptor s= H,,
i%%&899999 -,r c l t j dk r | j | dS dS )zRegisters the given file descriptor in the local database.
Args:
file_descriptor (FileDescriptor): The file descriptor to register.
r% N)r r& r
_InternalAddFileDescriptor)r file_descriptors r RegisterFileDescriptorz%SymbolDatabase.RegisterFileDescriptor s< H,,
i**?;;;;; -,r c L | j | j | S )aw Tries 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 r
FindMessageTypeByName)r symbols r GetSymbolzSymbolDatabase.GetSymbol s! =88@@ AAr c fdi }|D ]l}| j | }|j D ]6} | D ](} | j | ||j < # t $ r Y %w xY w7m|S )a Gets 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.
c 3 L K | V | j D ]} | D ]}|V dS )zCWalk a message Descriptor and recursively yields all message names.N) nested_types)r" msg_desc nested_desc_GetAllMessagess r r> z3SymbolDatabase.GetMessages.._GetAllMessages sY jjj' (*?844 K
r )r
FindFileByNamemessage_types_by_namevaluesr full_nameKeyError)r r result file_name file_descr< r" r> s @r r zSymbolDatabase.GetMessages s & F )**955i 5<<>> (#OH-- D %)]4%8F4> " " D Ms A&&
A3 2A3 )N)__name__
__module__ __qualname____doc__r r
r r r r# r r, r0 r4 r8 r r r r C s // (9 9 9 97 7 7 C C C G G G $3 3 3 : : :< < <B B B$$ $ $ $ $r r )r
c t S )z#Returns the default SymbolDatabase.)_DEFAULTrK r r DefaultrN s /r )
rJ r google.protobuf.internalr google.protobufr r r rN rM rK r r rQ s > : 7 7 7 7 7 7 + + + + + + + + + + + +Q Q Q Q Q Q Q Qh > 6 6 8 8 9 9 9 r