Cloned DMOJ

judge/management/commands/copy_language.py

@@ -0,0 +1,27 @@
+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))