29 lines
891 B
Go
29 lines
891 B
Go
package generator
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// GenerateRepository generates the repository class
|
|
func GenerateRepository(ctx *Context) (string, error) {
|
|
var b strings.Builder
|
|
|
|
// Imports
|
|
b.WriteString("from televend_core.databases.base_repository import CRUDRepository\n")
|
|
b.WriteString(fmt.Sprintf("from televend_core.databases.cloud_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.cloud_repositories.%s.model import %s\n",
|
|
ctx.ModuleName, ctx.EntityName))
|
|
b.WriteString("\n\n")
|
|
|
|
// Class definition
|
|
b.WriteString(fmt.Sprintf("class %sRepository(CRUDRepository[%s, %sFilter]):\n",
|
|
ctx.EntityName, ctx.EntityName, ctx.EntityName))
|
|
b.WriteString(fmt.Sprintf(" model_cls = %s\n", ctx.EntityName))
|
|
|
|
return b.String(), nil
|
|
}
|