xuyue/OnlineJudge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add some tests

Browse Source
This commit is contained in:
virusdefender
2016-10-30 02:17:35 +08:00
parent 078de956e5
commit 39857d1b56
118 changed files with 326 additions and 19131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
announcement/views.py
View File

@@ -2,11 +2,9 @@
from __future__ import unicode_literals
from django.utils.translation import ugettext as _
from rest_framework.views import APIView
from account.decorators import super_admin_required
from utils.shortcuts import paginate
from utils.shortcuts import serializer_invalid_response, error_response, success_response
from utils.shortcuts import paginate_data, APIView
from .models import Announcement
from .serializers import (CreateAnnouncementSerializer, AnnouncementSerializer,
EditAnnouncementSerializer)
@@ -24,9 +22,9 @@ class AnnouncementAdminAPIView(APIView):
announcement = Announcement.objects.create(title=data["title"],
content=data["content"],
created_by=request.user)
return success_response(AnnouncementSerializer(announcement).data)
return self.success(AnnouncementSerializer(announcement).data)
else:
return serializer_invalid_response(serializer)
return self.invalid_serializer(serializer)
@super_admin_required
def put(self, request):
@@ -39,16 +37,16 @@ class AnnouncementAdminAPIView(APIView):
try:
announcement = Announcement.objects.get(id=data["id"])
except Announcement.DoesNotExist:
return error_response(_("Announcement does not exist"))
return self.error(_("Announcement does not exist"))
announcement.title = data["title"]
announcement.content = data["content"]
announcement.visible = data["visible"]
announcement.save()
return success_response(AnnouncementSerializer(announcement).data)
return self.success(AnnouncementSerializer(announcement).data)
else:
return serializer_invalid_response(serializer)
return self.invalid_serializer(serializer)
@super_admin_required
def get(self, request):
@@ -59,8 +57,8 @@ class AnnouncementAdminAPIView(APIView):
if announcement_id:
try:
announcement = Announcement.objects.get(id=announcement_id)
return success_response(AnnouncementSerializer(announcement).data)
return self.success(AnnouncementSerializer(announcement).data)
except Announcement.DoesNotExist:
return error_response(_("Announcement does not exist"))
return self.error(_("Announcement does not exist"))
announcement = Announcement.objects.all().order_by("-create_time")
return paginate(request, announcement, AnnouncementSerializer)
return self.success(paginate_data(request, announcement, AnnouncementSerializer))