1
0
Fork 0
mirror of https://github.com/seanmorley15/AdventureLog.git synced 2025-07-19 21:09:37 +02:00

feat: update get_num_visits method to improve user visit count retrieval

This commit is contained in:
Sean Morley 2025-05-31 15:46:15 -04:00
parent 514ee85767
commit 724aec1f3a

View file

@ -22,8 +22,11 @@ class CountrySerializer(serializers.ModelSerializer):
def get_num_visits(self, obj): def get_num_visits(self, obj):
request = self.context.get('request') request = self.context.get('request')
if request and hasattr(request, 'user'): user = getattr(request, 'user', None)
return VisitedRegion.objects.filter(region__country=obj, user_id=request.user).count()
if user and user.is_authenticated:
return VisitedRegion.objects.filter(region__country=obj, user_id=user).count()
return 0 return 0
class Meta: class Meta: