from django.core.management.base import BaseCommand, CommandError

from judge.models import Language, LanguageLimit


class Command(BaseCommand):
    help = "allows the problems that allow <source> to be submitted in <target>"

    def add_arguments(self, parser):
        parser.add_argument("source", help="language to copy from")
        parser.add_argument("target", help="language to copy to")

    def handle(self, *args, **options):
        try:
            source = Language.objects.get(key=options["source"])
        except Language.DoesNotExist:
            raise CommandError("Invalid source language: %s" % options["source"])

        try:
            target = Language.objects.get(key=options["target"])
        except Language.DoesNotExist:
            raise CommandError("Invalid target language: %s" % options["target"])

        target.problem_set.set(source.problem_set.all())
        LanguageLimit.objects.bulk_create(
            LanguageLimit(
                problem=ll.problem,
                language=target,
                time_limit=ll.time_limit,
                memory_limit=ll.memory_limit,
            )
            for ll in LanguageLimit.objects.filter(language=source)
        )