130 lines
5 KiB
Python
130 lines
5 KiB
Python
# Protocol Buffers - Google's data interchange format
|
|
# Copyright 2008 Google Inc. All rights reserved.
|
|
# 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
|
|
# met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above
|
|
# copyright notice, this list of conditions and the following disclaimer
|
|
# in the documentation and/or other materials provided with the
|
|
# distribution.
|
|
# * Neither the name of Google Inc. nor the names of its
|
|
# contributors may be used to endorse or promote products derived from
|
|
# this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
"""Builds descriptors, message classes and services for generated _pb2.py.
|
|
|
|
This file is only called in python generated _pb2.py files. It builds
|
|
descriptors, message classes and services that users can directly use
|
|
in generated code.
|
|
"""
|
|
|
|
__author__ = 'jieluo@google.com (Jie Luo)'
|
|
|
|
from protobuf.internal import enum_type_wrapper
|
|
from protobuf import message as _message
|
|
from protobuf import reflection as _reflection
|
|
from protobuf import symbol_database as _symbol_database
|
|
|
|
_sym_db = _symbol_database.Default()
|
|
|
|
|
|
def BuildMessageAndEnumDescriptors(file_des, module):
|
|
"""Builds message and enum descriptors.
|
|
|
|
Args:
|
|
file_des: FileDescriptor of the .proto file
|
|
module: Generated _pb2 module
|
|
"""
|
|
|
|
def BuildNestedDescriptors(msg_des, prefix):
|
|
for (name, nested_msg) in msg_des.nested_types_by_name.items():
|
|
module_name = prefix + name.upper()
|
|
module[module_name] = nested_msg
|
|
BuildNestedDescriptors(nested_msg, module_name + '_')
|
|
for enum_des in msg_des.enum_types:
|
|
module[prefix + enum_des.name.upper()] = enum_des
|
|
|
|
for (name, msg_des) in file_des.message_types_by_name.items():
|
|
module_name = '_' + name.upper()
|
|
module[module_name] = msg_des
|
|
BuildNestedDescriptors(msg_des, module_name + '_')
|
|
|
|
|
|
def BuildTopDescriptorsAndMessages(file_des, module_name, module):
|
|
"""Builds top level descriptors and message classes.
|
|
|
|
Args:
|
|
file_des: FileDescriptor of the .proto file
|
|
module_name: str, the name of generated _pb2 module
|
|
module: Generated _pb2 module
|
|
"""
|
|
|
|
def BuildMessage(msg_des):
|
|
create_dict = {}
|
|
for (name, nested_msg) in msg_des.nested_types_by_name.items():
|
|
create_dict[name] = BuildMessage(nested_msg)
|
|
create_dict['DESCRIPTOR'] = msg_des
|
|
create_dict['__module__'] = module_name
|
|
message_class = _reflection.GeneratedProtocolMessageType(
|
|
msg_des.name, (_message.Message,), create_dict)
|
|
_sym_db.RegisterMessage(message_class)
|
|
return message_class
|
|
|
|
# top level enums
|
|
for (name, enum_des) in file_des.enum_types_by_name.items():
|
|
module['_' + name.upper()] = enum_des
|
|
module[name] = enum_type_wrapper.EnumTypeWrapper(enum_des)
|
|
for enum_value in enum_des.values:
|
|
module[enum_value.name] = enum_value.number
|
|
|
|
# top level extensions
|
|
for (name, extension_des) in file_des.extensions_by_name.items():
|
|
module[name.upper() + '_FIELD_NUMBER'] = extension_des.number
|
|
module[name] = extension_des
|
|
|
|
# services
|
|
for (name, service) in file_des.services_by_name.items():
|
|
module['_' + name.upper()] = service
|
|
|
|
# Build messages.
|
|
for (name, msg_des) in file_des.message_types_by_name.items():
|
|
module[name] = BuildMessage(msg_des)
|
|
|
|
|
|
def BuildServices(file_des, module_name, module):
|
|
"""Builds services classes and services stub class.
|
|
|
|
Args:
|
|
file_des: FileDescriptor of the .proto file
|
|
module_name: str, the name of generated _pb2 module
|
|
module: Generated _pb2 module
|
|
"""
|
|
# pylint: disable=g-import-not-at-top
|
|
from protobuf import service as _service
|
|
from protobuf import service_reflection
|
|
# pylint: enable=g-import-not-at-top
|
|
for (name, service) in file_des.services_by_name.items():
|
|
module[name] = service_reflection.GeneratedServiceType(
|
|
name, (_service.Service,),
|
|
dict(DESCRIPTOR=service, __module__=module_name))
|
|
stub_name = name + '_Stub'
|
|
module[stub_name] = service_reflection.GeneratedServiceStubType(
|
|
stub_name, (module[name],),
|
|
dict(DESCRIPTOR=service, __module__=module_name))
|