| # 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 google.protobuf.internal import enum_type_wrapper |
| from google.protobuf import message as _message |
| from google.protobuf import reflection as _reflection |
| from google.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 google.protobuf import service as _service |
| from google.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)) |