23 lines
873 B
Python
23 lines
873 B
Python
from django.core.management.base import BaseCommand
|
|
|
|
from judge.models import Problem, ProblemGroup, ProblemType
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "create an empty problem"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument("code", help="problem code")
|
|
parser.add_argument("name", help="problem title")
|
|
parser.add_argument("body", help="problem description")
|
|
parser.add_argument("type", help="problem type")
|
|
parser.add_argument("group", help="problem group")
|
|
|
|
def handle(self, *args, **options):
|
|
problem = Problem()
|
|
problem.code = options["code"]
|
|
problem.name = options["name"]
|
|
problem.description = options["body"]
|
|
problem.group = ProblemGroup.objects.get(name=options["group"])
|
|
problem.types = [ProblemType.objects.get(name=options["type"])]
|
|
problem.save()
|