Skip to content

Commit dee661a

Browse files
authored
Bugfix: Implement __str__ for enum props to fix query string (#259)
* Implement __str__ for enum props to fix conversion to query string
1 parent 4556d9a commit dee661a

File tree

8 files changed

+24
-0
lines changed

8 files changed

+24
-0
lines changed

end_to_end_tests/golden-record-custom/custom_e2e/models/an_enum.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@
44
class AnEnum(str, Enum):
55
FIRST_VALUE = "FIRST_VALUE"
66
SECOND_VALUE = "SECOND_VALUE"
7+
8+
def __str__(self) -> str:
9+
return str(self.value)

end_to_end_tests/golden-record-custom/custom_e2e/models/an_int_enum.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,6 @@ class AnIntEnum(IntEnum):
55
VALUE_NEGATIVE_1 = -1
66
VALUE_1 = 1
77
VALUE_2 = 2
8+
9+
def __str__(self) -> str:
10+
return str(self.value)

end_to_end_tests/golden-record-custom/custom_e2e/models/different_enum.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@
44
class DifferentEnum(str, Enum):
55
DIFFERENT = "DIFFERENT"
66
OTHER = "OTHER"
7+
8+
def __str__(self) -> str:
9+
return str(self.value)

end_to_end_tests/golden-record/my_test_api_client/models/an_enum.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@
44
class AnEnum(str, Enum):
55
FIRST_VALUE = "FIRST_VALUE"
66
SECOND_VALUE = "SECOND_VALUE"
7+
8+
def __str__(self) -> str:
9+
return str(self.value)

end_to_end_tests/golden-record/my_test_api_client/models/an_int_enum.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,6 @@ class AnIntEnum(IntEnum):
55
VALUE_NEGATIVE_1 = -1
66
VALUE_1 = 1
77
VALUE_2 = 2
8+
9+
def __str__(self) -> str:
10+
return str(self.value)

end_to_end_tests/golden-record/my_test_api_client/models/different_enum.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@
44
class DifferentEnum(str, Enum):
55
DIFFERENT = "DIFFERENT"
66
OTHER = "OTHER"
7+
8+
def __str__(self) -> str:
9+
return str(self.value)

openapi_python_client/templates/int_enum.pyi

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@ class {{ enum.reference.class_name }}(IntEnum):
44
{% for key, value in enum.values.items() %}
55
{{ key }} = {{ value }}
66
{% endfor %}
7+
8+
def __str__(self) -> str:
9+
return str(self.value)

openapi_python_client/templates/str_enum.pyi

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@ class {{ enum.reference.class_name }}(str, Enum):
44
{% for key, value in enum.values.items() %}
55
{{ key }} = "{{ value }}"
66
{% endfor %}
7+
8+
def __str__(self) -> str:
9+
return str(self.value)

0 commit comments

Comments
 (0)