diff options
Diffstat (limited to 'python/intex/load_intex_collateral.py')
| -rw-r--r-- | python/intex/load_intex_collateral.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/python/intex/load_intex_collateral.py b/python/intex/load_intex_collateral.py index 601c450f..08c780fe 100644 --- a/python/intex/load_intex_collateral.py +++ b/python/intex/load_intex_collateral.py @@ -1,7 +1,7 @@ import psycopg2 import os, csv, datetime import pdb -from common import root, sanitize_float +from common import sanitize_float from db import dbconn import sys import uuid @@ -35,7 +35,7 @@ def upload_data(conn, dealnames, workdate): sqlstr = "INSERT INTO ET_COLLATERAL({0}) VALUES({1})".format(",".join(sql_fields), ",".join(["%s"] * len(sql_fields))) for dealname in dealnames: - basedir = os.path.join(root, "data", "Collaterals_" + workdate) + basedir = os.path.join(os.environ['DATA_DIR'], "Collaterals_" + workdate) if sys.version_info[0]==2: fh = open(os.path.join(basedir, dealname.upper() + "_AD.txt")) fh = windows1252_encoder(fh) @@ -135,8 +135,8 @@ def upload_data(conn, dealnames, workdate): def intex_data(conn, workdate): dealnames = [d.replace("_AD.txt", "").lower() for d in - os.listdir(os.path.join(root, "data", "Collaterals_" + workdate))] - basedir = os.path.join(root, "data", "Indicative_" + workdate) + os.listdir(os.path.join(os.environ['DATA_DIR'], "Collaterals_" + workdate))] + basedir = os.path.join(os.environ['DATA_DIR'], "Indicative_" + workdate) files = [os.path.join(basedir, f) for f in os.listdir(basedir)] cusip_files = [f for f in files if "TrInfo" in f] deal_files = [f for f in files if "TrInfo" not in f] |