diff options
Diffstat (limited to 'python/markit/cds.py')
| -rw-r--r-- | python/markit/cds.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/python/markit/cds.py b/python/markit/cds.py index 2ae7692f..29e1f623 100644 --- a/python/markit/cds.py +++ b/python/markit/cds.py @@ -5,7 +5,6 @@ import requests import shutil
import zipfile
import time
-from common import root
from pandas.tseries.offsets import BDay
import pandas as pd
logger = logging.getLogger(__name__)
@@ -15,7 +14,8 @@ def convertToNone(v): def download_cds_data(payload):
r = requests.get('https://www.markit.com/export.jsp', params=payload)
- f2 = open(os.path.join(root, "Tranche_data", "CDS", "cds eod {0}.csv".format(payload['date'])), "wb")
+ f2 = open(os.path.join(os.environ['BASE_DIR'], "Tranche_data", "CDS",
+ "cds eod {0}.csv".format(payload['date'])), "wb")
with zipfile.ZipFile(io.BytesIO(r.content)) as z:
for f in z.namelist():
if "csv" in f:
@@ -39,7 +39,9 @@ def download_composite_data(payload, historical=False): with zipfile.ZipFile(io.BytesIO(r.content)) as z:
for f in z.namelist():
if "csv" in f:
- path = z.extract(f, path=os.path.join(root, "Tranche_data", "Composite_reports"))
+ path = z.extract(f, path=os.path.join(os.environ['BASE_DIR'],
+ "Tranche_data",
+ "Composite_reports"))
if historical:
os.utime(path, (ts, ts))
except zipfile.BadZipfile:
|