31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
|
from django.http import JsonResponse
|
||
|
|
from rest_framework import permissions, viewsets
|
||
|
|
from .models import Recommendation, Species, Trip, WaterBody
|
||
|
|
from .serializers import RecommendationSerializer, SpeciesSerializer, TripSerializer, WaterBodySerializer
|
||
|
|
|
||
|
|
def health(_request):
|
||
|
|
return JsonResponse({"status": "ok", "service": "fishiq-api"})
|
||
|
|
|
||
|
|
class ReadOnlyCatalogViewSet(viewsets.ReadOnlyModelViewSet):
|
||
|
|
permission_classes = [permissions.AllowAny]
|
||
|
|
|
||
|
|
class SpeciesViewSet(ReadOnlyCatalogViewSet):
|
||
|
|
queryset = Species.objects.all().order_by("common_name")
|
||
|
|
serializer_class = SpeciesSerializer
|
||
|
|
|
||
|
|
class WaterBodyViewSet(ReadOnlyCatalogViewSet):
|
||
|
|
queryset = WaterBody.objects.prefetch_related("species").all().order_by("name")
|
||
|
|
serializer_class = WaterBodySerializer
|
||
|
|
|
||
|
|
class TripViewSet(viewsets.ModelViewSet):
|
||
|
|
serializer_class = TripSerializer
|
||
|
|
permission_classes = [permissions.IsAuthenticated]
|
||
|
|
def get_queryset(self): return Trip.objects.filter(owner=self.request.user).select_related("water_body")
|
||
|
|
def perform_create(self, serializer): serializer.save(owner=self.request.user)
|
||
|
|
|
||
|
|
class RecommendationViewSet(viewsets.ReadOnlyModelViewSet):
|
||
|
|
serializer_class = RecommendationSerializer
|
||
|
|
permission_classes = [permissions.IsAuthenticated]
|
||
|
|
def get_queryset(self): return Recommendation.objects.filter(trip__owner=self.request.user)
|
||
|
|
|