diff options
Diffstat (limited to 'python/markit/__main__.py')
| -rw-r--r-- | python/markit/__main__.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py index 70bfa598..6f0d6c54 100644 --- a/python/markit/__main__.py +++ b/python/markit/__main__.py @@ -23,14 +23,17 @@ 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("-i", "--insert-only", action="store_true", + help="do not redownload data") 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() +else: + workdate = args.workdate if args.loans: log_file = os.path.join(root, 'logs', 'markit_loans.log') @@ -69,12 +72,14 @@ else: 'report': 'FIXED_COUPON', 'date': workdate.strftime("%Y%m%d"), 'type': 'CDS'} - download_cds_data(payload) - payload.update({'type':'CredIndex','version':4}) - download_composite_data(payload, historical) - engine = create_engine('postgresql://serenitas_user@debian/serenitasdb') + if not args.insert_only: + download_cds_data(payload) + payload.update({'type':'CredIndex','version':4}) + download_composite_data(payload, historical) serenitasdb = dbconn('serenitasdb') insert_cds(serenitasdb, workdate) - insert_index(engine, workdate) - insert_tranche(engine, workdate) serenitasdb.close() + if not args.insert_only: + engine = create_engine('postgresql://serenitas_user@debian/serenitasdb') + insert_index(engine, workdate) + insert_tranche(engine, workdate) |