
    hx                         d Z ddlZddlmZ ddlmZ ddlmZ  G d d          Z e ej                              Z	d	 ZdS )
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                   H    e Zd ZdZi ZddZd Zd Zd Zd Z	d Z
d	 Zd
 ZdS )SymbolDatabasez'A database of Python generated symbols.Nc                 :    |pt          j                    | _        dS )z!Initializes a new SymbolDatabase.N)r   DescriptorPoolpool)selfr
   s     c/var/www/html/e360mart/e360mart_env/lib/python3.11/site-packages/google/protobuf/symbol_database.py__init__zSymbolDatabase.__init__2   s    8688DIII    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_classesRegisterMessageDescriptor)r   messagedescs      r   RegisterMessagezSymbolDatabase.RegisterMessage6   s1     D!DM$""4(((Nr   c                 l    t          j                    dk    r| j                            |           dS dS )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.RegisterMessageDescriptorH   s=       H,,
i122222 -,r   c                 h    t          j                    dk    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.RegisterEnumDescriptorR   s4       H,,
i""?333r   c                 l    t          j                    dk    r| j                            |           dS dS )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%%&899999 -,r   c                 l    t          j                    dk    r| j                            |           dS dS )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.RegisterFileDescriptork   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   	GetSymbolzSymbolDatabase.GetSymbolu   s!      =88@@AAr   c                     fdi }|D ]l}| j                             |          }|j                                        D ]6} |          D ](}	 | j        |         ||j        <   # t          $ r Y %w xY w7m|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  dS )zCWalk a message Descriptor and recursively yields all message names.N)nested_types)r   msg_descnested_desc_GetAllMessagess      r   r0   z3SymbolDatabase.GetMessages.<locals>._GetAllMessages   sY      jjj'  (*?844 	 	K



	 r   )r
   FindFileByNamemessage_types_by_namevaluesr   	full_nameKeyError)r   filesresult	file_name	file_descr.   r   r0   s          @r   GetMessageszSymbolDatabase.GetMessages   s    &     F  	)**955i5<<>>  (#OH-- 	 	D%)]4%8F4>""   D	 Ms   A&&
A3	2A3	)N)__name__
__module____qualname____doc__r   r   r   r   r   r"   r&   r*   r:    r   r   r   r   ,   s        // (9 9 9 9  $3 3 3  	: 	: 	:< < <B B B$$ $ $ $ $r   r   )r
   c                      t           S )z#Returns the default SymbolDatabase.)_DEFAULTr?   r   r   DefaultrB      s    	/r   )
r>   warningsgoogle.protobuf.internalr   google.protobufr   r   r   rB   rA   r?   r   r   <module>rF      s    :  7 7 7 7 7 7 + + + + + + + + + + + +       D >6688999    r   