diff options
| -rw-r--r-- | python/quote_diff_bowdst.py | 39 |
1 files changed, 26 insertions, 13 deletions
diff --git a/python/quote_diff_bowdst.py b/python/quote_diff_bowdst.py index c4a03030..72503d1f 100644 --- a/python/quote_diff_bowdst.py +++ b/python/quote_diff_bowdst.py @@ -9,29 +9,38 @@ import argparse from serenitas.utils import SerenitasFileHandler import logging +from pandas.tseries.offsets import BDay -conn = dbconn("dawndb") -parser = argparse.ArgumentParser(description="determine sender destination") -parser.add_argument("--globeop", action="store_true", help="send to globeop") +def main(): + conn = dbconn("dawndb") -args = parser.parse_args() + parser = argparse.ArgumentParser(description="determine sender destination") + parser.add_argument("--globeop", action="store_true", help="send to globeop") + parser.add_argument( + "--workdate", + type=datetime.date.fromisoformat, + default=(datetime.date.today() - BDay(1)).date(), + ) + + args = parser.parse_args() -args = parser.parse_args() + args = parser.parse_args() + logger = logging.getLogger(__name__) + if not logger.handlers: + fh = SerenitasFileHandler("quote_diff_bowdst.log") + logger.addHandler(fh) + logger.setLevel(logging.INFO) -logger = logging.getLogger(__name__) -if not logger.handlers: - fh = SerenitasFileHandler("quote_diff_bowdst.log") - logger.addHandler(fh) -logger.setLevel(logging.INFO) + recon(conn, logger, args) -with conn.cursor() as c: +def recon(conn, logger, args): df = pd.read_sql( sql="SELECT * FROM list_bowd_quotes(%s)", con=conn, - params=(datetime.date.today() - datetime.timedelta(1),), + params=(args.workdate,), ) try: mask = np.isclose(df["bowd_price"], df["closeprice"], atol=0.15) @@ -79,4 +88,8 @@ with conn.cursor() as c: attach=attachments, ) logger.info(diff) -logger.info("Program Executed") + logger.info("Program Executed") + + +if __name__ == "__main__": + main() |