Spaces:
Paused
Paused
| # 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__ = '[email protected] (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)) | |