From 85dbc20a80a2b4c233708be1460da7909e205388 Mon Sep 17 00:00:00 2001 From: Christopher Baklid Date: Fri, 25 May 2018 14:21:10 +0200 Subject: lint --- snekbox.py | 23 ++++++++++------------- 1 file changed, 10 insertions(+), 13 deletions(-) (limited to 'snekbox.py') diff --git a/snekbox.py b/snekbox.py index 4dfcc48..ee82be8 100644 --- a/snekbox.py +++ b/snekbox.py @@ -1,23 +1,14 @@ -import traceback import sys -import time -import pika import io import json from rmq.consumer import consume from rmq.publisher import publish -from config import USERNAME -from config import PASSWORD from config import HOST -from config import PORT -from config import EXCHANGE from config import EXCHANGE_TYPE from config import QUEUE -from config import RETURN_QUEUE -from config import RETURN_EXCHANGE -from config import RETURN_ROUTING_KEY + def execute(snippet): old_stdout = sys.stdout @@ -38,15 +29,21 @@ def execute(snippet): def message_handler(ch, method, properties, body): msg = body.decode('utf-8') - # Execute code snippets here print(f"incoming: {msg}", flush=True) snek_msg = json.loads(msg) + for snekid, snekcode in snek_msg.items(): result = execute(snekcode) print(f"outgoing: {result}", flush=True) - publish(result, host=HOST, queue=snekid, routingkey=snekid, exchange=snekid, exchange_type=EXCHANGE_TYPE) + publish(result, + host=HOST, + queue=snekid, + routingkey=snekid, + exchange=snekid, + exchange_type=EXCHANGE_TYPE) + + ch.basic_ack(delivery_tag=method.delivery_tag) - ch.basic_ack(delivery_tag = method.delivery_tag) if __name__ == '__main__': consume(host=HOST, queue=QUEUE, callback=message_handler) -- cgit v1.2.3