Skip to content

Commit

Permalink
Avoid overwriting '__module__' of messages from shared modules. (#5364)
Browse files Browse the repository at this point in the history
Note that we *are* still overwriting it for messages from modules defined
within the current package.

See #4715.
  • Loading branch information
tseaver authored May 22, 2018
1 parent 6877b31 commit b709e77
Showing 1 changed file with 23 additions and 12 deletions.
35 changes: 23 additions & 12 deletions packages/google-cloud-os-login/google/cloud/oslogin_v1/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit b709e77

Please sign in to comment.