diff options
Diffstat (limited to 'python/risk/__main__.py')
| -rw-r--r-- | python/risk/__main__.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/python/risk/__main__.py b/python/risk/__main__.py index 596b7333..35c39031 100644 --- a/python/risk/__main__.py +++ b/python/risk/__main__.py @@ -1,4 +1,6 @@ import argparse +import logging +import os import pandas as pd from db import dbconn, dbengine from pandas.tseries.offsets import BDay @@ -15,6 +17,15 @@ if args.workdate is None: else: workdate = args.workdate +fh = logging.FileHandler(filename=os.path.join(os.getenv("LOG_DIR"), "risk.log")) +formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') +fh.setFormatter(formatter) + +loggers = [logging.getLogger('analytics'), logging.getLogger('risk')] +for logger in loggers: + logger.setLevel(logging.INFO) + logger.addHandler(fh) + with dbconn('dawndb') as conn: portf = get_swaption_portfolio(workdate, conn, source_list=["GS"]) insert_swaption_portfolio(portf, conn) |
