mirror of
https://github.com/seanmorley15/AdventureLog.git
synced 2025-07-21 13:59:36 +02:00
51 lines
No EOL
2.2 KiB
Python
51 lines
No EOL
2.2 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import action
|
|
from django.shortcuts import get_object_or_404
|
|
from worldtravel.models import City, Region, Country, VisitedCity, VisitedRegion
|
|
from adventures.models import Adventure, Collection
|
|
from users.serializers import CustomUserDetailsSerializer as PublicUserSerializer
|
|
from django.contrib.auth import get_user_model
|
|
|
|
User = get_user_model()
|
|
|
|
class StatsViewSet(viewsets.ViewSet):
|
|
"""
|
|
A simple ViewSet for listing the stats of a user.
|
|
"""
|
|
@action(detail=False, methods=['get'], url_path=r'counts/(?P<username>[\w.@+-]+)')
|
|
def counts(self, request, username):
|
|
if request.user.username == username:
|
|
user = get_object_or_404(User, username=username)
|
|
else:
|
|
user = get_object_or_404(User, username=username, public_profile=True)
|
|
# serializer = PublicUserSerializer(user)
|
|
|
|
# remove the email address from the response
|
|
user.email = None
|
|
|
|
# get the counts for the user
|
|
adventure_count = Adventure.objects.filter(
|
|
user_id=user.id).count()
|
|
trips_count = Collection.objects.filter(
|
|
user_id=user.id).count()
|
|
visited_city_count = VisitedCity.objects.filter(
|
|
user_id=user.id).count()
|
|
total_cities = City.objects.count()
|
|
visited_region_count = VisitedRegion.objects.filter(
|
|
user_id=user.id).count()
|
|
total_regions = Region.objects.count()
|
|
visited_country_count = VisitedRegion.objects.filter(
|
|
user_id=user.id).values('region__country').distinct().count()
|
|
total_countries = Country.objects.count()
|
|
return Response({
|
|
'adventure_count': adventure_count,
|
|
'trips_count': trips_count,
|
|
'visited_city_count': visited_city_count,
|
|
'total_cities': total_cities,
|
|
'visited_region_count': visited_region_count,
|
|
'total_regions': total_regions,
|
|
'visited_country_count': visited_country_count,
|
|
'total_countries': total_countries
|
|
}) |