Skip to content

Commit e1d4a4a

Browse files
authored
fix: add enums to types/__init__.py (#695)
1 parent 2e38473 commit e1d4a4a

File tree

1 file changed

+15
-6
lines changed

1 file changed

+15
-6
lines changed
Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,22 @@
11
{% extends '_base.py.j2' %}
22

33
{% block content %}
4-
{% for p in api.protos.values() if p.file_to_generate and p.messages -%}
5-
from .{{p.module_name }} import ({% for m in p.messages.values() %}{{ m.name }}, {% endfor %})
6-
{% endfor %}
4+
{%- for proto in api.protos.values() if proto.file_to_generate and proto.messages %}
5+
from .{{proto.module_name }} import (
6+
{%- for message in proto.messages.values() %}
7+
{{message.name }}, {% endfor %}
8+
{%- for enum in proto.enums.values() %}
9+
{{ enum.name }}, {% endfor %}
10+
){% endfor %}
711

812
__all__ = (
9-
{%- for p in api.protos.values() if p.file_to_generate %}{% for m in p.messages.values() %}
10-
'{{ m.name }}',
11-
{%- endfor %}{% endfor %}
13+
{%- for proto in api.protos.values() if proto.file_to_generate %}
14+
{%- for message in proto.messages.values() %}
15+
'{{ message.name }}',
16+
{%- endfor -%}
17+
{%- for enum in proto.enums.values() %}
18+
'{{ enum.name }}',
19+
{%- endfor -%}
20+
{%- endfor %}
1221
)
1322
{% endblock %}

0 commit comments

Comments
 (0)