3
SÉLj, ã @ sD d Z dZG dd„ deƒZG dd„ deƒZG dd„ deƒZdd „ Zd
S ) z*Provides a container for DescriptorProtos.z"matthewtoia@google.com (Matt Toia)c @ s e Zd ZdS )ÚErrorN)Ú__name__Ú
__module__Ú __qualname__© r r ú)/usr/lib/python3.6/descriptor_database.pyr $ s r c @ s e Zd ZdZdS )Ú,DescriptorDatabaseConflictingDefinitionErrorzGRaised when a proto is added with the same name & different descriptor.N)r r r Ú__doc__r r r r r ( s r c @ s0 e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
S ) ÚDescriptorDatabasezEA container accepting FileDescriptorProtos and maps DescriptorProtos.c C s i | _ i | _d S )N)Ú_file_desc_protos_by_fileÚ_file_desc_protos_by_symbol)Úselfr r r Ú__init__/ s zDescriptorDatabase.__init__c sâ ˆ j }|| jkr ˆ | j|< n| j| ˆ kr6td| ƒ‚ˆ j}x.ˆ jD ]$}| jj‡ fdd„t||ƒD ƒƒ qDW x$ˆ jD ]}ˆ | jdj ||j fƒ< qtW x$ˆ j
D ]}ˆ | jdj ||j fƒ< qšW x$ˆ j D ]}ˆ | jdj ||j fƒ< qÀW dS )aM Adds the FileDescriptorProto and its types to this database.
Args:
file_desc_proto: The FileDescriptorProto to add.
Raises:
DescriptorDatabaseConflictingDefinitionError: if an attempt is made to
add a proto with the same name but different definition than an
exisiting proto in the database.
z0%s already added, but with different descriptor.c 3 s | ]}|ˆ fV qd S )Nr )Ú.0Úname)Úfile_desc_protor r ú H s z)DescriptorDatabase.Add..Ú.N) r r
r ÚpackageZ message_typer ÚupdateÚ_ExtractSymbolsÚ enum_typeÚjoinÚ extensionÚservice)r r Z
proto_namer ÚmessageÚenumr r r )r r ÚAdd3 s
zDescriptorDatabase.Addc C s
| j | S )a¸ Finds the file descriptor proto by file name.
Typically the file name is a relative path ending to a .proto file. The
proto with the given name will have to have been added to this database
using the Add method or else an error will be raised.
Args:
name: The file name to find.
Returns:
The file descriptor proto matching the name.
Raises:
KeyError if no file by the given name was added.
)r
)r r r r r ÚFindFileByNameS s z!DescriptorDatabase.FindFileByNamec C s: y
| j | S tk
r4 |jdƒ\}}}| j | S X dS )a¢ Finds the file descriptor proto containing the specified symbol.
The symbol should be a fully qualified name including the file descriptor's
package and any containing messages. Some examples:
'some.package.name.Message'
'some.package.name.Message.NestedEnum'
'some.package.name.Message.some_field'
The file descriptor proto containing the specified symbol must be added to
this database using the Add method or else an error will be raised.
Args:
symbol: The fully qualified symbol name.
Returns:
The file descriptor proto containing the symbol.
Raises:
KeyError if no file contains the specified symbol.
r N)r ÚKeyErrorÚ
rpartition)r ÚsymbolZ top_levelÚ_r r r ÚFindFileContainingSymbolf s
z+DescriptorDatabase.FindFileContainingSymbolN)r r r r r
r r r" r r r r r , s
r c c sn |r|d | j n| j }|V x(| jD ] }xt||ƒD ]
}|V q6W q&W x | jD ]}dj||j fƒV qPW dS )zëPulls out all the symbols from a descriptor proto.
Args:
desc_proto: The proto to extract symbols from.
package: The package containing the descriptor type.
Yields:
The fully qualified name found in the descriptor.
r N)r Ú nested_typer r r )Z
desc_protor Z message_namer# r r r r r r ˆ s
r N)r Ú
__author__Ú Exceptionr r Úobjectr r r r r r Ú s
\