1 files changed, 7 insertions, 7 deletions
diff --git a/python/facility_update.py b/python/facility_update.py
index 95d38064..88179e0b 100644
--- a/python/facility_update.py
+++ b/python/facility_update.py
@@ -2,9 +2,9 @@ import requests
 import os
 import os.path
 import datetime
-import common
+from common import root
 import csv
-import pdb
+from db import conn
 import psycopg2
 
 legal = 'serecap'
@@ -17,7 +17,7 @@ def convertToNone(v):
     return v if v else None
 
 date = str(datetime.date.today())
-filename = os.path.join(common.root, "data", "Facility files", "facility_{0}.csv".format(date))
+filename = os.path.join(root, "data", "Facility files", "facility_{0}.csv".format(date))
 with open( filename, "wb") as fh:
      fh.write(r.content)
 
@@ -51,11 +51,11 @@ with open( filename, "wb") as fh:
 
 with open( filename, "r") as fh:
     reader = csv.reader(fh)
-    reader.next()
-    c = common.conn.cursor()
+    next(reader)
+    c = conn.cursor()
     sqlstring = "INSERT INTO markit_facility VALUES( {0} )".format( ",".join(["%s"] * 13))
     for line in reader:
         newline = [convertToNone(v) for v in line]
         c.execute(sqlstring, newline)
-common.conn.commit()
-common.conn.close()
+conn.commit()
+conn.close()