Skip to content

Commit 8a76df5

Browse files
author
Adam Gray
committed
fix output case for enums
1 parent 4b7ef35 commit 8a76df5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

openapi_python_client/openapi_parser/reference.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,10 @@ class Reference:
2121
def from_ref(ref: str) -> Reference:
2222
""" Get a Reference from the openapi #/schemas/blahblah string """
2323
ref_value = ref.split("/")[-1]
24-
class_name = stringcase.pascalcase(ref_value)
24+
# ugly hack to avoid stringcase ugly pascalcase output when ref_value isn't snake case
25+
class_name = stringcase.pascalcase(ref_value.replace(" ", ""))
2526

2627
if class_name in class_overrides:
2728
return class_overrides[class_name]
2829

29-
return Reference(class_name=class_name, module_name=stringcase.snakecase(ref_value),)
30+
return Reference(class_name=class_name, module_name=stringcase.snakecase(class_name),)

0 commit comments

Comments
 (0)