blob: a8a971be9a666cf158015ba2c6ab7ca9e2670341 (
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
27
28
29
30
|
import logging
import os
import redis
import socket
import tasks
from json import loads, dumps
from db import dbconn
from process_queue import get_redis_queue
def run():
hostname = socket.gethostname()
ET = dbconn('etdb')
q = get_redis_queue()
while True:
f, args = loads(q.blpop("tasks")[1].decode('utf-8'))
if f == 'generate_scenarios':
args += [ET]
getattr(tasks, f)(*args)
if f == "build_portfolio":
q.rpush("tasks", dumps(("build_scenarios", args)))
if f == "build_scenarios":
q.rpush("tasks", dumps(("generate_scenarios", args[:-1])))
ET.close()
if __name__=="__main__":
logger = logging.getLogger('intex')
logger.setLevel(logging.INFO)
logger.addHandler(logging.StreamHandler())
run()
|
