pub/python-betterproto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix services using non-pythonified field names

Browse Source
This commit is contained in:
andrei 2020-05-04 15:12:57 -07:00 committed by Nat Noordanus
parent 4a2baf3f0a
commit 983e0895a2
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
betterproto/templates/template.py
View File

@ -63,7 +63,7 @@ class {{ service.py_name }}Stub(betterproto.ServiceStub):
{% endif %} {% endif %}
{% for method in service.methods %} {% for method in service.methods %}
async def {{ method.py_name }}(self{% if method.input_message and method.input_message.properties %}, *, {% for field in method.input_message.properties %}{{ field.name }}: {% if field.zero == "None" and not field.type.startswith("Optional[") %}Optional[{{ field.type }}]{% else %}{{ field.type }}{% endif %} = {{ field.zero }}{% if not loop.last %}, {% endif %}{% endfor %}{% endif %}) -> {% if method.server_streaming %}AsyncGenerator[{{ method.output }}, None]{% else %}{{ method.output }}{% endif %}: async def {{ method.py_name }}(self{% if method.input_message and method.input_message.properties %}, *, {% for field in method.input_message.properties %}{{ field.py_name }}: {% if field.zero == "None" and not field.type.startswith("Optional[") %}Optional[{{ field.type }}]{% else %}{{ field.type }}{% endif %} = {{ field.zero }}{% if not loop.last %}, {% endif %}{% endfor %}{% endif %}) -> {% if method.server_streaming %}AsyncGenerator[{{ method.output }}, None]{% else %}{{ method.output }}{% endif %}:
{% if method.comment %} {% if method.comment %}
{{ method.comment }} {{ method.comment }}
@ -71,10 +71,10 @@ class {{ service.py_name }}Stub(betterproto.ServiceStub):
request = {{ method.input }}() request = {{ method.input }}()
{% for field in method.input_message.properties %} {% for field in method.input_message.properties %}
{% if field.field_type == 'message' %} {% if field.field_type == 'message' %}
if {{ field.name }} is not None: if {{ field.py_name }} is not None:
request.{{ field.name }} = {{ field.name }} request.{{ field.py_name }} = {{ field.py_name }}
{% else %} {% else %}
request.{{ field.name }} = {{ field.name }} request.{{ field.py_name }} = {{ field.py_name }}
{% endif %} {% endif %}
{% endfor %} {% endfor %}

4
betterproto/tests/inputs/keywords/keywords.proto
View File

@ -5,3 +5,7 @@ message Test {
int32 with = 2; int32 with = 2;
int32 as = 3; int32 as = 3;
} }
service TestService {
rpc GetTest(Test) returns (Test) {}
}