blob: cb111dd996a5bf0849ca7125d8afd5c7953e235a (
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
|
import redis
import tasks
from pickle import loads, dumps
import socket
import daemon
from daemon.pidlockfile import TimeoutPIDLockFile
def run():
hostname = socket.gethostname()
if hostname == 'debian':
q = redis.Redis(unix_socket_path='/var/run/redis/redis.sock')
elif hostname == 'gomez':
q = redis.Redis(host='debian')
while True:
f, args = loads(q.blpop("tasks")[1])
getattr(tasks, f)(*args)
if f == "build_portfolio":
q.rpush("tasks", dumps(("build_scenarios", args), protocol=2))
if f == "build_scenarios":
q.rpush("tasks", dumps(("generate_scenarios", args[:-1]), protocol=2))
if __name__=="__main__":
with daemon.DaemonContext(pidfile = TimeoutPIDLockFile('/var/run/client.pid',10)):
run()
|
