ó X¾÷Xc@sadZddlmZddlmZdejfd„ƒYZedejƒƒZd„ZdS( sRA 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')() iÿÿÿÿ(tdescriptor_pool(tmessage_factorytSymbolDatabasecBs;eZdZd„Zd„Zd„Zd„Zd„ZRS(s'A database of Python generated symbols.cCs-|j}||j|j<|jj|ƒ|S(s÷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. (t DESCRIPTORt_classest full_nametpoolt AddDescriptor(tselftmessagetdesc((sA/tmp/pip-build-h1VYrz/protobuf/google/protobuf/symbol_database.pytRegisterMessageDs cCs|jj|ƒ|S(s«Registers the given enum descriptor in the local database. Args: enum_descriptor: a descriptor.EnumDescriptor. Returns: The provided descriptor. (RtAddEnumDescriptor(Rtenum_descriptor((sA/tmp/pip-build-h1VYrz/protobuf/google/protobuf/symbol_database.pytRegisterEnumDescriptorUs cCs|jj|ƒdS(s«Registers the given file descriptor in the local database. Args: file_descriptor: a descriptor.FileDescriptor. Returns: The provided descriptor. N(RtAddFileDescriptor(Rtfile_descriptor((sA/tmp/pip-build-h1VYrz/protobuf/google/protobuf/symbol_database.pytRegisterFileDescriptoras cCs |j|S(sxTries 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. (R(Rtsymbol((sA/tmp/pip-build-h1VYrz/protobuf/google/protobuf/symbol_database.pyt GetSymbollscs–‡fd†‰i}xz|D]r}|jj|ƒ}xW|jjƒD]F}x=ˆ|ƒD]/}y|j|||:s d