Use remote plugin to generate typing stubs

Author: Ronen Hilewicz

.github/workflows/ci.yaml

@@ -59,9 +59,6 @@ jobs:
       with:
         python-version: '3.10'
 
-    - name: Install Poetry
-      uses: snok/install-poetry@v1
-
     - name: Install dependencies
       run: |
           mkdir -p $HOME/.ssh
@@ -77,16 +74,14 @@ jobs:
 
           go run mage.go deps
 
-          poetry install --no-interaction
-
     - name: Clean generated code
       run: go run mage.go clean
 
     - name: Setup Buf Registry
       run: echo -e "machine buf.build\npassword ${ASERTO_BUF_TOKEN}" >> ~/.netrc
 
     - name: Generate
-      run: poetry run go run mage.go generate
+      run: go run mage.go generate
 
     - name: Commit changes
       if: github.event_name == 'workflow_dispatch'
@@ -161,7 +156,6 @@ jobs:
       - name: Install Poetry
         uses: snok/install-poetry@v1
 
-
       - name: Build and push the python package
         run: go run mage.go release
 

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