First working version
This commit is contained in:
34
internal/generator/manager.go
Normal file
34
internal/generator/manager.go
Normal file
@ -0,0 +1,34 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user