aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--python/intex/intex_scenarios.py22
1 files changed, 13 insertions, 9 deletions
diff --git a/python/intex/intex_scenarios.py b/python/intex/intex_scenarios.py
index 4b51af9f..16bf73f5 100644
--- a/python/intex/intex_scenarios.py
+++ b/python/intex/intex_scenarios.py
@@ -1,16 +1,20 @@
-import os
-import os.path
-import datetime
import csv
-from csv import reader
+import datetime
import json
-import re
+import logging
+import os
+import os.path
import psycopg2
-from common import root
+import re
import sys
+import yaml
+
sys.path.append('.')
+from csv import reader
+from common import root
from db import dbconn
-import yaml
+
+logger = logging.getLogger(__name__)
pattern1 = re.compile("REINVEST\[\w+::REINV_TBA(\d)\]\[DEAL,(\d+)\]=.*$")
# reinv end date
@@ -73,7 +77,7 @@ def generate_scenarios(workdate, dealname, conn):
with open(os.path.join(basedir, "csv", dealname + ".config")) as fh:
config = yaml.load(fh)
except IOError:
- print("{0}: config file doesn't exist".format(dealname))
+ logger.error("{0}: config file doesn't exist".format(dealname))
return
reinvflag = config['reinvflag']
if reinvflag:
@@ -219,7 +223,7 @@ def generate_scenarios(workdate, dealname, conn):
fhsss.close()
fhrecovery.close()
fhcdr.close()
- print("generated scenarios for: {0}".format(dealname))
+ logger.info("generated scenarios for: {0}".format(dealname))
if __name__ == "__main__":
if len(sys.argv) > 1: