diff options
| -rw-r--r-- | python/markit/__main__.py | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py index 05ea50a0..8d881365 100644 --- a/python/markit/__main__.py +++ b/python/markit/__main__.py @@ -11,7 +11,6 @@ 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 utils import SerenitasFileHandler from utils.db import dbconn @@ -30,13 +29,19 @@ parser.add_argument( "workdate", nargs="?", type=lambda s: pd.datetime.strptime(s, "%Y-%m-%d").date() ) + +def default_date(): + return (pd.datetime.today() - BDay(1)).date() + + args = parser.parse_args() historical = True -if args.workdate is None and not arg.rates: - historical = False - workdate = pd.datetime.today() - BDay(1) -elif arg.rates: - workdate = pd.datetime.today() +if args.workdate is None: + if not args.rates: + historical = False + workdate = default_date() + else: + workdate = pd.datetime.today().date() else: workdate = args.workdate |