Merge pull request #53 from ZhengYang/master

import package for no-pk models
This commit is contained in:
astaxie 2014-08-19 17:51:57 +08:00
commit 320fd1631e
1 changed files with 4 additions and 1 deletions

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)
// if table contains time field, import time.Time package
timePkg := ""
importTimePkg := ""
if tb.ImportTimePkg {
timePkg = "\"time\"\n"
importTimePkg = "import \"time\"\n"
}
fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1)
fileStr = strings.Replace(fileStr, "{{importTimePkg}}", importTimePkg, -1)
if _, err := f.WriteString(fileStr); err != nil {
ColorLog("[ERRO] Could not write model file to %s\n", fpath)
os.Exit(2)
@ -707,7 +710,7 @@ func getPackagePath(curpath string) (packpath string) {
const (
STRUCT_MODEL_TPL = `package models
{{importTimePkg}}
{{modelStruct}}
`