Instantiate a module with the generated code

tools/distrib/python/grpcio_tools/grpc_tools/BUILD

@@ -30,7 +30,10 @@ pyx_library(
 py_library(
     name = "grpc_tools",
     srcs = ["__init__.py", "protoc.py"],
-    deps = [":cyprotoc"],
+    deps = [
+      ":cyprotoc",
+      "@six_archive//:six",
+    ],
     # TODO: Think about whether we should include well-known protos.
     srcs_version = "PY2AND3",
     imports = [NON_BAZEL_ROOT],

tools/distrib/python/grpcio_tools/grpc_tools/_protoc_compiler.pyx

@@ -73,7 +73,7 @@ cdef _c_protoc_error_to_protoc_error(cProtocError c_protoc_error):
 cdef _c_protoc_warning_to_protoc_warning(cProtocWarning c_protoc_warning):
     return ProtocWarning(c_protoc_warning.filename, c_protoc_warning.line, c_protoc_warning.column, c_protoc_warning.message)
 
-def run_protoc_in_memory(bytes protobuf_path, bytes include_path):
+def get_protos(bytes protobuf_path, bytes include_path):
   cdef map[string, string] files
   cdef vector[cProtocError] errors
   # NOTE: Abbreviated name used to shadowing of the module name.

tools/distrib/python/grpcio_tools/grpc_tools/protoc.py

@@ -17,6 +17,10 @@
 import pkg_resources
 import sys
 
+# TODO: Figure out how to add this dependency to setuptools.
+import six
+import imp
+
 from grpc_tools import _protoc_compiler
 
 def main(command_arguments):
@@ -29,8 +33,16 @@ def main(command_arguments):
     command_arguments = [argument.encode() for argument in command_arguments]
     return _protoc_compiler.run_main(command_arguments)
 
-def run_protoc_in_memory(protobuf_path, include_path):
-  return _protoc_compiler.run_protoc_in_memory(protobuf_path.encode('ascii'), include_path.encode('ascii'))
+def get_protos(protobuf_path, include_path):
+  files = _protoc_compiler.get_protos(protobuf_path.encode('ascii'), include_path.encode('ascii'))
+  modules = []
+  # TODO: Ensure pointer equality between two invocations of this function.
+  for filename, code in six.iteritems(files):
+    module = imp.new_module(filename.decode('ascii'))
+    six.exec_(code, module.__dict__)
+    modules.append(module)
+  return tuple(modules)
+
 
 
 if __name__ == '__main__':

tools/distrib/python/grpcio_tools/grpc_tools/protoc_test.py

@@ -26,9 +26,10 @@ class ProtocTest(unittest.TestCase):
         original_dir = os.getcwd()
         # TODO: Completely get rid of this chdir stuff.
         os.chdir(os.path.join(original_dir, "tools/distrib/python/grpcio_tools/"))
-        files = protoc.run_protoc_in_memory("grpc_tools/simple.proto", "")
+        protos, = protoc.get_protos("grpc_tools/simple.proto", "")
         os.chdir(original_dir)
-        print("Files: {}".format(files))
+        # print("Protos: {}".format(vars(protos)))
+        print(protos.SimpleMessageRequest)
 
 
 if __name__ == '__main__':