✨ support aio

1. add grpc.aio.Server to generated `Servicer_to_server` stub Signed-off-by: weiwee <[email protected]>
sagewe · Mar 30, 2021 · e8b81df · e8b81df
1 parent f8ccdf3
commit e8b81df
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mypy_protobuf/main.py b/mypy_protobuf/main.py
@@ -683,10 +683,12 @@ def write_grpc_services(self, services: Iterable[d.ServiceDescriptorProto]) -> N
                 self.write_grpc_methods(service)
             l("")
             l(
-                "def add_{}Servicer_to_server(servicer: {}Servicer, server: {}) -> None: ...",
+                "def add_{}Servicer_to_server(servicer: {}Servicer, server: {}[{}, {}]) -> None: ...",
                 service.name,
                 service.name,
+                self._import("typing", "Union"),
                 self._import("grpc", "Server"),
+                self._import("grpc.aio", "Server"),
             )
             l("")