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")