diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/risk/__init__.py | 3 | ||||
| -rw-r--r-- | python/risk/__main__.py | 6 |
2 files changed, 4 insertions, 5 deletions
diff --git a/python/risk/__init__.py b/python/risk/__init__.py index ccf50cb9..cf6570a4 100644 --- a/python/risk/__init__.py +++ b/python/risk/__init__.py @@ -1,5 +1,6 @@ import sys sys.path.append("..") -from db import dbengine +from utils.db import dbconn, dbengine +from utils import SerenitasFileHandler mysql_engine = dbengine('rmbs_model') 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: |