diff --git a/backend/server/worldtravel/urls.py b/backend/server/worldtravel/urls.py index f3e8651..8f0610f 100644 --- a/backend/server/worldtravel/urls.py +++ b/backend/server/worldtravel/urls.py @@ -5,9 +5,9 @@ from rest_framework.routers import DefaultRouter from .views import CountryViewSet, RegionViewSet, VisitedRegionViewSet, regions_by_country, visits_by_country router = DefaultRouter() -router.register(r'countries', CountryViewSet) -router.register(r'regions', RegionViewSet) -router.register(r'visitedregion', VisitedRegionViewSet) +router.register(r'countries', CountryViewSet, basename='countries') +router.register(r'regions', RegionViewSet, basename='regions') +router.register(r'visitedregion', VisitedRegionViewSet, basename='visitedregion') urlpatterns = [ path('', include(router.urls)), diff --git a/backend/server/worldtravel/views.py b/backend/server/worldtravel/views.py index 0d320ff..276c58f 100644 --- a/backend/server/worldtravel/views.py +++ b/backend/server/worldtravel/views.py @@ -37,7 +37,10 @@ class RegionViewSet(viewsets.ReadOnlyModelViewSet): serializer_class = RegionSerializer permission_classes = [IsAuthenticated] + class VisitedRegionViewSet(viewsets.ModelViewSet): - queryset = VisitedRegion.objects.all() serializer_class = VisitedRegionSerializer - permission_classes = [IsAuthenticated] \ No newline at end of file + permission_classes = [IsAuthenticated] + + def get_queryset(self): + return VisitedRegion.objects.filter(user_id=self.request.user.id) \ No newline at end of file