| Index: third_party/protobuf/python/google/protobuf/reflection.py
|
| diff --git a/third_party/protobuf/python/google/protobuf/reflection.py b/third_party/protobuf/python/google/protobuf/reflection.py
|
| index 9570fd50883f35b2bbfc30f6846fcd165fab08e2..82fca6619820307ef2abd0ab1e285809d5c111d0 100755
|
| --- a/third_party/protobuf/python/google/protobuf/reflection.py
|
| +++ b/third_party/protobuf/python/google/protobuf/reflection.py
|
| @@ -1,6 +1,6 @@
|
| # Protocol Buffers - Google's data interchange format
|
| # Copyright 2008 Google Inc. All rights reserved.
|
| -# http://code.google.com/p/protobuf/
|
| +# https://developers.google.com/protocol-buffers/
|
| #
|
| # Redistribution and use in source and binary forms, with or without
|
| # modification, are permitted provided that the following conditions are
|
| @@ -56,18 +56,12 @@ _FieldDescriptor = descriptor_mod.FieldDescriptor
|
|
|
|
|
| if api_implementation.Type() == 'cpp':
|
| - if api_implementation.Version() == 2:
|
| - from google.protobuf.internal.cpp import cpp_message
|
| - _NewMessage = cpp_message.NewMessage
|
| - _InitMessage = cpp_message.InitMessage
|
| - else:
|
| - from google.protobuf.internal import cpp_message
|
| - _NewMessage = cpp_message.NewMessage
|
| - _InitMessage = cpp_message.InitMessage
|
| + from google.protobuf.pyext import cpp_message as message_impl
|
| else:
|
| - from google.protobuf.internal import python_message
|
| - _NewMessage = python_message.NewMessage
|
| - _InitMessage = python_message.InitMessage
|
| + from google.protobuf.internal import python_message as message_impl
|
| +
|
| +_NewMessage = message_impl.NewMessage
|
| +_InitMessage = message_impl.InitMessage
|
|
|
|
|
| class GeneratedProtocolMessageType(type):
|
| @@ -91,6 +85,10 @@ class GeneratedProtocolMessageType(type):
|
| myproto_instance = MyProtoClass()
|
| myproto.foo_field = 23
|
| ...
|
| +
|
| + The above example will not work for nested types. If you wish to include them,
|
| + use reflection.MakeClass() instead of manually instantiating the class in
|
| + order to create the appropriate class structure.
|
| """
|
|
|
| # Must be consistent with the protocol-compiler code in
|
| @@ -123,7 +121,6 @@ class GeneratedProtocolMessageType(type):
|
| superclass = super(GeneratedProtocolMessageType, cls)
|
|
|
| new_class = superclass.__new__(cls, name, bases, dictionary)
|
| - setattr(descriptor, '_concrete_class', new_class)
|
| return new_class
|
|
|
| def __init__(cls, name, bases, dictionary):
|
| @@ -159,11 +156,43 @@ def ParseMessage(descriptor, byte_str):
|
| Returns:
|
| Newly created protobuf Message object.
|
| """
|
| + result_class = MakeClass(descriptor)
|
| + new_msg = result_class()
|
| + new_msg.ParseFromString(byte_str)
|
| + return new_msg
|
|
|
| - class _ResultClass(message.Message):
|
| +
|
| +def MakeClass(descriptor):
|
| + """Construct a class object for a protobuf described by descriptor.
|
| +
|
| + Composite descriptors are handled by defining the new class as a member of the
|
| + parent class, recursing as deep as necessary.
|
| + This is the dynamic equivalent to:
|
| +
|
| + class Parent(message.Message):
|
| __metaclass__ = GeneratedProtocolMessageType
|
| DESCRIPTOR = descriptor
|
| + class Child(message.Message):
|
| + __metaclass__ = GeneratedProtocolMessageType
|
| + DESCRIPTOR = descriptor.nested_types[0]
|
|
|
| - new_msg = _ResultClass()
|
| - new_msg.ParseFromString(byte_str)
|
| - return new_msg
|
| + Sample usage:
|
| + file_descriptor = descriptor_pb2.FileDescriptorProto()
|
| + file_descriptor.ParseFromString(proto2_string)
|
| + msg_descriptor = descriptor.MakeDescriptor(file_descriptor.message_type[0])
|
| + msg_class = reflection.MakeClass(msg_descriptor)
|
| + msg = msg_class()
|
| +
|
| + Args:
|
| + descriptor: A descriptor.Descriptor object describing the protobuf.
|
| + Returns:
|
| + The Message class object described by the descriptor.
|
| + """
|
| + attributes = {}
|
| + for name, nested_type in descriptor.nested_types_by_name.items():
|
| + attributes[name] = MakeClass(nested_type)
|
| +
|
| + attributes[GeneratedProtocolMessageType._DESCRIPTOR_KEY] = descriptor
|
| +
|
| + return GeneratedProtocolMessageType(str(descriptor.name), (message.Message,),
|
| + attributes)
|
|
|