Changes config params.

1 files changed, 18 insertions, 16 deletions

diff --git a/crawler/config.go b/crawler/config.go
index f05ad18..af02a6c 100644
--- a/crawler/config.go
+++ b/crawler/config.go
@@ -6,14 +6,16 @@ import (
 )
 
 type Config struct {
-	DBDriver   string
-	DBDBName   string
-	DBHost     string
-	DBPort     string
-	DBUser     string
-	DBPassword string
-	DBOptions  string
-	DBPath     string // for sqlite
+	DBDriver     string
+	DBDBName     string
+	DBHost       string
+	DBPort       string
+	DBUser       string
+	DBPassword   string
+	DBOptions    string
+	DBPath       string // for sqlite
+	Polr_URL     string
+	Polr_API_Key string
 
 	Debug bool
 }
@@ -64,13 +66,13 @@ func (c *Config) parseConfig(configFile string) {
 
 // Actually sets the config struct
 func (c *Config) setsConfig() {
-	c.DBDriver = viper.GetString("DBDriver")
-	c.DBHost = viper.GetString("DBHost")
-	c.DBPort = viper.GetString("DBPort")
-	c.DBUser = viper.GetString("DBUser")
-	c.DBPassword = viper.GetString("DBPassword")
-	c.DBDBName = viper.GetString("DBName")
-	c.DBOptions = viper.GetString("DBOptions")
-	c.DBPath = viper.GetString("DBPath")
+	c.DBDriver = viper.GetString("DB_Driver")
+	c.DBHost = viper.GetString("DB_Host")
+	c.DBPort = viper.GetString("DB_Port")
+	c.DBUser = viper.GetString("DB_User")
+	c.DBPassword = viper.GetString("DB_Password")
+	c.DBDBName = viper.GetString("DB_Name")
+	c.DBOptions = viper.GetString("DB_Options")
+	c.DBPath = viper.GetString("DB_Path")
 	c.Debug = viper.GetBool("Debug")
 }