Added yield for server streaming, fixes danielgtaylor#412

Signed-off-by: Antonín Říha <[email protected]>
anriha · Aug 29, 2022 · 2365ef8 · 2365ef8
1 parent 591ec5e
commit 2365ef8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/betterproto/templates/template.py.j2 b/src/betterproto/templates/template.py.j2
@@ -179,6 +179,9 @@ class {{ service.py_name }}Base(ServiceBase):
 
         {% endif %}
         raise grpclib.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+        {% if method.server_streaming %}
+        yield {{ method.py_output_message_type }}()
+        {% endif %}
 
     {% endfor %}