add constructor typing

Signed-off-by: Aidan Jensen <aidandj.github@gmail.com>

Author: aidandj

mypy_protobuf/main.py

@@ -1022,6 +1022,18 @@ def write_grpc_services(
                         self._import("grpc.aio", "Channel"),
                     )
                     wl("")
+                elif self.grpc_type == GRPCType.SYNC:
+                    wl(
+                        "def __init__(self, channel: {}) -> None: ...",
+                        self._import("grpc", "Channel"),
+                    )
+                    wl("")
+                elif self.grpc_type == GRPCType.ASYNC:
+                    wl(
+                        "def __init__(self, channel: {}) -> None: ...",
+                        self._import("grpc.aio", "Channel"),
+                    )
+                    wl("")
 
                 self.write_grpc_stub_methods(service, scl)
 

test/async_only/test_async_only.py

@@ -24,6 +24,7 @@ async def UnaryUnary(
         return dummy_pb2.DummyReply(value=request.value[::-1])
 
 
-async def main():
+async def noop() -> None:
+    """Don't actually run anything; this is just for type-checking."""
     stub = dummy_pb2_grpc.DummyServiceStub(channel=grpc.aio.insecure_channel("localhost:50051"))
     await stub.UnaryUnary(dummy_pb2.DummyRequest(value="test"))

test/generated_async_only/testproto/grpc/dummy_pb2_grpc.pyi

@@ -42,6 +42,8 @@ _DummyServiceStreamStreamType: typing_extensions.TypeAlias = grpc.aio.StreamStre
 class DummyServiceStub:
     """DummyService"""
 
+    def __init__(self, channel: grpc.aio.Channel) -> None: ...
+
     UnaryUnary: _DummyServiceUnaryUnaryType
     """UnaryUnary"""
 
@@ -103,6 +105,8 @@ _DeprecatedServiceDeprecatedMethodNotDeprecatedRequestType: typing_extensions.Ty
 class DeprecatedServiceStub:
     """Marking the service as deprecated"""
 
+    def __init__(self, channel: grpc.aio.Channel) -> None: ...
+
     DeprecatedMethod: _DeprecatedServiceDeprecatedMethodType
     """DeprecatedMethod"""
 

test/generated_async_only/testproto/grpc/import_pb2_grpc.pyi

@@ -34,6 +34,8 @@ _SimpleServiceNoCommentType: typing_extensions.TypeAlias = grpc.aio.UnaryUnaryMu
 class SimpleServiceStub:
     """SimpleService"""
 
+    def __init__(self, channel: grpc.aio.Channel) -> None: ...
+
     UnaryUnary: _SimpleServiceUnaryUnaryType
     """UnaryUnary"""
 

test/generated_sync_only/testproto/grpc/dummy_pb2_grpc.pyi

@@ -42,6 +42,8 @@ _DummyServiceStreamStreamType: typing_extensions.TypeAlias = grpc.StreamStreamMu
 class DummyServiceStub:
     """DummyService"""
 
+    def __init__(self, channel: grpc.Channel) -> None: ...
+
     UnaryUnary: _DummyServiceUnaryUnaryType
     """UnaryUnary"""
 
@@ -103,6 +105,8 @@ _DeprecatedServiceDeprecatedMethodNotDeprecatedRequestType: typing_extensions.Ty
 class DeprecatedServiceStub:
     """Marking the service as deprecated"""
 
+    def __init__(self, channel: grpc.Channel) -> None: ...
+
     DeprecatedMethod: _DeprecatedServiceDeprecatedMethodType
     """DeprecatedMethod"""
 

test/generated_sync_only/testproto/grpc/import_pb2_grpc.pyi

@@ -34,6 +34,8 @@ _SimpleServiceNoCommentType: typing_extensions.TypeAlias = grpc.UnaryUnaryMultiC
 class SimpleServiceStub:
     """SimpleService"""
 
+    def __init__(self, channel: grpc.Channel) -> None: ...
+
     UnaryUnary: _SimpleServiceUnaryUnaryType
     """UnaryUnary"""
 

test/sync_only/test_sync_only.py

@@ -21,5 +21,7 @@ def UnaryUnary(
         return dummy_pb2.DummyReply(value=request.value[::-1])
 
 
-stub = dummy_pb2_grpc.DummyServiceStub(channel=grpc.insecure_channel("localhost:50051"))
-stub.UnaryUnary(dummy_pb2.DummyRequest(value="test"))
+def noop() -> None:
+    """Don't actually run anything; this is just for type-checking."""
+    stub = dummy_pb2_grpc.DummyServiceStub(channel=grpc.insecure_channel("localhost:50051"))
+    stub.UnaryUnary(dummy_pb2.DummyRequest(value="test"))