fix when a $ref is used in additionalProperties

GetStream · Sep 18, 2023 · 32317dc · 32317dc
1 parent 0bd06ea
commit 32317dc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/openapi-gen/openapi.go b/openapi-gen/openapi.go
@@ -389,5 +389,12 @@ func getReferencesFromTypes(schema *openapi3.Schema) []string {
 		}
 	}
 
+	if schema.AdditionalProperties.Schema != nil {
+		if schema.AdditionalProperties.Schema.Ref != "" {
+			refs = append(refs, schema.AdditionalProperties.Schema.Ref)
+		} else if schema.AdditionalProperties.Schema.Value != nil {
+			refs = append(refs, getReferencesFromTypes(schema.AdditionalProperties.Schema.Value)...)
+		}
+	}
 	return refs
 }
diff --git a/openapi-gen/templates/python/type.tmpl b/openapi-gen/templates/python/type.tmpl
@@ -20,6 +20,9 @@
         {{- if and (.Value.AdditionalProperties.Schema) (not (contains $typingImports "Dict"))}}
             {{- $typingImports = append $typingImports "Dict"}}
         {{- end}}
+        {{- if and (.Value.AdditionalProperties.Schema) (.Value.AdditionalProperties.Schema.Ref) }}
+        {{- $_ := $imports.Add .Value.AdditionalProperties.Schema.Ref }}
+        {{- end}}
     {{- end}}
 {{- end}}