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

Merge pull request #139 from seanmorley15/development

Browse source 
chore: Update VisitedRegionViewSet to set user_id in create method
This commit is contained in:
Sean Morley 2024-07-23 12:09:49 -04:00 • committed by GitHub
commit f1748a2cf0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

16
backend/server/worldtravel/views.py
View file

@ -1,7 +1,7 @@
from django.shortcuts import render
from .models import Country, Region, VisitedRegion
from .serializers import CountrySerializer, RegionSerializer, VisitedRegionSerializer
from rest_framework import viewsets
from rest_framework import viewsets, status
from rest_framework.permissions import IsAuthenticated
from django.shortcuts import get_object_or_404
from rest_framework.response import Response
@ -16,7 +16,6 @@ def regions_by_country(request, country_code):
serializer = RegionSerializer(regions, many=True)
return Response(serializer.data)
@api_view(['GET'])
@permission_classes([IsAuthenticated])
def visits_by_country(request, country_code):
@ -26,7 +25,6 @@ def visits_by_country(request, country_code):
serializer = VisitedRegionSerializer(visits, many=True)
return Response(serializer.data)
class CountryViewSet(viewsets.ReadOnlyModelViewSet):
queryset = Country.objects.all()
serializer_class = CountrySerializer
@ -37,10 +35,18 @@ class RegionViewSet(viewsets.ReadOnlyModelViewSet):
serializer_class = RegionSerializer
permission_classes = [IsAuthenticated]
class VisitedRegionViewSet(viewsets.ModelViewSet):
serializer_class = VisitedRegionSerializer
permission_classes = [IsAuthenticated]
def get_queryset(self):
return VisitedRegion.objects.filter(user_id=self.request.user.id)
return VisitedRegion.objects.filter(user_id=self.request.user.id)
def create(self, request, *args, **kwargs):
# Set the user_id to the request user's ID
request.data['user_id'] = request.user.id
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
self.perform_create(serializer)
headers = self.get_success_headers(serializer.data)
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)