Files
OnlineJudge/tutorial/views/admin.py
2025-06-15 14:41:01 +08:00

93 lines
2.8 KiB
Python

from account.decorators import super_admin_required
from utils.api import APIView, validate_serializer
from tutorial.models import Tutorial
from tutorial.serializers import (
TutorialSerializer,
TutorialListSerializer,
CreateTutorialSerializer,
EditTutorialSerializer,
)
class TutorialAdminAPI(APIView):
@validate_serializer(CreateTutorialSerializer)
@super_admin_required
def post(self, request):
"""
create tutorial
"""
data = request.data
tutorial = Tutorial.objects.create(**data, created_by=request.user)
return self.success(TutorialSerializer(tutorial).data)
@validate_serializer(EditTutorialSerializer)
@super_admin_required
def put(self, request):
"""
edit tutorial
"""
data = request.data
try:
tutorial = Tutorial.objects.get(id=data.pop("id"))
except Tutorial.DoesNotExist:
return self.error("Tutorial does not exist")
for k, v in data.items():
setattr(tutorial, k, v)
tutorial.save()
return self.success(TutorialSerializer(tutorial).data)
@super_admin_required
def get(self, request):
"""
get tutorial list / get one tutorial
"""
tutorial_id = request.GET.get("id")
if tutorial_id:
try:
tutorial = Tutorial.objects.get(id=tutorial_id)
return self.success(TutorialSerializer(tutorial).data)
except Tutorial.DoesNotExist:
return self.error("Tutorial does not exist")
tutorials = Tutorial.objects.all().order_by("order", "-created_at")
# 按 type 分组返回数据
result = {
"python": TutorialListSerializer(
tutorials.filter(type="python"),
many=True
).data,
"c": TutorialListSerializer(
tutorials.filter(type="c"),
many=True
).data
}
return self.success(result)
@super_admin_required
def delete(self, request):
if request.GET.get("id"):
Tutorial.objects.filter(id=request.GET["id"]).delete()
return self.success()
class TutorialVisibilityAPI(APIView):
@super_admin_required
def put(self, request):
"""
change tutorial visibility
"""
tutorial_id = request.data.get("id")
is_public = request.data.get("is_public")
if tutorial_id is None or is_public is None:
return self.error("Invalid parameter")
try:
tutorial = Tutorial.objects.get(id=tutorial_id)
except Tutorial.DoesNotExist:
return self.error("Tutorial does not exist")
tutorial.is_public = is_public
tutorial.save()
return self.success(TutorialSerializer(tutorial).data)