From 724aec1f3a480aa8fc7e7348570400b0a99e8cfa Mon Sep 17 00:00:00 2001 From: Sean Morley Date: Sat, 31 May 2025 15:46:15 -0400 Subject: [PATCH] feat: update get_num_visits method to improve user visit count retrieval --- backend/server/worldtravel/serializers.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/server/worldtravel/serializers.py b/backend/server/worldtravel/serializers.py index 6472125..962914b 100644 --- a/backend/server/worldtravel/serializers.py +++ b/backend/server/worldtravel/serializers.py @@ -22,8 +22,11 @@ class CountrySerializer(serializers.ModelSerializer): def get_num_visits(self, obj): request = self.context.get('request') - if request and hasattr(request, 'user'): - return VisitedRegion.objects.filter(region__country=obj, user_id=request.user).count() + user = getattr(request, 'user', None) + + if user and user.is_authenticated: + return VisitedRegion.objects.filter(region__country=obj, user_id=user).count() + return 0 class Meta: