Intis/entity-maker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Generate cloud instead televendcloud entities

Browse Source
This commit is contained in:
Eden Kirin
2025-12-15 10:37:28 +01:00
parent 0a4030c389
commit 5f5f1ad114
8 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/generator/factory.go
View File

@ -26,23 +26,23 @@ func GenerateFactory(ctx *Context) (string, error) {
}
for moduleName, entityName := range fkImports {
b.WriteString(fmt.Sprintf("from televend_core.databases.televend_repositories.%s.factory import (\n",
b.WriteString(fmt.Sprintf("from televend_core.databases.cloud_repositories.%s.factory import (\n",
moduleName))
b.WriteString(fmt.Sprintf(" %sFactory,\n", entityName))
b.WriteString(")\n")
}
b.WriteString(fmt.Sprintf("from televend_core.databases.televend_repositories.%s.model import (\n",
b.WriteString(fmt.Sprintf("from televend_core.databases.cloud_repositories.%s.model import (\n",
ctx.ModuleName))
b.WriteString(fmt.Sprintf(" %s,\n", ctx.EntityName))
b.WriteString(")\n")
b.WriteString("from televend_core.test_extras.factory_boy_utils import (\n")
b.WriteString(" CustomSelfAttribute,\n")
b.WriteString(" TelevendBaseFactory,\n")
b.WriteString(" CloudBaseFactory,\n")
b.WriteString(")\n\n\n")
// Class definition
b.WriteString(fmt.Sprintf("class %sFactory(TelevendBaseFactory):\n", ctx.EntityName))
b.WriteString(fmt.Sprintf("class %sFactory(CloudBaseFactory):\n", ctx.EntityName))
// Add boolean fields with defaults
for _, col := range ctx.TableInfo.Columns {