1 files changed, 9 insertions, 8 deletions
diff --git a/python/intex/load_indicative.py b/python/intex/load_indicative.py
index 0686ca92..145325df 100644
--- a/python/intex/load_indicative.py
+++ b/python/intex/load_indicative.py
@@ -5,7 +5,8 @@ from datetime import date
 import csv, sys, re
 import pdb
 from common import root, sanitize_float
-from db import conn
+from db import dbconn
+from contextlib import closing
 import logging
 
 logger = logging.getLogger(__name__)
@@ -184,10 +185,10 @@ if __name__=="__main__":
     cusip_files = [f for f in files if "TrInfo" in f]
     deal_files = [f for f in files if "TrInfo" not in f]
 
-    #first load deal data
-    for deal in deal_files:
-        upload_deal_data(conn, deal)
-    #then load tranche data
-    for cusip in cusip_files:
-        upload_cusip_data(conn, cusip)
-    conn.close()
+    with closing(dbconn('etdb')) as etdb:
+        #first load deal data
+        for deal in deal_files:
+            upload_deal_data(etdb, deal)
+            #then load tranche data
+        for cusip in cusip_files:
+            upload_cusip_data(etdb, cusip)