3d001a2a1a
Cached data include, - lookup table between groups and fields of "oneof" fields - default value creator of each field - type hint of each field
97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
# sources: {{ ', '.join(description.files) }}
|
|
# plugin: python-betterproto
|
|
{% if description.datetime_imports %}
|
|
from datetime import {% for i in description.datetime_imports %}{{ i }}{% if not loop.last %}, {% endif %}{% endfor %}
|
|
|
|
{% endif%}
|
|
{% if description.typing_imports %}
|
|
from typing import {% for i in description.typing_imports %}{{ i }}{% if not loop.last %}, {% endif %}{% endfor %}
|
|
|
|
{% endif %}
|
|
|
|
import betterproto
|
|
{% if description.services %}
|
|
import grpclib
|
|
{% endif %}
|
|
|
|
{% for i in description.imports %}
|
|
{{ i }}
|
|
{% endfor %}
|
|
|
|
|
|
{% if description.enums %}{% for enum in description.enums %}
|
|
class {{ enum.py_name }}(betterproto.Enum):
|
|
{% if enum.comment %}
|
|
{{ enum.comment }}
|
|
|
|
{% endif %}
|
|
{% for entry in enum.entries %}
|
|
{% if entry.comment %}
|
|
{{ entry.comment }}
|
|
{% endif %}
|
|
{{ entry.name }} = {{ entry.value }}
|
|
{% endfor %}
|
|
|
|
|
|
{% endfor %}
|
|
{% endif %}
|
|
{% for message in description.messages %}
|
|
@betterproto.protoclass
|
|
class {{ message.py_name }}(betterproto.Message):
|
|
{% if message.comment %}
|
|
{{ message.comment }}
|
|
|
|
{% endif %}
|
|
{% for field in message.properties %}
|
|
{% if field.comment %}
|
|
{{ field.comment }}
|
|
{% endif %}
|
|
{{ field.py_name }}: {{ field.type }} = betterproto.{{ field.field_type }}_field({{ field.number }}{% if field.field_type == 'map'%}, betterproto.{{ field.map_types[0] }}, betterproto.{{ field.map_types[1] }}{% endif %}{% if field.one_of %}, group="{{ field.one_of }}"{% endif %}{% if field.field_wraps %}, wraps={{ field.field_wraps }}{% endif %})
|
|
{% endfor %}
|
|
{% if not message.properties %}
|
|
pass
|
|
{% endif %}
|
|
|
|
|
|
{% endfor %}
|
|
{% for service in description.services %}
|
|
class {{ service.py_name }}Stub(betterproto.ServiceStub):
|
|
{% if service.comment %}
|
|
{{ service.comment }}
|
|
|
|
{% endif %}
|
|
{% 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 %}:
|
|
{% if method.comment %}
|
|
{{ method.comment }}
|
|
|
|
{% endif %}
|
|
request = {{ method.input }}()
|
|
{% for field in method.input_message.properties %}
|
|
{% if field.field_type == 'message' %}
|
|
if {{ field.name }} is not None:
|
|
request.{{ field.name }} = {{ field.name }}
|
|
{% else %}
|
|
request.{{ field.name }} = {{ field.name }}
|
|
{% endif %}
|
|
{% endfor %}
|
|
|
|
{% if method.server_streaming %}
|
|
async for response in self._unary_stream(
|
|
"{{ method.route }}",
|
|
request,
|
|
{{ method.output }},
|
|
):
|
|
yield response
|
|
{% else %}
|
|
return await self._unary_unary(
|
|
"{{ method.route }}",
|
|
request,
|
|
{{ method.output }},
|
|
)
|
|
{% endif %}
|
|
|
|
{% endfor %}
|
|
{% endfor %}
|