28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
|
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))
|