aboutsummaryrefslogtreecommitdiffstats
path: root/python/markit/__main__.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/markit/__main__.py')
-rw-r--r--python/markit/__main__.py13
1 files changed, 5 insertions, 8 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py
index 100f9d1b..765d1537 100644
--- a/python/markit/__main__.py
+++ b/python/markit/__main__.py
@@ -1,22 +1,18 @@
import argparse
-import datetime
import logging
import numpy as np
import pandas as pd
import os
-import sys
-import logging
from .cds import download_cds_data, download_composite_data
from .loans import download_facility, insert_facility, download_marks, update_facility
from .rates import downloadMarkitIRData
from .import_quotes import insert_cds, insert_index, insert_tranche
-from pandas.tseries.offsets import BDay
from sqlalchemy import create_engine
from .utils import default_date
from db import dbconn
-## parse arguments
+# parse arguments
parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("-l", "--loans", action="store_true",
@@ -27,7 +23,8 @@ group.add_argument("-r", "--rates", action="store_true",
help="download markit IR data")
parser.add_argument("-i", "--insert-only", action="store_true",
help="do not re-download data")
-parser.add_argument('workdate', nargs='?', type = lambda s: pd.datetime.strptime(s, "%Y-%m-%d").date())
+parser.add_argument('workdate', nargs='?',
+ type=lambda s: pd.datetime.strptime(s, "%Y-%m-%d").date())
args = parser.parse_args()
historical = True
@@ -44,7 +41,7 @@ elif args.cds:
elif args.rates:
log_file = os.path.join(os.environ['LOG_DIR'], 'markit_rates.log')
-## set up logging
+# set up logging
logger = logging.getLogger('markit')
fh = logging.FileHandler(filename=log_file)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -78,7 +75,7 @@ elif args.cds:
'type': 'CDS'}
if not args.insert_only:
download_cds_data(payload)
- payload.update({'type':'CredIndex','version':4})
+ payload.update({'type': 'CredIndex', 'version': 4})
download_composite_data(payload, historical)
serenitasdb = dbconn('serenitasdb')
insert_cds(serenitasdb, workdate)