Skip to content

Commit e48f987

Browse files
authored
Merge branch 'master' into add-model_template_dir-option
2 parents f0036cc + 7801309 commit e48f987

File tree

4 files changed

+108
-95
lines changed

4 files changed

+108
-95
lines changed

fastapi_code_generator/__main__.py

+1
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ def generate_code(
144144
data_type_manager_type=data_model_types.data_type_manager,
145145
dump_resolve_reference_action=data_model_types.dump_resolve_reference_action,
146146
custom_template_dir=model_template_dir,
147+
target_python_version=python_version,
147148
)
148149

149150
with chdir(output_dir):

fastapi_code_generator/parser.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -262,11 +262,12 @@ def get_parameter_type(
262262
path: List[str],
263263
) -> Optional[Argument]:
264264
parameters = self.resolve_object(parameters, ParameterObject)
265+
if parameters.name is None:
266+
raise RuntimeError("parameters.name is None") # pragma: no cover
265267
orig_name = parameters.name
268+
name = self.model_resolver.get_valid_field_name(parameters.name)
266269
if snake_case:
267-
name = stringcase.snakecase(parameters.name)
268-
else:
269-
name = parameters.name
270+
name = stringcase.snakecase(name)
270271

271272
schema: Optional[JsonSchemaObject] = None
272273
data_type: Optional[DataType] = None
@@ -307,8 +308,6 @@ def get_parameter_type(
307308
default = repr(schema.default) if schema.has_default else None
308309
self.imports_for_fastapi.append(field.imports)
309310
self.data_types.append(field.data_type)
310-
if field.name is None:
311-
raise RuntimeError("field.name is None") # pragma: no cover
312311
return Argument(
313312
name=UsefulStr(field.name),
314313
type_hint=UsefulStr(field.type_hint),

0 commit comments

Comments
 (0)