新增自学模块
This commit is contained in:
@@ -1,17 +1,92 @@
|
||||
from rest_framework import viewsets, permissions
|
||||
from ..models import Tutorial
|
||||
from ..serializers import TutorialSerializer
|
||||
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 IsSuperAdminUser(permissions.BasePermission):
|
||||
def has_permission(self, request, view):
|
||||
return bool(request.user and request.user.is_super_admin())
|
||||
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 AdminTutorialViewSet(viewsets.ModelViewSet):
|
||||
queryset = Tutorial.objects.all()
|
||||
serializer_class = TutorialSerializer
|
||||
permission_classes = [IsSuperAdminUser]
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(created_by=self.request.user)
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user