Use correct grpc stub

* use grpc stub * fix grpc stub import problem * do not track .idea * delete .idea * Update .gitignore
RussellLuo · Nov 14, 2017 · e7bd735 · e7bd735
1 parent e1ae5a0
commit e7bd735
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,3 +7,4 @@ build
 *.egg
 *.egg-info
 .tox
+.idea
diff --git a/grpc_pytools/pythonic.py b/grpc_pytools/pythonic.py
@@ -44,7 +44,8 @@ def write_module_header(self):
             '# -*- coding: utf-8 -*-\n'
             '{import_enum}'
             '\nimport grpc'
-            '\n\n{import_pb2}'.format(
+            '\n\n{import_pb2}'
+            '\n{import_pb2}_grpc'.format(
                 import_enum='\nimport enum' if self.has_enum_types() else '',
                 import_pb2=import_pb2
             )
@@ -90,7 +91,7 @@ def write_stub_property(self, stub_class_name):
             '\n    @property\n'
             '    def stub(self):\n'
             '        channel = grpc.insecure_channel(self.target)\n'
-            '        return {pb2_name}.{stub_class_name}(channel)\n'.format(
+            '        return {pb2_name}_grpc.{stub_class_name}(channel)\n'.format(
                 pb2_name=self.pb2_name,
                 stub_class_name=stub_class_name
             )
-Original file line number
+Diff line change
@@ Expand Up / @@ -7,3 +7,4 @@ build @@
     *.egg
     *.egg-info
     .tox
+    .idea