parent
b528d4dbb4
commit
0c8ba5eb87
@ -1,52 +1,84 @@
|
||||
from rest_framework import viewsets
|
||||
from django_filters.rest_framework import DjangoFilterBackend
|
||||
from rest_framework import viewsets, generics, filters
|
||||
from rest_framework.permissions import IsAuthenticated, IsAdminUser
|
||||
|
||||
from rest_framework.response import Response
|
||||
|
||||
from .serializers import ContestSerializer, BandSerializer, FrequencySerializer, EntryCategorySerializer, \
|
||||
ReferenceSerializer, QSOSerializer, ShadowCallSerializer
|
||||
ReferenceSerializer, QSOSerializer, ShadowCallSerializer, UserSerializer
|
||||
from contest.models import Contest, Band, Frequency, EntryCategory, Reference, QSO, ShadowCall
|
||||
|
||||
|
||||
class ContestViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
"""
|
||||
Resource to list and view all available contests. Use `current/` to get the current Contest.
|
||||
"""
|
||||
queryset = Contest.objects.all()
|
||||
serializer_class = ContestSerializer
|
||||
filterset_fields = ['shortName']
|
||||
|
||||
def get_object(self):
|
||||
lookup_url_kwarg = self.lookup_url_kwarg or self.lookup_field
|
||||
if self.kwargs.get(lookup_url_kwarg) == "current":
|
||||
obj = Contest.get_current_contest()
|
||||
self.check_object_permissions(self.request, obj)
|
||||
else:
|
||||
obj = super(ContestViewSet, self).get_object()
|
||||
|
||||
return obj
|
||||
|
||||
|
||||
class BandViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
queryset = Band.objects.all()
|
||||
serializer_class = BandSerializer
|
||||
filterset_fields = ['name', 'contest']
|
||||
|
||||
|
||||
class FrequencyViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
queryset = Frequency.objects.all()
|
||||
serializer_class = FrequencySerializer
|
||||
filterset_fields = ['band', 'channel']
|
||||
|
||||
|
||||
class EntryCategoryViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
queryset = EntryCategory.objects.all()
|
||||
serializer_class = EntryCategorySerializer
|
||||
filterset_fields = ['name']
|
||||
|
||||
|
||||
class ReferenceViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
permission_classes = [IsAdminUser]
|
||||
queryset = Reference.objects.all()
|
||||
serializer_class = ReferenceSerializer
|
||||
filterset_fields = ['name']
|
||||
|
||||
|
||||
class QSOViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
class QSOViewSet(viewsets.ModelViewSet):
|
||||
permission_classes = [IsAuthenticated]
|
||||
serializer_class = QSOSerializer
|
||||
filterset_fields = ['time', 'ownNo', 'band', 'call', 'refStr']
|
||||
|
||||
def get_queryset(self):
|
||||
return QSO.objects.filter(owner=self.request.user)
|
||||
|
||||
def perform_create(self, serializer):
|
||||
return serializer.save(owner=self.request.user)
|
||||
|
||||
|
||||
class UserProfile(viewsets.ReadOnlyModelViewSet):
|
||||
"""Return the currently authenticated user as a single item"""
|
||||
pass
|
||||
class UserProfileViewSet(generics.UpdateAPIView, viewsets.GenericViewSet):
|
||||
permission_classes = [IsAuthenticated]
|
||||
serializer_class = UserSerializer
|
||||
|
||||
def list(self, request, format=None):
|
||||
user = request.user
|
||||
serializer = UserSerializer(user)
|
||||
return Response(serializer.data)
|
||||
|
||||
def get_queryset(self):
|
||||
return self.request.user
|
||||
|
||||
|
||||
class ShadowCallViewSet(viewsets.ReadOnlyModelViewSet):
|
||||
permission_classes = [IsAdminUser]
|
||||
queryset = ShadowCall.objects.all()
|
||||
serializer_class = ShadowCallSerializer
|
||||
filterset_fields = ['username', 'ref']
|
||||
|
@ -1,3 +1,4 @@
|
||||
Django<1.12
|
||||
django-crispy-forms
|
||||
django-rest-framework
|
||||
django-filter
|
||||
|
Loading…
Reference in new issue