Skip to content

Commit 84ca046

Browse files
committed
Replaced enum special characters such that no compile error occurs anymore if there are whitespaces/operators etc in the enum.
1 parent 10402b1 commit 84ca046

File tree

3 files changed

+9
-2
lines changed

3 files changed

+9
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "openapi-python-generator"
3-
version = "0.3.4"
3+
version = "0.3.5"
44
description = "Openapi Python Generator"
55
authors = ["Marco Müllner <[email protected]>"]
66
license = "MIT"

src/openapi_python_generator/language_converters/python/model_generator.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import itertools
2+
import re
23
from typing import List
34
from typing import Optional
45

@@ -218,10 +219,16 @@ def generate_models(components: Components) -> List[Model]:
218219

219220
for name, schema_or_reference in components.schemas.items():
220221
if schema_or_reference.enum is not None:
222+
value_dict = schema_or_reference.dict()
223+
regex = re.compile(r"[\s\/=\*\+]+")
224+
value_dict["enum"] = [
225+
re.sub(regex, "_", i) if isinstance(i, str) else f"value_{i}"
226+
for i in value_dict["enum"]
227+
]
221228
m = Model(
222229
file_name=name,
223230
content=JINJA_ENV.get_template(ENUM_TEMPLATE).render(
224-
name=name, **schema_or_reference.dict()
231+
name=name, **value_dict
225232
),
226233
openapi_object=schema_or_reference,
227234
properties=[],

src/openapi_python_generator/language_converters/python/templates/enums.jinja2

Whitespace-only changes.

0 commit comments

Comments
 (0)