googleapis
diff --git a/‎packages/google-cloud-os-login/google/cloud/oslogin_v1/types.py
Copy file name to clipboardExpand all lines: packages/google-cloud-os-login/google/cloud/oslogin_v1/types.py
+23-12Lines changed: 23 additions & 12 deletions b/‎packages/google-cloud-os-login/google/cloud/oslogin_v1/types.py
Copy file name to clipboardExpand all lines: packages/google-cloud-os-login/google/cloud/oslogin_v1/types.py
+23-12Lines changed: 23 additions & 12 deletions
@@ -15,24 +15,35 @@
 from __future__ import absolute_import
 import sys
 
-from google.api_core.protobuf_helpers import get_messages
-
 from google.api import http_pb2
-from google.cloud.oslogin_v1.proto import common_pb2
-from google.cloud.oslogin_v1.proto import oslogin_pb2
 from google.protobuf import descriptor_pb2
 from google.protobuf import empty_pb2
 from google.protobuf import field_mask_pb2
 
+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 (
-        http_pb2,
-        common_pb2,
-        oslogin_pb2,
-        descriptor_pb2,
-        empty_pb2,
-        field_mask_pb2,
-):
+
+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)