Redefine method definition for sync stubs

Author: AdrienVannson

src/betterproto2_compiler/templates/service_stub.py.j2

@@ -10,24 +10,7 @@ class {% block class_name %}{% endblock %}({% block inherit_from %}{% endblock %
     {% block class_content %}{% endblock %}
 
     {% for method in service.methods %}
-    async def {{ method.py_name }}(self
-        {%- if not method.client_streaming -%}
-            , message:
-                {%- if method.is_input_msg_empty -%}
-                    "{{ method.py_input_message_type }} | None" = None
-                {%- else -%}
-                    "{{ method.py_input_message_type }}"
-                {%- endif -%}
-        {%- else -%}
-            {# Client streaming: need a request iterator instead #}
-            , messages: "AsyncIterable[{{ method.py_input_message_type }}] | Iterable[{{ method.py_input_message_type }}]"
-        {%- endif -%}
-            ,
-            *
-            , timeout: "float | None" = None
-            , deadline: "Deadline | None" = None
-            , metadata: "MetadataLike | None" = None
-            ) -> "{% if method.server_streaming %}AsyncIterator[{{ method.py_output_message_type }}]{% else %}{{ method.py_output_message_type }}{% endif %}":
+    {% block method_definition scoped required %}{% endblock %}
         {% if method.comment %}
         """
         {{ method.comment | indent(8) }}

src/betterproto2_compiler/templates/service_stub_async.py.j2

@@ -3,3 +3,25 @@
 {# Class definition #}
 {% block class_name %}{{ service.py_name }}Stub{% endblock %}
 {% block inherit_from %}betterproto2.ServiceStub{% endblock %}
+
+{# Methods definition #}
+{% block method_definition %}
+    async def {{ method.py_name }}(self
+        {%- if not method.client_streaming -%}
+            , message:
+                {%- if method.is_input_msg_empty -%}
+                    "{{ method.py_input_message_type }} | None" = None
+                {%- else -%}
+                    "{{ method.py_input_message_type }}"
+                {%- endif -%}
+        {%- else -%}
+            {# Client streaming: need a request iterator instead #}
+            , messages: "AsyncIterable[{{ method.py_input_message_type }}] | Iterable[{{ method.py_input_message_type }}]"
+        {%- endif -%}
+            ,
+            *
+            , timeout: "float | None" = None
+            , deadline: "Deadline | None" = None
+            , metadata: "MetadataLike | None" = None
+            ) -> "{% if method.server_streaming %}AsyncIterator[{{ method.py_output_message_type }}]{% else %}{{ method.py_output_message_type }}{% endif %}":
+{% endblock %}

src/betterproto2_compiler/templates/service_stub_sync.py.j2

@@ -8,3 +8,25 @@
     def __init__(self, channel: grpc.Channel):
         self._channel = channel
 {% endblock %}
+
+{# Methods definition #}
+{% block method_definition %}
+    def {{ method.py_name }}(self
+        {%- if not method.client_streaming -%}
+            , message:
+                {%- if method.is_input_msg_empty -%}
+                    "{{ method.py_input_message_type }} | None" = None
+                {%- else -%}
+                    "{{ method.py_input_message_type }}"
+                {%- endif -%}
+        {%- else -%}
+            {# Client streaming: need a request iterator instead #}
+            , messages: "Iterable[{{ method.py_input_message_type }}]"
+        {%- endif -%}
+            ,
+            *
+            , timeout: "float | None" = None
+            , deadline: "Deadline | None" = None
+            , metadata: "MetadataLike | None" = None
+            ) -> "{% if method.server_streaming %}Iterator[{{ method.py_output_message_type }}]{% else %}{{ method.py_output_message_type }}{% endif %}":
+{% endblock %}