Docker/AdventureLog
Docker/AdventureLog
1
0
Fork 0
mirror of https://github.com/seanmorley15/AdventureLog.git synced 2025-07-22 22:39:36 +02:00
Code Issues Releases Activity

chore: Add TripViewSet and TripSerializer for managing trips

Browse source
This commit is contained in:
Sean Morley 2024-07-09 13:26:39 -04:00
parent c0900876c6
commit 8f0a8b2cc8
5 changed files with 78 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

50
backend/server/adventures/views.py
View file

@ -1,19 +1,17 @@
from rest_framework.decorators import action
from rest_framework import viewsets
from rest_framework.response import Response
from .models import Adventure
from .serializers import AdventureSerializer
from .models import Adventure, Trip
from .serializers import AdventureSerializer, TripSerializer
from rest_framework.permissions import IsAuthenticated
from django.db.models import Q
# Create your views here.
from .permissions import IsOwnerOrReadOnly
class AdventureViewSet(viewsets.ModelViewSet):
serializer_class = AdventureSerializer
permission_classes = [IsAuthenticated]
permission_classes = [IsAuthenticated, IsOwnerOrReadOnly]
def get_queryset(self):
# Allow any user to see public adventures or their own adventures
return Adventure.objects.filter(
Q(is_public=True) | Q(user_id=self.request.user.id)
)
@ -21,24 +19,56 @@ class AdventureViewSet(viewsets.ModelViewSet):
def perform_create(self, serializer):
serializer.save(user_id=self.request.user)
# Custom actions to return visited and planned adventures
@action(detail=False, methods=['get'])
def visited(self, request):
visited_adventures = Adventure.objects.filter(
type='visited', user_id=request.user.id, trip_id=None)
type='visited', user_id=request.user.id, trip=None)
serializer = self.get_serializer(visited_adventures, many=True)
return Response(serializer.data)
@action(detail=False, methods=['get'])
def planned(self, request):
planned_adventures = Adventure.objects.filter(
type='planned', user_id=request.user.id, trip_id=None)
type='planned', user_id=request.user.id, trip=None)
serializer = self.get_serializer(planned_adventures, many=True)
return Response(serializer.data)
@action(detail=False, methods=['get'])
def featured(self, request):
featured_adventures = Adventure.objects.filter(
type='featured', is_public=True, trip_id=None)
type='featured', is_public=True, trip=None)
serializer = self.get_serializer(featured_adventures, many=True)
return Response(serializer.data)
class TripViewSet(viewsets.ModelViewSet):
serializer_class = TripSerializer
permission_classes = [IsAuthenticated, IsOwnerOrReadOnly]
def get_queryset(self):
return Trip.objects.filter(
Q(is_public=True) | Q(user_id=self.request.user.id)
)
def perform_create(self, serializer):
serializer.save(user_id=self.request.user)
@action(detail=False, methods=['get'])
def visited(self, request):
trips = Trip.objects.filter(
type='visited', user_id=request.user.id)
serializer = self.get_serializer(trips, many=True)
return Response(serializer.data)
@action(detail=False, methods=['get'])
def planned(self, request):
trips = Trip.objects.filter(
type='planned', user_id=request.user.id)
serializer = self.get_serializer(trips, many=True)
return Response(serializer.data)
@action(detail=False, methods=['get'])
def featured(self, request):
trips = Trip.objects.filter(
type='featured', is_public=True)
serializer = self.get_serializer(trips, many=True)
return Response(serializer.data)