Fix bug where num_visits was not user specific

seanmorley15 · Nov 16, 2024 · ae92fc2 · ae92fc2
1 parent 4a7f720
commit ae92fc2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/backend/server/worldtravel/serializers.py b/backend/server/worldtravel/serializers.py
@@ -19,7 +19,10 @@ def get_num_regions(self, obj):
         return Region.objects.filter(country=obj).count()
 
     def get_num_visits(self, obj):
-        return VisitedRegion.objects.filter(region__country=obj).count()
+        request = self.context.get('request')
+        if request and hasattr(request, 'user'):
+            return VisitedRegion.objects.filter(region__country=obj, user_id=request.user).count()
+        return 0
 
     class Meta:
         model = Country