diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/external_deriv_marks.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/python/external_deriv_marks.py b/python/external_deriv_marks.py index c6eb7ccb..1b741f52 100644 --- a/python/external_deriv_marks.py +++ b/python/external_deriv_marks.py @@ -90,6 +90,7 @@ def baml_navs(date: datetime.date = None): if __name__ == "__main__": import argparse + import logging from utils.db import dbconn from pandas.tseries.offsets import BDay @@ -103,16 +104,22 @@ if __name__ == "__main__": parser.add_argument( "-a", "--all", action="store_true", default=False, help="download everything" ) + parser.add_argument( + "-d", "--debug", action="store_true", default=False, help="more verbose logging" + ) args = parser.parse_args() date = None if args.all else args.date - + logging.basicConfig() + logger = logging.getLogger("external_marks") + logger.setLevel(logging.DEBUG if args.debug else logging.INFO) for cp in ["MS", "CITI", "GS", "BAML"]: + logger.info(cp) if cp != "CITI": date_arg = (date - BDay()).date() else: date_arg = date df = globals()[f"{cp.lower()}_navs"](date_arg) - print(df) + logger.info(df) with dbconn("dawndb") as conn: with conn.cursor() as c: for k, v in df[["nav"]].iterrows(): |