Add host command line parameter in load_demo.py, use the same command… (

stanford-futuredata#229)

* Add host command line parameter in load_demo.py, use the same command line for load_demo2.py

* Change default password to postgres

tools/load_demo.py

@@ -7,15 +7,18 @@
 if(len(sys.argv) > 1):
     port = sys.argv[1]
 
-port = None
 user = None
 password = None
 if(len(sys.argv) > 2):
-    port = sys.argv[1]
     user = sys.argv[2]
     password = sys.argv[3]
 
-conn = psycopg2.connect("dbname='postgres' host='localhost'" +
+host = "localhost"
+if (len(sys.argv) > 3):
+    host = sys.argv[4]
+
+conn = psycopg2.connect("dbname='postgres'" +
+                         (" host='" + host + "'") +
                          (" port="+port if port else "") +
                          (" user="+user if user else "") +
                          (" password="+password if password else ""))

tools/load_demo2.py

@@ -12,6 +12,17 @@
 if(len(sys.argv) > 2):
     port = sys.argv[2]
 
+user = "postgres"
+password = "postgres"
+if(len(sys.argv) > 3):
+    user = sys.argv[3]
+    password = sys.argv[4]
+
+host = "localhost"
+if (len(sys.argv) > 4):
+    host = sys.argv[5]
+
+
 # TODO: These can be made configurable later
 NUM_ATTRIBUTES = 4
 NUM_METRICS = 5
@@ -24,7 +35,12 @@
   ["CA", "MA", "NY", "WY", "AR", "NV", "PA", "WA", "WI"]
 ]
 
-conn = psycopg2.connect("dbname='postgres' host='localhost'" + " port="+port if port else "" + " user=postgres password=mamikonyan")
+conn = psycopg2.connect("dbname='postgres'" +
+                         (" host='" + host + "'") +
+                         (" port="+port if port else "") +
+                         (" user="+user if user else "") +
+                         (" password="+password if password else ""))
+
 cur = conn.cursor()
 
 print "Creating table..."