def test_get_messages(): answer = protobuf_helpers.get_messages(date_pb2) # Ensure that Date was exported properly. assert answer['Date'] is date_pb2.Date # Ensure that no non-Message objects were exported. for value in answer.values(): assert issubclass(value, message.Message)
from google.api_core.protobuf_helpers import get_messages from google.cloud.oslogin_v1.proto import common_pb2 from google.cloud.oslogin_v1.proto import oslogin_pb2 _shared_modules = [ http_pb2, descriptor_pb2, empty_pb2, field_mask_pb2, ] _local_modules = [ common_pb2, oslogin_pb2, ] names = [] for module in _shared_modules: for name, message in get_messages(module).items(): setattr(sys.modules[__name__], name, message) names.append(name) for module in _local_modules: for name, message in get_messages(module).items(): message.__module__ = 'google.cloud.oslogin_v1.types' setattr(sys.modules[__name__], name, message) names.append(name) __all__ = tuple(sorted(names))
import sys from google.api_core.protobuf_helpers import get_messages from google.cloud.videointelligence_v1p3beta1.proto import video_intelligence_pb2 from google.longrunning import operations_pb2 from google.protobuf import any_pb2 from google.protobuf import duration_pb2 from google.protobuf import timestamp_pb2 from google.rpc import status_pb2 _shared_modules = [operations_pb2, any_pb2, duration_pb2, timestamp_pb2, status_pb2] _local_modules = [video_intelligence_pb2] names = [] for module in _shared_modules: # pragma: NO COVER for name, message in get_messages(module).items(): setattr(sys.modules[__name__], name, message) names.append(name) for module in _local_modules: for name, message in get_messages(module).items(): message.__module__ = "google.cloud.videointelligence_v1p3beta1.types" setattr(sys.modules[__name__], name, message) names.append(name) __all__ = tuple(sorted(names))
'max_request_batch_size', 'max_request_batch_latency', 'max_lease_duration'], ) FlowControl.__new__.__defaults__ = ( psutil.virtual_memory().total * 0.2, # max_bytes: 20% of total RAM float('inf'), # max_messages: no limit 0.8, # resume_threshold: 80% 100, # max_requests: 100 100, # max_request_batch_size: 100 0.01, # max_request_batch_latency: 0.01s 2 * 60 * 60, # max_lease_duration: 2 hours. ) names = ['BatchSettings', 'FlowControl'] for name, message in get_messages(pubsub_pb2).items(): message.__module__ = 'google.cloud.pubsub_v1.types' setattr(sys.modules[__name__], name, message) names.append(name) for module in ( http_pb2, pubsub_pb2, iam_policy_pb2, policy_pb2, audit_data_pb2, descriptor_pb2, duration_pb2, empty_pb2, field_mask_pb2,
[ 'max_bytes', 'max_messages', 'resume_threshold', 'max_requests', 'max_request_batch_size', 'max_request_batch_latency' ], ) FlowControl.__new__.__defaults__ = ( psutil.virtual_memory().total * 0.2, # max_bytes: 20% of total RAM float('inf'), # max_messages: no limit 0.8, # resume_threshold: 80% 100, # max_requests: 100 100, # max_request_batch_size: 100 0.01, # max_request_batch_latency: 0.01s ) names = ['BatchSettings', 'FlowControl'] for name, message in get_messages(pubsub_pb2).items(): message.__module__ = 'google.cloud.pubsub_v1.types' setattr(sys.modules[__name__], name, message) names.append(name) for module in ( http_pb2, pubsub_pb2, iam_policy_pb2, policy_pb2, audit_data_pb2, descriptor_pb2, duration_pb2, empty_pb2, field_mask_pb2, timestamp_pb2,