models.py -> db_models.py

simo7 · simo7 · commit 4d2625e9b5ee · 2024-10-04T14:12:57.000+02:00
diff --git a/internal/gen.go b/internal/gen.go
@@ -998,8 +998,8 @@ func Generate(_ context.Context, req *plugin.GenerateRequest) (*plugin.GenerateR
 
 	output := map[string]string{}
 	result := pyprint.Print(buildModelsTree(&tctx, i), pyprint.Options{})
-	tctx.SourceName = "models.py"
-	output["models.py"] = string(result.Python)
+	tctx.SourceName = "db_models.py"
+	output["db_models.py"] = string(result.Python)
 
 	files := map[string]struct{}{}
 	for _, q := range queries {
diff --git a/internal/imports.go b/internal/imports.go
@@ -69,7 +69,7 @@ func queryValueUses(name string, qv QueryValue) bool {
 }
 
 func (i *importer) Imports(fileName string) []string {
-	if fileName == "models.py" {
+	if fileName == "db_models.py" {
 		return i.modelImports()
 	}
 	return i.queryImports(fileName)

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ func queryValueUses(name string, qv QueryValue) bool {`
`69`	`69`	`}`
`70`	`70`
`71`	`71`	`func (i *importer) Imports(fileName string) []string {`
`72`		`- if fileName == "models.py" {`
	`72`	`+ if fileName == "db_models.py" {`
`73`	`73`	`return i.modelImports()`
`74`	`74`	`}`
`75`	`75`	`return i.queryImports(fileName)`