3

���h�@sHdZd
dl
mZ
d
dl
mZ
Gdd�dej�Zeej�d�
Zdd�Zd	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@sHeZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dS)�SymbolDatabasez'A database of Python generated symbols.cCs|
j}|
|j
|<|j|�

|
S)
z�Registers 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.
    )Z
DESCRIPTOR�_classes�RegisterMessageDescriptor)�self�message�desc�r
�%/usr/lib/python3.6/symbol_database.py�RegisterMessageDs




zSymbolDatabase.RegisterMessagecCs|jj
|
�

d
S)z�Registers the given message descriptor in the local database.

    Args:
      message_descriptor: a descriptor.MessageDescriptor.
    N)�poolZ
AddDescriptor)rZmessage_descriptorr
r
rrUsz(SymbolDatabase.RegisterMessageDescriptorcCs|jj
|
�

|
S)
z�Registers the given enum descriptor in the local database.

    Args:
      enum_descriptor: a descriptor.EnumDescriptor.

    Returns:
      The provided descriptor.
    )r
ZAddEnumDescriptor)rZenum_descriptorr
r
r�RegisterEnumDescriptor]s	
z%SymbolDatabase.RegisterEnumDescriptorcCs|jj
|
�

d
S)z�Registers the given service descriptor in the local database.

    Args:
      service_descriptor: a descriptor.ServiceDescriptor.

    Returns:
      The provided descriptor.
    N)r
ZAddServiceDescriptor)rZservice_descriptorr
r
r�RegisterServiceDescriptoris	z(SymbolDatabase.RegisterServiceDescriptorcCs|jj
|
�

d
S)z�Registers the given file descriptor in the local database.

    Args:
      file_descriptor: a descriptor.FileDescriptor.

    Returns:
      The provided descriptor.
    N)r
ZAddFileDescriptor)rZfile_descriptorr
r
r�RegisterFileDescriptorts	z%SymbolDatabase.RegisterFileDescriptorcCs|j|j
j|
�
S)
ax
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: A str, a protocol buffer symbol.

    Returns:
      A Python class corresponding to the symbol.

    Raises:
      KeyError: if the symbol could not be found.
    )rr
ZFindMessageTypeByName)rZsymbolr
r
r�	GetSymbolszSymbolDatabase.GetSymbolcs��f
d
d��i}xl|
D]d}|jj
|�
}xRt|jj��
D]@}x:�|�
D].}y|j|||j<WqDtk
rp


YqDXqDWq6WqW|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.
    c
3s2|V
x&|jD]}
x�|
�
D]
}|V
qWqWd
S)zCWalk a message Descriptor and recursively yields all message names.N)
Znested_types)r	�msg_descZnested_desc)
�_GetAllMessagesr
rr�s


z3SymbolDatabase.GetMessages.<locals>._GetAllMessages)r
ZFindFileByName�listZmessage_types_by_name�valuesrZ	full_name�KeyError)r�files�result�	file_nameZ	file_descrr	r
)
rr�GetMessages�s








zSymbolDatabase.GetMessagesN)�__name__�
__module__�__qualname__�__doc__rrrrrrrr
r
r
rrAs
r)
r
c
CstS)
z#Returns the default SymbolDatabase.)
�_DEFAULTr
r
r
r�Default�sr N)rZgoogle.protobufrrZMessageFactoryrr rr
r
r
r�<module>:s

w