Skip to content

Commit f160c08

Browse files
diorcetyYann Diorcet
authored andcommitted
Fix python-fastapi signature of parameters for method
1 parent 7e94c87 commit f160c08

3 files changed

Lines changed: 18 additions & 1 deletion

File tree

modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/AbstractPythonCodegen.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
import org.openapitools.codegen.CodegenOperation;
5252
import org.openapitools.codegen.CodegenParameter;
5353
import org.openapitools.codegen.CodegenProperty;
54+
import org.openapitools.codegen.CodegenResponse;
5455
import org.openapitools.codegen.DefaultCodegen;
5556
import org.openapitools.codegen.GeneratorLanguage;
5657
import org.openapitools.codegen.IJsonSchemaValidationProperties;
@@ -1263,6 +1264,20 @@ public OperationsMap postProcessOperationsWithModels(OperationsMap objs, List<Mo
12631264
);
12641265
}
12651266

1267+
// update typing import for operation responses type
1268+
for (CodegenResponse response : operation.responses) {
1269+
// Not interested in the result, only in the update of the imports
1270+
getPydanticType(
1271+
response.returnProperty,
1272+
modelImports,
1273+
exampleImports,
1274+
postponedModelImports,
1275+
postponedExampleImports,
1276+
moduleImports,
1277+
null
1278+
);
1279+
}
1280+
12661281
// add import for code samples
12671282
// import models one by one
12681283
if (!exampleImports.isEmpty()) {

modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/PythonFastAPIServerCodegen.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,8 @@ public String getTypeDeclaration(Schema p) {
234234

235235
@Override
236236
public OperationsMap postProcessOperationsWithModels(OperationsMap objs, List<ModelMap> allModels) {
237+
super.postProcessOperationsWithModels(objs, allModels);
238+
237239
OperationMap operations = objs.getOperations();
238240
// Set will make sure that no duplicated items are used.
239241
Set<String> securityImports = new HashSet<>();
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
{{#isString}}str{{/isString}}{{#isInteger}}int{{/isInteger}}{{#isLong}}int{{/isLong}}{{#isFloat}}float{{/isFloat}}{{#isDouble}}float{{/isDouble}}{{#isByteArray}}str{{/isByteArray}}{{#isBinary}}str{{/isBinary}}{{#isBoolean}}bool{{/isBoolean}}{{#isDate}}str{{/isDate}}{{#isDateTime}}str{{/isDateTime}}{{#isModel}}{{dataType}}{{/isModel}}{{#isContainer}}{{dataType}}{{/isContainer}}
1+
{{{vendorExtensions.x-py-typing}}}

0 commit comments

Comments
 (0)