35 lines
1.2 KiB
Go
35 lines
1.2 KiB
Go
package generator
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// GenerateManager generates the manager class
|
|
func GenerateManager(ctx *Context) (string, error) {
|
|
var b strings.Builder
|
|
|
|
// Imports
|
|
b.WriteString("from televend_core.databases.base_manager import CRUDManager\n")
|
|
b.WriteString(fmt.Sprintf("from televend_core.databases.televend_repositories.%s.filter import (\n",
|
|
ctx.ModuleName))
|
|
b.WriteString(fmt.Sprintf(" %sFilter,\n", ctx.EntityName))
|
|
b.WriteString(")\n")
|
|
b.WriteString(fmt.Sprintf("from televend_core.databases.televend_repositories.%s.model import %s\n",
|
|
ctx.ModuleName, ctx.EntityName))
|
|
b.WriteString(fmt.Sprintf("from televend_core.databases.televend_repositories.%s.repository import (\n",
|
|
ctx.ModuleName))
|
|
b.WriteString(fmt.Sprintf(" %sRepository,\n", ctx.EntityName))
|
|
b.WriteString(")\n")
|
|
b.WriteString("\n\n")
|
|
|
|
// Class definition
|
|
b.WriteString(fmt.Sprintf("class %sManager(\n", ctx.EntityName))
|
|
b.WriteString(fmt.Sprintf(" CRUDManager[%s, %sFilter, %sRepository]\n",
|
|
ctx.EntityName, ctx.EntityName, ctx.EntityName))
|
|
b.WriteString("):\n")
|
|
b.WriteString(fmt.Sprintf(" repository_cls = %sRepository\n", ctx.EntityName))
|
|
|
|
return b.String(), nil
|
|
}
|