1 files changed, 20 insertions, 2 deletions
diff --git a/python/master.py b/python/master.py
index e1e488d2..bc89a8aa 100644
--- a/python/master.py
+++ b/python/master.py
@@ -1,8 +1,26 @@
 import redis
 from pickle import dumps
 import datetime
+import os
+import common
+import time
+import pdb
 
+cusipsfile = os.path.join(common.root, "scripts", "cusips_to_price.txt")
 q = redis.Redis()
 workdate = str(datetime.date.today())
-for dealname in ['ares26', 'carlg131', 'cata131']:
-    q.rpush("tasks", dumps(("generate_scenarios", [workdate, dealname])))
+
+while True:
+    if os.path.exists(cusipsfile):
+        with open(cusipsfile) as fh:
+            cusip, reinvflag = zip(*[line.rstrip().split("\t") for line in fh])
+        c = common.conn.cursor()
+        sqlstr = "select * from dealname_from_cusip({0})".format(",",join(["%s"]*len(cusip)))
+        c.execute(sqlstr, params = cusip)
+        dealnames = [d[0] for d in c.fetchall()]
+
+        q.rpush("tasks", dumps(("build_portfolio", [workdate, dealname, reinvflag])))
+        os.unlink(cusipsfile)
+    time.sleep(3)
+
+common.conn.close()