Use remote plugin to generate typing stubs

Author: Ronen Hilewicz

buf/buf.gen.yaml

@@ -6,7 +6,7 @@ plugins:
     out: ./src
   - plugin: buf.build/protocolbuffers/python
     out: ./src
-  - name: "mypy"
+  - plugin: buf.build/protocolbuffers/pyi
     out: "./src"
-  - name: "mypy_grpc"
+  - plugin: "buf.build/community/nipunn1313-mypy-grpc"
     out: "./src"

init_gen.py

@@ -90,7 +90,7 @@ def __init__(self):
         self._in_const: bool = False
 
     def visit_ClassDef(self, node: ast.ClassDef):
-        if self.is_public(node.name):
+        if self.is_public(node.name) and not node.name.endswith("AsyncStub"):
             self.symbols.append(node.name)
 
     def is_public(self, name: str) -> bool:

poetry.lock

@@ -30,14 +30,15 @@ packages = [
 
 [tool.poetry.dependencies]
 python = ">=3.8"
+grpcio = ">=1.58.0"
+typing-extensions = ">=3.10.0"
 
 [tool.poetry.dev-dependencies]
-mypy = "^1.6.1"
-mypy-protobuf = "^3.5.0"
 black = "^23.10.1"
 isort = "^5.10.1"
 
 [tool.mypy]
+explicit_package_bases = true
 show_error_codes = true
 check_untyped_defs = false
 disallow_any_decorated = false