diff --git a/object/ormer.go b/object/ormer.go index 4fc0d54d..11a8aa17 100644 --- a/object/ormer.go +++ b/object/ormer.go @@ -38,10 +38,12 @@ import ( var ( ormer *Ormer = nil createDatabase = true + configPath = "conf/app.conf" ) func InitFlag() { createDatabase = getCreateDatabaseFlag() + configPath = getConfigFlag() } func getCreateDatabaseFlag() bool { @@ -50,6 +52,12 @@ func getCreateDatabaseFlag() bool { return *res } +func getConfigFlag() string { + res := flag.String("config", "conf/app.conf", "set it to \"/your/path/app.conf\" if your config file is not in: \"/conf/app.conf\"") + flag.Parse() + return *res +} + func InitConfig() { err := beego.LoadAppConfig("ini", "../conf/app.conf") if err != nil { @@ -64,7 +72,7 @@ func InitConfig() { func InitAdapter() { if conf.GetConfigString("driverName") == "" { - if !util.FileExist("conf/app.conf") { + if !util.FileExist(configPath) { dir, err := os.Getwd() if err != nil { panic(err)