diff options
Diffstat (limited to 'python/markit/import_quotes.py')
| -rw-r--r-- | python/markit/import_quotes.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/python/markit/import_quotes.py b/python/markit/import_quotes.py index 88166dd8..a584be96 100644 --- a/python/markit/import_quotes.py +++ b/python/markit/import_quotes.py @@ -6,7 +6,6 @@ import pandas as pd import os from collections import defaultdict -from common import root from pandas.tseries.offsets import BDay logger = logging.getLogger(__name__) @@ -87,7 +86,7 @@ def insert_cds(database, workdate): with database.cursor() as c: c.execute("DELETE from cds_quotes where date=%s", (workdate,)) database.commit() - with open(os.path.join(root, "Tranche_data", "CDS", filename)) as fh: + with open(os.path.join(os.environ['BASE_DIR'], "Tranche_data", "CDS", filename)) as fh: csvreader = csv.DictReader(fh) with database.cursor() as c: for line in csvreader: @@ -110,7 +109,7 @@ def insert_index(engine, workdate=None): :param workdate: date. If None, we will try to reinsert all files """ - basedir = os.path.join(root, 'Tranche_data', 'Composite_reports') + basedir = os.path.join(os.environ['BASE_DIR'], 'Tranche_data', 'Composite_reports') filenames = [os.path.join(basedir, f) for f in os.listdir(basedir) if 'Indices' in f] name_mapping = {"CDXNAHY":"HY", "CDXNAIG":"IG",'iTraxx Eur': "EU", 'iTraxx Eur Xover': "XO"} @@ -150,7 +149,7 @@ def insert_tranche(engine, workdate=None): :type workdate: pd.Timestamp """ - basedir = os.path.join(root, 'Tranche_data', 'Composite_reports') + basedir = os.path.join(os.environ['BASE_DIR'], 'Tranche_data', 'Composite_reports') filenames = [os.path.join(basedir, f) for f in os.listdir(basedir) if f.startswith('Tranche Composites')] index_version = pd.read_sql_table("index_version", engine, index_col='redindexcode') for f in filenames: |