lbsadmin/bee
1
0
Fork 0
mirror of https://github.com/beego/bee.git synced 2024-11-22 10:10:53 +00:00
Code Issues Releases Activity

import package for no-pk models

Browse Source
This commit is contained in:
ZhengYang 2014-08-19 17:43:08 +08:00
parent c487e9d6f7
commit 82b7ab3e31
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
g_appcode.go
View File

@ -487,10 +487,13 @@ func writeModelFiles(tables []*Table, mPath string, selectedTables map[string]bo
fileStr = strings.Replace(fileStr, "{{modelName}}", camelCase(tb.Name), -1) fileStr = strings.Replace(fileStr, "{{modelName}}", camelCase(tb.Name), -1)
// if table contains time field, import time.Time package // if table contains time field, import time.Time package
timePkg := "" timePkg := ""
importTimePkg := ""
if tb.ImportTimePkg { if tb.ImportTimePkg {
timePkg = "\"time\"\n" timePkg = "\"time\"\n"
importTimePkg = "import \"time\"\n"
} }
fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1) fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1)
fileStr = strings.Replace(fileStr, "{{importTimePkg}}", importTimePkg, -1)
if _, err := f.WriteString(fileStr); err != nil { if _, err := f.WriteString(fileStr); err != nil {
ColorLog("[ERRO] Could not write model file to %s\n", fpath) ColorLog("[ERRO] Could not write model file to %s\n", fpath)
os.Exit(2) os.Exit(2)
@ -707,6 +710,7 @@ func getPackagePath(curpath string) (packpath string) {
const ( const (
STRUCT_MODEL_TPL = `package models STRUCT_MODEL_TPL = `package models
{{importTimePkg}}
{{modelStruct}} {{modelStruct}}
` `