aboutsummaryrefslogtreecommitdiffstats
path: root/python/master.py
blob: bc89a8aa40ccfeaeefaed8fefd7fc8a782d449fb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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())

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()