| 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..0c757264f440fdcb59e36f1cbfc596ffda52603b 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
|
| @@ -49,104 +49,23 @@ __author__ = 'robinson@google.com (Will Robinson)'
|
|
|
|
|
| from google.protobuf.internal import api_implementation
|
| -from google.protobuf import descriptor as descriptor_mod
|
| from google.protobuf import message
|
|
|
| -_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
|
| -
|
| -
|
| -class GeneratedProtocolMessageType(type):
|
| -
|
| - """Metaclass for protocol message classes created at runtime from Descriptors.
|
| -
|
| - We add implementations for all methods described in the Message class. We
|
| - also create properties to allow getting/setting all fields in the protocol
|
| - message. Finally, we create slots to prevent users from accidentally
|
| - "setting" nonexistent fields in the protocol message, which then wouldn't get
|
| - serialized / deserialized properly.
|
| -
|
| - The protocol compiler currently uses this metaclass to create protocol
|
| - message classes at runtime. Clients can also manually create their own
|
| - classes at runtime, as in this example:
|
| + from google.protobuf.internal import python_message as message_impl
|
|
|
| - mydescriptor = Descriptor(.....)
|
| - class MyProtoClass(Message):
|
| - __metaclass__ = GeneratedProtocolMessageType
|
| - DESCRIPTOR = mydescriptor
|
| - myproto_instance = MyProtoClass()
|
| - myproto.foo_field = 23
|
| - ...
|
| - """
|
| -
|
| - # Must be consistent with the protocol-compiler code in
|
| - # proto2/compiler/internal/generator.*.
|
| - _DESCRIPTOR_KEY = 'DESCRIPTOR'
|
| -
|
| - def __new__(cls, name, bases, dictionary):
|
| - """Custom allocation for runtime-generated class types.
|
| -
|
| - We override __new__ because this is apparently the only place
|
| - where we can meaningfully set __slots__ on the class we're creating(?).
|
| - (The interplay between metaclasses and slots is not very well-documented).
|
| -
|
| - Args:
|
| - name: Name of the class (ignored, but required by the
|
| - metaclass protocol).
|
| - bases: Base classes of the class we're constructing.
|
| - (Should be message.Message). We ignore this field, but
|
| - it's required by the metaclass protocol
|
| - dictionary: The class dictionary of the class we're
|
| - constructing. dictionary[_DESCRIPTOR_KEY] must contain
|
| - a Descriptor object describing this protocol message
|
| - type.
|
| -
|
| - Returns:
|
| - Newly-allocated class.
|
| - """
|
| - descriptor = dictionary[GeneratedProtocolMessageType._DESCRIPTOR_KEY]
|
| - bases = _NewMessage(bases, descriptor, dictionary)
|
| - 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):
|
| - """Here we perform the majority of our work on the class.
|
| - We add enum getters, an __init__ method, implementations
|
| - of all Message methods, and properties for all fields
|
| - in the protocol type.
|
| -
|
| - Args:
|
| - name: Name of the class (ignored, but required by the
|
| - metaclass protocol).
|
| - bases: Base classes of the class we're constructing.
|
| - (Should be message.Message). We ignore this field, but
|
| - it's required by the metaclass protocol
|
| - dictionary: The class dictionary of the class we're
|
| - constructing. dictionary[_DESCRIPTOR_KEY] must contain
|
| - a Descriptor object describing this protocol message
|
| - type.
|
| - """
|
| - descriptor = dictionary[GeneratedProtocolMessageType._DESCRIPTOR_KEY]
|
| - _InitMessage(descriptor, cls)
|
| - superclass = super(GeneratedProtocolMessageType, cls)
|
| - superclass.__init__(name, bases, dictionary)
|
| +# The type of all Message classes.
|
| +# Part of the public interface.
|
| +#
|
| +# Used by generated files, but clients can also use it at runtime:
|
| +# mydescriptor = pool.FindDescriptor(.....)
|
| +# class MyProtoClass(Message):
|
| +# __metaclass__ = GeneratedProtocolMessageType
|
| +# DESCRIPTOR = mydescriptor
|
| +GeneratedProtocolMessageType = message_impl.GeneratedProtocolMessageType
|
|
|
|
|
| def ParseMessage(descriptor, byte_str):
|
| @@ -159,11 +78,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)
|
|
|