diff options
Diffstat (limited to 'python/risk')
| -rw-r--r-- | python/risk/__init__.py | 8 | ||||
| -rw-r--r-- | python/risk/__main__.py | 8 |
2 files changed, 9 insertions, 7 deletions
diff --git a/python/risk/__init__.py b/python/risk/__init__.py index 49c8342c..7468b863 100644 --- a/python/risk/__init__.py +++ b/python/risk/__init__.py @@ -1,5 +1,13 @@ +import logging import sys sys.path.append("..") from utils.db import dbconn, dbengine from utils import SerenitasFileHandler + +fh = SerenitasFileHandler("risk.log") + +loggers = [logging.getLogger("analytics"), logging.getLogger(__name__)] +for logger in loggers: + logger.setLevel(logging.INFO) + logger.addHandler(fh) diff --git a/python/risk/__main__.py b/python/risk/__main__.py index 346150f8..49a85eca 100644 --- a/python/risk/__main__.py +++ b/python/risk/__main__.py @@ -1,5 +1,4 @@ import argparse -import logging import pandas as pd from . import dbconn, dbengine from pandas.tseries.offsets import BDay @@ -7,7 +6,6 @@ from .bonds import subprime_risk, clo_risk, crt_risk, insert_subprime_risk from analytics import init_ontr from .swaptions import get_swaption_portfolio, insert_swaption_portfolio from .tranches import get_tranche_portfolio, insert_tranche_portfolio -from . import SerenitasFileHandler parser = argparse.ArgumentParser() parser.add_argument( @@ -19,13 +17,9 @@ if args.workdate is None: else: workdate = args.workdate -fh = SerenitasFileHandler("risk.log") + init_ontr(workdate) -loggers = [logging.getLogger("analytics"), logging.getLogger("risk")] -for logger in loggers: - logger.setLevel(logging.INFO) - logger.addHandler(fh) mysql_engine = dbengine("rmbs_model") mysqlcrt_engine = dbengine("crt") |