45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
from os.path import join, exists
|
|
from shutil import rmtree
|
|
|
|
import pika
|
|
from django.core.management.base import BaseCommand
|
|
from pika.adapters.utils.connection_workflow import AMQPConnectorException
|
|
|
|
from Main.models import Solution
|
|
from Sprint import settings
|
|
from SprintLib.testers import *
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Tests solution"
|
|
|
|
def handle(self, *args, **options):
|
|
print("Enter worker")
|
|
while True:
|
|
try:
|
|
connection = pika.BlockingConnection(
|
|
pika.ConnectionParameters(host=settings.RABBIT_HOST)
|
|
)
|
|
channel = connection.channel()
|
|
channel.queue_declare(queue="test")
|
|
|
|
def callback(ch, method, properties, body):
|
|
id = int(str(body, encoding="utf-8"))
|
|
print(f"Received id {id}")
|
|
solution = Solution.objects.get(id=id)
|
|
try:
|
|
eval(solution.language.work_name + "Tester")(solution).execute()
|
|
except Exception as e:
|
|
print(e)
|
|
solution.result = "TE"
|
|
solution.save()
|
|
finally:
|
|
path = join("solutions", str(id))
|
|
if exists(path):
|
|
rmtree(path)
|
|
|
|
channel.basic_consume(queue="test", on_message_callback=callback, auto_ack=True)
|
|
channel.start_consuming()
|
|
except AMQPConnectorException:
|
|
print("connection to rabbit failed: reconnecting")
|