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.
BuildMessageAndEnumDescriptors(file_des, module)
Builds message and enum descriptors.
Parameters:
| Name | Type | Description | Default |
file_des | | FileDescriptor of the .proto file | required |
module | | | required |
Source code in client/ayon_hiero/vendor/google/protobuf/internal/builder.py
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67 | 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 + '_')
|
BuildServices(file_des, module_name, module)
Builds services classes and services stub class.
Parameters:
| Name | Type | Description | Default |
file_des | | FileDescriptor of the .proto file | required |
module_name | | str, the name of generated _pb2 module | required |
module | | | required |
Source code in client/ayon_hiero/vendor/google/protobuf/internal/builder.py
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130 | 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))
|
BuildTopDescriptorsAndMessages(file_des, module_name, module)
Builds top level descriptors and message classes.
Parameters:
| Name | Type | Description | Default |
file_des | | FileDescriptor of the .proto file | required |
module_name | | str, the name of generated _pb2 module | required |
module | | | required |
Source code in client/ayon_hiero/vendor/google/protobuf/internal/builder.py
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108 | 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)
|