aboutsummaryrefslogtreecommitdiffstats
path: root/python/risk/__main__.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/risk/__main__.py')
-rw-r--r--python/risk/__main__.py11
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)