Changed to use .yml file

supportdiva · supportdiva · commit 9d0232d6ffd0 · 2016-11-03T14:35:08.000-07:00
diff --git a/.DS_Store b/.DS_Store
diff --git a/README.md b/README.md
@@ -5,7 +5,7 @@ This script allows users to download a CSV of raw SQL from queries in a report.
 # Steps to use this script:
 
 1.  In Mode, generate API token (under Settings -> Your Name -> API Tokens).
-2.  Add the token and password values to the python.properties file.
+2.  Add the token and password values to the mode.yml file.
 3.  Run this script using: 
     `python demo.py -org={{organization_username}} -reporttoken={{report_token}}`
 
diff --git a/demo.py b/demo.py
@@ -1,14 +1,14 @@
 
-import json, requests, datetime, ConfigParser, argparse, sys, csv
+import json, requests, datetime, ConfigParser, argparse, sys, csv, yaml
 from requests.auth import HTTPBasicAuth
 
 
 def get_auth(whichAuth):
-	file = 'python.properties'
-	config = ConfigParser.RawConfigParser()
-	config.read(file)
-	token = config.get('ModeSection', 'token')
-	password = config.get('ModeSection', 'password')
+	with open ('mode.yml', 'r') as f:
+		mode = yaml.load(f) 
+
+	token = mode["mode"]["token"]
+	password = mode["mode"]["password"]
 	auth = (token, password)
 	return auth
 
diff --git a/mode.yml b/mode.yml
@@ -0,0 +1,3 @@
+mode:
+  token: 
+  password: 
diff --git a/python.properties b/python.properties
diff --git a/sql.csv b/sql.csv
@@ -0,0 +1,12 @@
+Query Token,SQL Query
+d832cd041c7a,"SELECT
+  *
+FROM
+  benn.nfl_touchdowns "
+790de3c8b5ad,"SELECT
+  team,
+  COUNT(touchdown_id) AS total_touchdowns
+FROM
+  benn.nfl_touchdowns 
+GROUP BY 1
+ORDER BY 2 DESC"
