From 016a566f91e9530f39ffc7a4b50ebb68348dad39 Mon Sep 17 00:00:00 2001 From: Adrien Vannson Date: Fri, 8 Nov 2024 15:23:25 +0100 Subject: [PATCH] Fix problem with __all__ --- src/betterproto/templates/header.py.j2 | 6 +++--- tests/test_all_definition.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/betterproto/templates/header.py.j2 b/src/betterproto/templates/header.py.j2 index b6d0a6c4..f3be1c6c 100644 --- a/src/betterproto/templates/header.py.j2 +++ b/src/betterproto/templates/header.py.j2 @@ -4,13 +4,13 @@ # This file has been @generated __all__ = ( - {%- for enum in output_file.enums -%} + {% for _, enum in output_file.enums|dictsort(by="key") %} "{{ enum.py_name }}", {%- endfor -%} - {%- for message in output_file.messages -%} + {% for _, message in output_file.messages|dictsort(by="key") %} "{{ message.py_name }}", {%- endfor -%} - {%- for service in output_file.services -%} + {% for _, service in output_file.services|dictsort(by="key") %} "{{ service.py_name }}Stub", "{{ service.py_name }}Base", {%- endfor -%} diff --git a/tests/test_all_definition.py b/tests/test_all_definition.py index 61abb5f3..ca0b03f5 100644 --- a/tests/test_all_definition.py +++ b/tests/test_all_definition.py @@ -16,4 +16,4 @@ def test_all_definition(): "TestStub", "TestBase", ) - assert enum.__all__ == ("Choice", "ArithmeticOperator", "Test") + assert enum.__all__ == ("ArithmeticOperator", "Choice", "Test")