diff options
Diffstat (limited to 'python/risk/__main__.py')
| -rw-r--r-- | python/risk/__main__.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/python/risk/__main__.py b/python/risk/__main__.py index 35c39031..bc105963 100644 --- a/python/risk/__main__.py +++ b/python/risk/__main__.py @@ -2,7 +2,7 @@ import argparse import logging import os import pandas as pd -from db import dbconn, dbengine +from . import dbconn, dbengine from pandas.tseries.offsets import BDay from .subprime import get_rmbs_portfolio, subprime_risk from .swaptions import get_swaption_portfolio, insert_swaption_portfolio @@ -17,9 +17,7 @@ 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) +fh = SerenitasFileHandler("risk.log") loggers = [logging.getLogger('analytics'), logging.getLogger('risk')] for logger in loggers: |