Sync stub: add constructor, remove parent class

Author: AdrienVannson

src/betterproto2_compiler/templates/header.py.j2

@@ -35,6 +35,7 @@ from dataclasses import dataclass
 import betterproto2
 {% if output_file.services %}
 from betterproto2.grpc.grpclib_server import ServiceBase
+import grpc
 import grpclib
 {% endif %}
 

src/betterproto2_compiler/templates/service_stub.py.j2

@@ -1,4 +1,4 @@
-class {% block class_name %}{% endblock %}(betterproto2.ServiceStub):
+class {% block class_name %}{% endblock %}({% block inherit_from %}{% endblock %}):
     {% if service.comment %}
     """
     {{ service.comment | indent(4) }}
@@ -7,6 +7,8 @@ class {% block class_name %}{% endblock %}(betterproto2.ServiceStub):
     pass
     {% endif %}
 
+    {% block class_content %}{% endblock %}
+
     {% for method in service.methods %}
     async def {{ method.py_name }}(self
         {%- if not method.client_streaming -%}

src/betterproto2_compiler/templates/service_stub_async.py.j2

@@ -1,3 +1,5 @@
 {% extends "service_stub.py.j2" %}
 
+{# Class definition #}
 {% block class_name %}{{ service.py_name }}Stub{% endblock %}
+{% block inherit_from %}betterproto2.ServiceStub{% endblock %}

src/betterproto2_compiler/templates/service_stub_sync.py.j2

@@ -1,3 +1,10 @@
 {% extends "service_stub.py.j2" %}
 
+{# Class definition #}
 {% block class_name %}{{ service.py_name }}SyncStub{% endblock %}
+
+{% block class_content %}
+    {# TODO move to parent class #}
+    def __init__(self, channel: grpc.Channel):
+        self._channel = channel
+{% endblock %}