Skip to content

Commit 03534ea

Browse files
committed
make sure only parent is imported
1 parent 53394a0 commit 03534ea

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/main/java/io/swagger/codegen/v3/generators/python/PythonClientCodegen.java

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
import java.util.List;
3030
import java.util.Map;
3131
import java.util.regex.Pattern;
32+
import java.util.stream.Collectors;
3233

3334
import static io.swagger.codegen.v3.generators.handlebars.ExtensionHelper.getBooleanValue;
3435

@@ -239,6 +240,17 @@ public void processOpts() {
239240

240241
}
241242

243+
@Override
244+
public CodegenModel fromModel(String name, Schema schema, Map<String, Schema> allDefinitions) {
245+
final CodegenModel codegenModel = super.fromModel(name, schema, allDefinitions);
246+
final List<String> imports = codegenModel.imports.stream()
247+
.filter(model -> model.equals(codegenModel.parent))
248+
.collect(Collectors.toList());
249+
codegenModel.imports.clear();
250+
codegenModel.imports.addAll(imports);
251+
return codegenModel;
252+
}
253+
242254
private static String dropDots(String str) {
243255
return str.replaceAll("\\.", "_");
244256
}

0 commit comments

Comments
 (0)