Allow disabling generation of grpclib stubs

Author: robin

src/betterproto/plugin/models.py

@@ -179,11 +179,14 @@ def comment(self) -> str:
             proto_file=self.proto_file, path=self.path, indent=self.comment_indent
         )
 
+@dataclass
+class Options:
+    grpc_kind: str = "grpclib"
 
 @dataclass
 class PluginRequestCompiler:
-
     plugin_request_obj: plugin.CodeGeneratorRequest
+    options: Options
     output_packages: Dict[str, "OutputTemplate"] = field(default_factory=dict)
 
     @property

src/betterproto/plugin/parser.py

@@ -29,6 +29,7 @@
     FieldCompiler,
     MapEntryCompiler,
     MessageCompiler,
+    Options,
     OneOfFieldCompiler,
     OutputTemplate,
     PluginRequestCompiler,
@@ -41,7 +42,6 @@
 if TYPE_CHECKING:
     from google.protobuf.descriptor import Descriptor
 
-
 def traverse(
     proto_file: FieldDescriptorProto,
 ) -> "itertools.chain[Tuple[Union[str, EnumDescriptorProto], List[int]]]":
@@ -68,13 +68,21 @@ def _traverse(
         _traverse([5], proto_file.enum_type), _traverse([4], proto_file.message_type)
     )
 
+def parse_options(plugin_options: List[str]) -> Options:
+    options = Options()
+    for option in plugin_options:
+        if option.startswith("grpc-kind="):
+            options.grpc_kind = option.split("=", 1)[1]
+    return options
 
 def generate_code(
     request: plugin.CodeGeneratorRequest, response: plugin.CodeGeneratorResponse
 ) -> None:
     plugin_options = request.parameter.split(",") if request.parameter else []
 
-    request_data = PluginRequestCompiler(plugin_request_obj=request)
+    options = parse_options(plugin_options)
+
+    request_data = PluginRequestCompiler(plugin_request_obj=request, options=options)
     # Gather output packages
     for proto_file in request.proto_file:
         if (

src/betterproto/templates/template.py.j2

@@ -15,8 +15,10 @@ from typing import {% for i in output_file.typing_imports|sort %}{{ i }}{% if no
 {% endif %}
 
 import betterproto
+{% if output_file.parent_request.options.grpc_kind == "grpclib" %}
 from betterproto.grpc.grpclib_server import ServiceBase
-{% if output_file.services %}
+{% endif %}
+{% if output_file.services and output_file.parent_request.options.grpc_kind == "grpclib" %}
 import grpclib
 {% endif %}
 
@@ -68,6 +70,9 @@ class {{ message.py_name }}(betterproto.Message):
 
 
 {% endfor %}
+
+{% if output_file.parent_request.options.grpc_kind == "grpclib" %}
+
 {% for service in output_file.services %}
 class {{ service.py_name }}Stub(betterproto.ServiceStub):
     {% if service.comment %}
@@ -237,6 +242,7 @@ class {{ service.py_name }}Base(ServiceBase):
         }
 
 {% endfor %}
+{% endif %}
 
 {% for i in output_file.imports|sort %}
 {{ i }}