diff options
Diffstat (limited to 'python/intex')
| -rw-r--r-- | python/intex/common.py | 5 | ||||
| -rw-r--r-- | python/intex/intex_scenarios.py | 7 | ||||
| -rw-r--r-- | python/intex/load_indicative.py | 6 | ||||
| -rw-r--r-- | python/intex/load_intex_collateral.py | 8 |
4 files changed, 10 insertions, 16 deletions
diff --git a/python/intex/common.py b/python/intex/common.py index 009f0ba6..d93f9c18 100644 --- a/python/intex/common.py +++ b/python/intex/common.py @@ -1,10 +1,5 @@ import os -if os.name =='nt': - root = "//WDsentinel/share/CorpCDOs" -elif os.name == 'posix': - root = '/home/serenitas/CorpCDOs' - def sanitize_float(intex_float): try: intex_float = intex_float.replace(",", "") diff --git a/python/intex/intex_scenarios.py b/python/intex/intex_scenarios.py index e2c9aa8b..2eaa1dbe 100644 --- a/python/intex/intex_scenarios.py +++ b/python/intex/intex_scenarios.py @@ -11,7 +11,6 @@ import yaml sys.path.append('.') from csv import reader -from common import root from db import dbconn logger = logging.getLogger(__name__) @@ -68,9 +67,9 @@ def get_reinvenddate(conn, dealname): raise Exception("missing reinvestment end date") def generate_scenarios(workdate, dealname, conn): - prometheus = os.path.join(root, "Scenarios", "prometheus.sss") + prometheus = os.path.join(os.environ['BASE_DIR'], "Scenarios", "prometheus.sss") n_scenarios = 100 - basedir = os.path.join(root, "Scenarios", "Intex curves_" + workdate) + basedir = os.path.join(os.environ['BASE_DIR'], "Scenarios", "Intex curves_" + workdate) defaultedprice = get_recovery(conn, dealname, workdate) replace = "\g<a>{0:.3f}".format(defaultedprice) try: @@ -234,7 +233,7 @@ if __name__ == "__main__": dealnames = sys.argv[2:] else: dealnames = [d.split(".")[0] for d in - os.listdir(os.path.join(root, "Scenarios", + os.listdir(os.path.join(os.environ('BASE_DIR'], "Scenarios", "Intex curves_" + workdate, "csv")) if "RData" in d] ET = dbconn('etdb') diff --git a/python/intex/load_indicative.py b/python/intex/load_indicative.py index c4fef63b..6feeb0ae 100644 --- a/python/intex/load_indicative.py +++ b/python/intex/load_indicative.py @@ -4,7 +4,7 @@ import datetime from datetime import date import csv, sys, re import pdb -from common import root, sanitize_float +from common import sanitize_float from db import dbconn from contextlib import closing import logging @@ -182,8 +182,8 @@ if __name__=="__main__": workdate = sys.argv[1] else: workdate = str(datetime.date.today()) - files = [os.path.join(root, "data", "Indicative_" + workdate, f) for f in - os.listdir(os.path.join(root, "data", "Indicative_" + workdate))] + files = [os.path.join(os.environ['DATA_DIR'], "Indicative_" + workdate, f) for f in + os.listdir(os.path.join(os.environ['DATA_DIR'], "Indicative_" + workdate))] cusip_files = [f for f in files if "TrInfo" in f] deal_files = [f for f in files if "TrInfo" not in f] 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] |