LBRYCommunity/sqlboiler
1
0
Fork 0
Code Issues Pull requests 1 Packages Projects Releases Wiki Activity

Add debug flag

Browse source
This commit is contained in:
Patrick O'brien 2016-08-16 23:54:26 +10:00
parent 506de19d39
commit b8efd30e1b
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
main.go
View file

@ -63,6 +63,7 @@ func main() {
rootCmd.PersistentFlags().StringP("output", "o", "models", "The name of the folder to output to")
rootCmd.PersistentFlags().StringP("pkgname", "p", "models", "The name you wish to assign to your generated package")
rootCmd.PersistentFlags().StringSliceP("exclude", "x", nil, "Tables to be excluded from the generated package")
rootCmd.PersistentFlags().BoolP("debug", "d", false, "Debug mode prints stack traces on error")
viper.SetDefault("postgres.sslmode", "require")
viper.SetDefault("postgres.port", "5432")
@ -71,11 +72,13 @@ func main() {
if err := rootCmd.Execute(); err != nil {
if e, ok := err.(commandFailure); ok {
fmt.Printf("%s\n", string(e))
return
fmt.Printf("%v\n", string(e))
} else if !viper.GetBool("debug") {
fmt.Printf("%v\n", err)
} else {
fmt.Printf("%+v\n", err)
}
fmt.Printf("%+v\n", err)
os.Exit(1)
}
}