diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/markit/__main__.py | 32 |
1 files changed, 16 insertions, 16 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py index a273860f..bb228695 100644 --- a/python/markit/__main__.py +++ b/python/markit/__main__.py @@ -1,17 +1,20 @@ import argparse import datetime import logging -import os +import numpy as np import pandas as pd +import os import sys import logging import pdb from common import root from .cds import download_cds_data, download_composite_data +from .loans import download_facility, insert_facility, download_marks, update_facility 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 @@ -21,23 +24,19 @@ group.add_argument("-l", "--loans", action="store_true", help="download markit loan data") group.add_argument("-c", "--cds", action="store_true", help="download markit cds data") -parser.add_argument('workdate', nargs='?', type = lambda s: pd.datetime.strptime(sys.argv[1], "%Y-%m-%d")), -# , -# default = pd.datetime.today() - BDay(1)) +parser.add_argument('workdate', nargs='?', type = lambda s: pd.datetime.strptime(s, "%Y-%m-%d").date()) args = parser.parse_args() + historical = True +if args.workdate is None: + historical = False + workdate = default_date() if args.loans: log_file = os.path.join(root, 'logs', 'markit_loans.log') - if args.workdate is None: - historical = False - workdate = datetime.date.today() else: log_file = os.path.join(root, 'logs', 'markit_cds.log') - if args.workdate is None: - historical = False - workdate = pd.datetime.today() - BDay(1) ## set up logging logger = logging.getLogger('markit') @@ -48,20 +47,21 @@ logger.addHandler(fh) logger.setLevel(logging.INFO) if args.loans: - workdate = str(workdate) payload={'LEGALENTITY': 'lmcg', 'USERNAME': 'serecapuser', 'PASSWORD': 'Welcome1'} + download_facility(workdate, payload) logger.info('facility downloaded') insert_facility(workdate) logger.info('facility inserted') - payload.update({'EOD':'Y'}) + payload.update({'RELATIVE': np.busday_count(default_date(), workdate)}) download_marks(workdate, payload) logger.info('marks downloaded') - payload.pop('EOD') + payload.pop('RELATIVE') update_facility(workdate, payload) logger.info('facility updated') + else: payload = {'user': 'GuillaumeHorel', 'password': 'password', @@ -75,7 +75,7 @@ else: download_composite_data(payload, historical) engine = create_engine('postgresql://serenitas_user@debian/serenitasdb') serenitasdb = dbconn('serenitasdb') - insert_cds(serenitasdb, workdate.date()) - insert_index(engine, workdate.date()) - insert_tranche(engine, workdate.date()) + insert_cds(serenitasdb, workdate) + insert_index(engine, workdate) + insert_tranche(engine, workdate) serenitasdb.close() |