Add literals alongside enums

theorm · theorm · commit 9f93f9abe58e · 2024-03-29T13:48:22.000+01:00
diff --git a/openapi_python_client/templates/int_enum.py.jinja b/openapi_python_client/templates/int_enum.py.jinja
@@ -1,3 +1,4 @@
+from typing import Literal
 from enum import IntEnum
 
 class {{ enum.class_info.name }}(IntEnum):
@@ -7,3 +8,10 @@ class {{ enum.class_info.name }}(IntEnum):
 
     def __str__(self) -> str:
         return str(self.value)
+
+
+{{ enum.class_info.name }}Literal = Literal[
+    {% for _, value in enum.values.items() %}
+    {{ value }},
+    {% endfor %}
+]
diff --git a/openapi_python_client/templates/str_enum.py.jinja b/openapi_python_client/templates/str_enum.py.jinja
@@ -1,3 +1,4 @@
+from typing import Literal
 from enum import Enum
 
 class {{ enum.class_info.name }}(str, Enum):
@@ -7,3 +8,10 @@ class {{ enum.class_info.name }}(str, Enum):
 
     def __str__(self) -> str:
         return str(self.value)
+
+
+{{ enum.class_info.name }}Literal = Literal[
+    {% for _, value in enum.values|dictsort(true) %}
+    "{{ value }}",
+    {% endfor %}
+]
