48 lines
2.4 KiB
Python
48 lines
2.4 KiB
Python
from datetime import timedelta
|
|
from django.http import JsonResponse
|
|
from django.utils import timezone
|
|
from rest_framework import permissions, status, viewsets
|
|
from rest_framework.decorators import api_view, permission_classes
|
|
from rest_framework.response import Response
|
|
from .models import Recommendation, Species, Trip, WaterBody
|
|
from .recommendations import build_bass_plan
|
|
from .serializers import RecommendationSerializer, SpeciesSerializer, TripSerializer, WaterBodySerializer
|
|
|
|
def health(_request):
|
|
return JsonResponse({"status": "ok", "service": "fishiq-api"})
|
|
|
|
@api_view(["GET", "POST"])
|
|
@permission_classes([permissions.AllowAny])
|
|
def demo_plan(request):
|
|
starts_at = timezone.now() + timedelta(days=3)
|
|
supplied_start = request.data.get("starts_at") if request.method == "POST" else request.query_params.get("starts_at")
|
|
if supplied_start:
|
|
try:
|
|
starts_at = timezone.datetime.fromisoformat(supplied_start.replace("Z", "+00:00"))
|
|
if timezone.is_naive(starts_at): starts_at = timezone.make_aware(starts_at)
|
|
except ValueError:
|
|
return Response({"starts_at": ["Use an ISO-8601 date and time."]}, status=status.HTTP_400_BAD_REQUEST)
|
|
return Response({"trip": {"water_body": "Lake Eufaula", "region": "Alabama / Georgia", "target_species": "Largemouth bass", "starts_at": starts_at.isoformat(), "duration_hours": 5}, "plan": build_bass_plan(starts_at)})
|
|
|
|
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)
|