mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2025-11-04 14:49:58 +08:00
add generate user api
This commit is contained in:
parent
334b67488a
commit
e8b06f0487
@ -67,7 +67,7 @@ class User(AbstractBaseUser):
|
|||||||
|
|
||||||
|
|
||||||
class UserProfile(models.Model):
|
class UserProfile(models.Model):
|
||||||
user = models.OneToOneField(User)
|
user = models.OneToOneField(User, on_delete=models.CASCADE)
|
||||||
# acm_problems_status examples:
|
# acm_problems_status examples:
|
||||||
# {
|
# {
|
||||||
# "problems": {
|
# "problems": {
|
||||||
|
|||||||
@ -35,6 +35,15 @@ class UserChangeEmailSerializer(serializers.Serializer):
|
|||||||
tfa_code = serializers.CharField(required=False, allow_blank=True)
|
tfa_code = serializers.CharField(required=False, allow_blank=True)
|
||||||
|
|
||||||
|
|
||||||
|
class GenerateUserSerializer(serializers.Serializer):
|
||||||
|
prefix = serializers.CharField(max_length=16, allow_blank=True)
|
||||||
|
suffix = serializers.CharField(max_length=16, allow_blank=True)
|
||||||
|
number_from = serializers.IntegerField()
|
||||||
|
number_to = serializers.IntegerField()
|
||||||
|
default_email = serializers.CharField(max_length=64)
|
||||||
|
password_length = serializers.IntegerField(required=False, max_value=16)
|
||||||
|
|
||||||
|
|
||||||
class UserSerializer(serializers.ModelSerializer):
|
class UserSerializer(serializers.ModelSerializer):
|
||||||
create_time = DateTimeTZField()
|
create_time = DateTimeTZField()
|
||||||
last_login = DateTimeTZField()
|
last_login = DateTimeTZField()
|
||||||
|
|||||||
@ -1,7 +1,8 @@
|
|||||||
from django.conf.urls import url
|
from django.conf.urls import url
|
||||||
|
|
||||||
from ..views.admin import UserAdminAPI
|
from ..views.admin import UserAdminAPI, GenerateUserAPI
|
||||||
|
|
||||||
urlpatterns = [
|
urlpatterns = [
|
||||||
url(r"^user/?$", UserAdminAPI.as_view(), name="user_admin_api"),
|
url(r"^user/?$", UserAdminAPI.as_view(), name="user_admin_api"),
|
||||||
|
url(r"^generate_user/?$", GenerateUserAPI.as_view(), name="generate_user_api"),
|
||||||
]
|
]
|
||||||
|
|||||||
@ -1,12 +1,16 @@
|
|||||||
|
import os
|
||||||
|
import re
|
||||||
|
import xlsxwriter
|
||||||
from django.db.models import Q
|
from django.db.models import Q
|
||||||
|
from django.http import HttpResponse
|
||||||
|
|
||||||
from submission.models import Submission
|
from submission.models import Submission
|
||||||
from utils.api import APIView, validate_serializer
|
from utils.api import APIView, validate_serializer
|
||||||
from utils.shortcuts import rand_str
|
from utils.shortcuts import rand_str
|
||||||
|
|
||||||
from ..decorators import super_admin_required
|
from ..decorators import super_admin_required
|
||||||
from ..models import AdminType, ProblemPermission, User
|
from ..models import AdminType, ProblemPermission, User, UserProfile
|
||||||
from ..serializers import EditUserSerializer, UserSerializer
|
from ..serializers import EditUserSerializer, UserSerializer, GenerateUserSerializer
|
||||||
|
|
||||||
|
|
||||||
class UserAdminAPI(APIView):
|
class UserAdminAPI(APIView):
|
||||||
@ -85,3 +89,82 @@ class UserAdminAPI(APIView):
|
|||||||
Q(userprofile__real_name__icontains=keyword) |
|
Q(userprofile__real_name__icontains=keyword) |
|
||||||
Q(email__icontains=keyword))
|
Q(email__icontains=keyword))
|
||||||
return self.success(self.paginate_data(request, user, UserSerializer))
|
return self.success(self.paginate_data(request, user, UserSerializer))
|
||||||
|
|
||||||
|
def delete_one(self, user_id):
|
||||||
|
try:
|
||||||
|
user = User.objects.get(id=user_id)
|
||||||
|
except User.DoesNotExist:
|
||||||
|
return f"User {user_id} does not exist"
|
||||||
|
profile = user.userprofile
|
||||||
|
if profile.submission_number:
|
||||||
|
return f"Can't delete the user {user_id} as he/she has submissions"
|
||||||
|
user.delete()
|
||||||
|
|
||||||
|
@super_admin_required
|
||||||
|
def delete(self, request):
|
||||||
|
id = request.GET.get("id")
|
||||||
|
if not id:
|
||||||
|
return self.error("Invalid Parameter, user_id is required")
|
||||||
|
for user_id in id.split(","):
|
||||||
|
if user_id:
|
||||||
|
error = self.delete_one(user_id)
|
||||||
|
if error:
|
||||||
|
return self.error(error)
|
||||||
|
return self.success()
|
||||||
|
|
||||||
|
|
||||||
|
class GenerateUserAPI(APIView):
|
||||||
|
@super_admin_required
|
||||||
|
def get(self, request):
|
||||||
|
"""
|
||||||
|
download users excel
|
||||||
|
"""
|
||||||
|
file_id = request.GET.get("file_id")
|
||||||
|
if not file_id:
|
||||||
|
return self.error("Invalid Parameter, file_id is required")
|
||||||
|
if not re.match(r"[a-zA-Z0-9]+", file_id):
|
||||||
|
return self.error("Illegal file_id")
|
||||||
|
file_path = f"/tmp/{file_id}.xlsx"
|
||||||
|
if not os.path.isfile(file_path):
|
||||||
|
return self.error("File does not exist")
|
||||||
|
with open(file_path, "rb") as f:
|
||||||
|
raw_data = f.read()
|
||||||
|
os.remove(file_path)
|
||||||
|
response = HttpResponse(raw_data)
|
||||||
|
response["Content-Disposition"] = f"attachment; filename=users.xlsx"
|
||||||
|
response["Content-Type"] = "application/xlsx"
|
||||||
|
return response
|
||||||
|
|
||||||
|
@validate_serializer(GenerateUserSerializer)
|
||||||
|
@super_admin_required
|
||||||
|
def post(self, request):
|
||||||
|
data = request.data
|
||||||
|
number_max_length = max(len(str(data["number_from"])), len(str(data["number_to"])))
|
||||||
|
if number_max_length + len(data["prefix"]) + len(data["suffix"]) > 32:
|
||||||
|
return self.error("Username should not more than 32 characters")
|
||||||
|
if data["number_from"] >= data["number_to"]:
|
||||||
|
return self.error("Start number must be lower than end number")
|
||||||
|
|
||||||
|
password_length = data.get("password_length", 8)
|
||||||
|
default_email = data.get("default_email")
|
||||||
|
|
||||||
|
file_id = rand_str(8)
|
||||||
|
filename = f"/tmp/{file_id}.xlsx"
|
||||||
|
workbook = xlsxwriter.Workbook(filename)
|
||||||
|
worksheet = workbook.add_worksheet()
|
||||||
|
worksheet.set_column("A:B", 20)
|
||||||
|
worksheet.write("A1", "Username")
|
||||||
|
worksheet.write("B1", "Password")
|
||||||
|
i = 1
|
||||||
|
for number in range(data["number_from"], data["number_to"] + 1):
|
||||||
|
username = f"{data['prefix']}{number}{data['suffix']}"
|
||||||
|
password = rand_str(password_length)
|
||||||
|
user = User.objects.create(username=username, email=default_email)
|
||||||
|
user.set_password(password)
|
||||||
|
user.save()
|
||||||
|
UserProfile.objects.create(user=user)
|
||||||
|
worksheet.write_string(i, 0, username)
|
||||||
|
worksheet.write_string(i, 1, password)
|
||||||
|
i += 1
|
||||||
|
workbook.close()
|
||||||
|
return self.success(file_id)
|
||||||
|
|||||||
@ -15,3 +15,4 @@ django-redis
|
|||||||
psycopg2
|
psycopg2
|
||||||
gunicorn
|
gunicorn
|
||||||
jsonfield
|
jsonfield
|
||||||
|
XlsxWriter
|
||||||
|
|||||||
@ -221,10 +221,10 @@ class JudgeDispatcher(DispatcherBase):
|
|||||||
user_profile.accepted_number += 1
|
user_profile.accepted_number += 1
|
||||||
else:
|
else:
|
||||||
if oi_problems_status[problem_id]["status"] == JudgeStatus.ACCEPTED and \
|
if oi_problems_status[problem_id]["status"] == JudgeStatus.ACCEPTED and \
|
||||||
self.submission.result != JudgeStatus.ACCEPTED:
|
self.submission.result != JudgeStatus.ACCEPTED:
|
||||||
user_profile.accepted_number -= 1
|
user_profile.accepted_number -= 1
|
||||||
elif oi_problems_status[problem_id]["status"] != JudgeStatus.ACCEPTED and \
|
elif oi_problems_status[problem_id]["status"] != JudgeStatus.ACCEPTED and \
|
||||||
self.submission.result == JudgeStatus:
|
self.submission.result == JudgeStatus:
|
||||||
user_profile.accepted_number += 1
|
user_profile.accepted_number += 1
|
||||||
|
|
||||||
# minus last time score, add this time score
|
# minus last time score, add this time score
|
||||||
|
|||||||
@ -49,6 +49,7 @@ class CreateOrEditProblemSerializer(serializers.Serializer):
|
|||||||
spj = serializers.BooleanField()
|
spj = serializers.BooleanField()
|
||||||
spj_language = serializers.ChoiceField(choices=spj_language_names, allow_blank=True, allow_null=True)
|
spj_language = serializers.ChoiceField(choices=spj_language_names, allow_blank=True, allow_null=True)
|
||||||
spj_code = serializers.CharField(allow_blank=True, allow_null=True)
|
spj_code = serializers.CharField(allow_blank=True, allow_null=True)
|
||||||
|
spj_compile_ok = serializers.BooleanField(default=False)
|
||||||
visible = serializers.BooleanField()
|
visible = serializers.BooleanField()
|
||||||
difficulty = serializers.ChoiceField(choices=[Difficulty.LOW, Difficulty.MID, Difficulty.HIGH])
|
difficulty = serializers.ChoiceField(choices=[Difficulty.LOW, Difficulty.MID, Difficulty.HIGH])
|
||||||
tags = serializers.ListField(child=serializers.CharField(max_length=32), allow_empty=False)
|
tags = serializers.ListField(child=serializers.CharField(max_length=32), allow_empty=False)
|
||||||
@ -80,7 +81,6 @@ class TagSerializer(serializers.ModelSerializer):
|
|||||||
|
|
||||||
|
|
||||||
class CompileSPJSerializer(serializers.Serializer):
|
class CompileSPJSerializer(serializers.Serializer):
|
||||||
id = serializers.IntegerField()
|
|
||||||
spj_language = serializers.ChoiceField(choices=spj_language_names)
|
spj_language = serializers.ChoiceField(choices=spj_language_names)
|
||||||
spj_code = serializers.CharField()
|
spj_code = serializers.CharField()
|
||||||
|
|
||||||
@ -131,5 +131,3 @@ class ContestProblemSafeSerializer(BaseProblemSerializer):
|
|||||||
class ContestProblemMakePublicSerializer(serializers.Serializer):
|
class ContestProblemMakePublicSerializer(serializers.Serializer):
|
||||||
id = serializers.IntegerField()
|
id = serializers.IntegerField()
|
||||||
display_id = serializers.CharField(max_length=32)
|
display_id = serializers.CharField(max_length=32)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -19,7 +19,7 @@ DEFAULT_PROBLEM_DATA = {"_id": "A-110", "title": "test", "description": "<p>test
|
|||||||
"output_description": "test", "time_limit": 1000, "memory_limit": 256, "difficulty": "Low",
|
"output_description": "test", "time_limit": 1000, "memory_limit": 256, "difficulty": "Low",
|
||||||
"visible": True, "tags": ["test"], "languages": ["C", "C++", "Java", "Python2"], "template": {},
|
"visible": True, "tags": ["test"], "languages": ["C", "C++", "Java", "Python2"], "template": {},
|
||||||
"samples": [{"input": "test", "output": "test"}], "spj": False, "spj_language": "C",
|
"samples": [{"input": "test", "output": "test"}], "spj": False, "spj_language": "C",
|
||||||
"spj_code": "", "test_case_id": "499b26290cc7994e0b497212e842ea85",
|
"spj_code": "", "spj_compile_ok": True, "test_case_id": "499b26290cc7994e0b497212e842ea85",
|
||||||
"test_case_score": [{"output_name": "1.out", "input_name": "1.in", "output_size": 0,
|
"test_case_score": [{"output_name": "1.out", "input_name": "1.in", "output_size": 0,
|
||||||
"stripped_output_md5": "d41d8cd98f00b204e9800998ecf8427e",
|
"stripped_output_md5": "d41d8cd98f00b204e9800998ecf8427e",
|
||||||
"input_size": 0, "score": 0}],
|
"input_size": 0, "score": 0}],
|
||||||
|
|||||||
@ -1,6 +1,7 @@
|
|||||||
import hashlib
|
import hashlib
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
|
import shutil
|
||||||
import zipfile
|
import zipfile
|
||||||
from wsgiref.util import FileWrapper
|
from wsgiref.util import FileWrapper
|
||||||
|
|
||||||
@ -10,6 +11,7 @@ from django.http import StreamingHttpResponse
|
|||||||
from account.decorators import problem_permission_required
|
from account.decorators import problem_permission_required
|
||||||
from judge.dispatcher import SPJCompiler
|
from judge.dispatcher import SPJCompiler
|
||||||
from contest.models import Contest
|
from contest.models import Contest
|
||||||
|
from submission.models import Submission
|
||||||
from utils.api import APIView, CSRFExemptAPIView, validate_serializer
|
from utils.api import APIView, CSRFExemptAPIView, validate_serializer
|
||||||
from utils.shortcuts import rand_str, natural_sort_key
|
from utils.shortcuts import rand_str, natural_sort_key
|
||||||
|
|
||||||
@ -140,23 +142,11 @@ class CompileSPJAPI(APIView):
|
|||||||
@problem_permission_required
|
@problem_permission_required
|
||||||
def post(self, request):
|
def post(self, request):
|
||||||
data = request.data
|
data = request.data
|
||||||
try:
|
|
||||||
problem = Problem.objects.get(pk=data["id"])
|
|
||||||
except Problem.DoesNotExist:
|
|
||||||
return self.error("Problem does not exist")
|
|
||||||
spj_version = rand_str(8)
|
spj_version = rand_str(8)
|
||||||
problem.spj = True
|
|
||||||
problem.spj_version = spj_version
|
|
||||||
problem.spj_language = data["spj_language"]
|
|
||||||
problem.spj_code = data["spj_code"]
|
|
||||||
error = SPJCompiler(data["spj_code"], spj_version, data["spj_language"]).compile_spj()
|
error = SPJCompiler(data["spj_code"], spj_version, data["spj_language"]).compile_spj()
|
||||||
if error:
|
if error:
|
||||||
problem.spj_compile_ok = False
|
|
||||||
problem.save()
|
|
||||||
return self.error(error)
|
return self.error(error)
|
||||||
else:
|
else:
|
||||||
problem.spj_compile_ok = True
|
|
||||||
problem.save()
|
|
||||||
return self.success()
|
return self.success()
|
||||||
|
|
||||||
|
|
||||||
@ -166,6 +156,8 @@ class ProblemBase(APIView):
|
|||||||
if data["spj"]:
|
if data["spj"]:
|
||||||
if not data["spj_language"] or not data["spj_code"]:
|
if not data["spj_language"] or not data["spj_code"]:
|
||||||
return "Invalid spj"
|
return "Invalid spj"
|
||||||
|
if not data["spj_compile_ok"]:
|
||||||
|
return "SPJ code must be compiled successfully"
|
||||||
data["spj_version"] = hashlib.md5(
|
data["spj_version"] = hashlib.md5(
|
||||||
(data["spj_language"] + ":" + data["spj_code"]).encode("utf-8")).hexdigest()
|
(data["spj_language"] + ":" + data["spj_code"]).encode("utf-8")).hexdigest()
|
||||||
else:
|
else:
|
||||||
@ -182,6 +174,23 @@ class ProblemBase(APIView):
|
|||||||
data["created_by"] = request.user
|
data["created_by"] = request.user
|
||||||
data["languages"] = list(data["languages"])
|
data["languages"] = list(data["languages"])
|
||||||
|
|
||||||
|
@problem_permission_required
|
||||||
|
def delete(self, request):
|
||||||
|
id = request.GET.get("id")
|
||||||
|
if not id:
|
||||||
|
return self.error("Invalid parameter, id is requred")
|
||||||
|
try:
|
||||||
|
problem = Problem.objects.get(id=id)
|
||||||
|
except Problem.DoesNotExist:
|
||||||
|
return self.error("Problem does not exists")
|
||||||
|
if Submission.objects.filter(problem=problem).exists():
|
||||||
|
return self.error("Can't delete the problem as it has submissions")
|
||||||
|
d = os.path.join(settings.TEST_CASE_DIR, problem.test_case_id)
|
||||||
|
if os.path.isdir(d):
|
||||||
|
shutil.rmtree(d, ignore_errors=True)
|
||||||
|
problem.delete()
|
||||||
|
return self.success()
|
||||||
|
|
||||||
|
|
||||||
class ProblemAPI(ProblemBase):
|
class ProblemAPI(ProblemBase):
|
||||||
@validate_serializer(CreateProblemSerializer)
|
@validate_serializer(CreateProblemSerializer)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user