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:
parent
514ee85767
commit
724aec1f3a
1 changed files with 5 additions and 2 deletions
|
@ -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:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue