diff options
Diffstat (limited to 'python/task_runner.py')
| -rw-r--r-- | python/task_runner.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/python/task_runner.py b/python/task_runner.py index 52bc1e98..ad2fb817 100644 --- a/python/task_runner.py +++ b/python/task_runner.py @@ -6,15 +6,16 @@ from common import get_redis_queue from subprocess import CalledProcessError from tasks import Rpc + def run(): - ET = dbconn('etdb') + ET = dbconn("etdb") q = get_redis_queue() notify("READY=1") while True: - rpc = Rpc.from_json(q.blpop("tasks")[1].decode('utf-8')) + rpc = Rpc.from_json(q.blpop("tasks")[1].decode("utf-8")) print("Running '{}' with {}".format(rpc.fun, rpc.args)) - if rpc.fun == 'generate_scenarios': + if rpc.fun == "generate_scenarios": rpc.args += [ET] try: rpc() @@ -23,13 +24,14 @@ def run(): else: print("'{}' completed".format(rpc.fun)) if rpc.fun == "build_portfolios": - q.rpush("tasks", str(Rpc('build_scenarios', rpc.args))) + q.rpush("tasks", str(Rpc("build_scenarios", rpc.args))) if rpc.fun == "build_scenarios": - q.rpush("tasks", str(Rpc('generate_scenarios', rpc.args[:-1]))) + q.rpush("tasks", str(Rpc("generate_scenarios", rpc.args[:-1]))) ET.close() -if __name__=="__main__": - logger = logging.getLogger('intex') + +if __name__ == "__main__": + logger = logging.getLogger("intex") logger.setLevel(logging.INFO) logger.addHandler(logging.StreamHandler()) run() |