Add flags to generate only sync or only async stubs

Author: WouldYouKindly

mypy_protobuf/main.py

@@ -135,6 +135,11 @@ MYPY_PROTOBUF_VENV=venv_$PY_VER_MYPY_PROTOBUF
     find proto -name "*.proto" -print0 | xargs -0 "$PROTOC" "${PROTOC_ARGS[@]}" --mypy_out=generate_concrete_servicer_stubs:test/generated-concrete
     find proto/testproto/grpc -name "*.proto" -print0 | xargs -0 "$PROTOC" "${PROTOC_ARGS[@]}" --mypy_grpc_out=generate_concrete_servicer_stubs:test/generated-concrete
 
+    # Generate with sync_only stubs for testing
+    find proto/testproto/grpc -name "*.proto" -print0 | xargs -0 "$PROTOC" "${PROTOC_ARGS[@]}" --mypy_grpc_out=only_sync:test/generated-sync-only
+
+    # Generate with async_only stubs for testing
+    find proto/testproto/grpc -name "*.proto" -print0 | xargs -0 "$PROTOC" "${PROTOC_ARGS[@]}" --mypy_grpc_out=only_async:test/generated-async-only
 
     if [[ -n $VALIDATE ]] && ! diff <(echo "$SHA_BEFORE") <(find test/generated -name "*.pyi" -print0 | xargs -0 sha1sum); then
         echo -e "${RED}Some .pyi files did not match. Please commit those files${NC}"
@@ -161,6 +166,14 @@ for PY_VER in $PY_VER_UNIT_TESTS; do
         CONCRETE_MODULES=( -m test.test_concrete )
         MYPYPATH=$MYPYPATH:test/generated-concrete mypy ${CUSTOM_TYPESHED_DIR_ARG:+"$CUSTOM_TYPESHED_DIR_ARG"} --python-executable="$UNIT_TESTS_VENV"/bin/python --python-version="$PY_VER_MYPY_TARGET" "${CONCRETE_MODULES[@]}"
 
+        # Run sync_only mypy
+        SYNC_ONLY_MODULES=( -m test.test_sync_only )
+        MYPYPATH=$MYPYPATH:test/generated-sync-only mypy ${CUSTOM_TYPESHED_DIR_ARG:+"$CUSTOM_TYPESHED_DIR_ARG"} --python-executable="$UNIT_TESTS_VENV"/bin/python --python-version="$PY_VER_MYPY_TARGET" "${SYNC_ONLY_MODULES[@]}"
+
+        # Run async_only mypy
+        ASYNC_ONLY_MODULES=( -m test.test_async_only )
+        MYPYPATH=$MYPYPATH:test/generated-async-only mypy ${CUSTOM_TYPESHED_DIR_ARG:+"$CUSTOM_TYPESHED_DIR_ARG"} --python-executable="$UNIT_TESTS_VENV"/bin/python --python-version="$PY_VER_MYPY_TARGET" "${ASYNC_ONLY_MODULES[@]}"
+
         export MYPYPATH=$MYPYPATH:test/generated
 
         # Run mypy
@@ -209,7 +222,7 @@ for PY_VER in $PY_VER_UNIT_TESTS; do
     (
         # Run unit tests.
         source "$UNIT_TESTS_VENV"/bin/activate
-        PYTHONPATH=test/generated py.test --ignore=test/generated -v
+        PYTHONPATH=test/generated py.test --ignore=test/generated --ignore=test/generated-sync-only --ignore=test/generated-async-only -v
     )
 done
 

run_test.sh

@@ -0,0 +1,66 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class lower(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    A_FIELD_NUMBER: builtins.int
+    a: builtins.int
+    def __init__(
+        self,
+        *,
+        a: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["a", b"a"]) -> None: ...
+
+Global___lower: typing_extensions.TypeAlias = lower
+
+@typing.final
+class Upper(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LOWER_FIELD_NUMBER: builtins.int
+    @property
+    def Lower(self) -> Global___lower: ...
+    def __init__(
+        self,
+        *,
+        Lower: Global___lower | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["Lower", b"Lower"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["Lower", b"Lower"]) -> None: ...
+
+Global___Upper: typing_extensions.TypeAlias = Upper
+
+@typing.final
+class lower2(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    UPPER_FIELD_NUMBER: builtins.int
+    @property
+    def upper(self) -> Global___Upper: ...
+    def __init__(
+        self,
+        *,
+        upper: Global___Upper | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["upper", b"upper"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["upper", b"upper"]) -> None: ...
+
+Global___lower2: typing_extensions.TypeAlias = lower2

test/generated-async-only/__init__.py

@@ -0,0 +1,82 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class Test(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    A_FIELD_NUMBER: builtins.int
+    B_FIELD_NUMBER: builtins.int
+    C_FIELD_NUMBER: builtins.int
+    D_FIELD_NUMBER: builtins.int
+    E_FIELD_NUMBER: builtins.int
+    F_FIELD_NUMBER: builtins.int
+    G_FIELD_NUMBER: builtins.int
+    H_FIELD_NUMBER: builtins.int
+    I_FIELD_NUMBER: builtins.int
+    J_FIELD_NUMBER: builtins.int
+    K_FIELD_NUMBER: builtins.int
+    a: builtins.str
+    """Ending with " """
+    b: builtins.str
+    """Ending with "" """
+    c: builtins.str
+    """Ending with \"\"\" """
+    d: builtins.str
+    """Ending with \\ """
+    e: builtins.str
+    """Containing bad escape: \\x"""
+    f: builtins.str
+    """Containing \"\"\"" quadruple"""
+    g: builtins.str
+    """Containing \"\"\""" quintuple"""
+    h: builtins.str
+    """Containing \"\"\"\"\"\" sextuple"""
+    i: builtins.str
+    """\"\"\" Multiple \"\"\" triples \"\"\" """
+    j: builtins.str
+    """"quotes" can be a problem in comments.
+    \"\"\"Triple quotes\"\"\" just as well
+    """
+    k: builtins.str
+    """\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"
+    "                                              "
+    " Super Duper comments with surrounding edges! "
+    "                                              "
+    "            Pay attention to me!!!!           "
+    "                                              "
+    \"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"\"
+    """
+    def __init__(
+        self,
+        *,
+        a: builtins.str = ...,
+        b: builtins.str = ...,
+        c: builtins.str = ...,
+        d: builtins.str = ...,
+        e: builtins.str = ...,
+        f: builtins.str = ...,
+        g: builtins.str = ...,
+        h: builtins.str = ...,
+        i: builtins.str = ...,
+        j: builtins.str = ...,
+        k: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["a", b"a", "b", b"b", "c", b"c", "d", b"d", "e", b"e", "f", b"f", "g", b"g", "h", b"h", "i", b"i", "j", b"j", "k", b"k"]) -> None: ...
+
+Global___Test: typing_extensions.TypeAlias = Test