Only include APPROVED users for stats on homepage

dandiapi/api/tests/test_stats.py

@@ -2,14 +2,15 @@
 
 import pytest
 
+from dandiapi.api.models import UserMetadata
+
 
 @pytest.mark.django_db
 def test_stats_baseline(api_client):
     assert api_client.get('/api/stats/').data == {
         'dandiset_count': 0,
         'published_dandiset_count': 0,
-        # django-guardian automatically creates an AnonymousUser
-        'user_count': 1,
+        'user_count': 0,
         'size': 0,
     }
 
@@ -32,11 +33,17 @@ def test_stats_published(api_client, published_version_factory):
 
 
 @pytest.mark.django_db
-def test_stats_user(api_client, user):
+def test_stats_user(api_client, user_factory):
+    # Create multiple users with different statuses
+    users_per_status = approved_user_count = 3
+
+    for status in UserMetadata.Status.choices:
+        [user_factory(metadata__status=status[0]) for _ in range(users_per_status)]
+
     stats = api_client.get('/api/stats/').data
 
-    # django-guardian automatically creates an AnonymousUser
-    assert stats['user_count'] == 2
+    # Assert that the user count only includes users with APPROVED status
+    assert stats['user_count'] == approved_user_count
 
 
 @pytest.mark.django_db

dandiapi/api/views/stats.py

@@ -5,7 +5,7 @@
 from rest_framework.decorators import api_view
 from rest_framework.response import Response
 
-from dandiapi.api.models import Asset, Dandiset
+from dandiapi.api.models import Asset, Dandiset, UserMetadata
 
 
 # Cache this response for 12 hours
@@ -14,7 +14,7 @@
 def stats_view(self):
     dandiset_count = Dandiset.objects.count()
     published_dandiset_count = Dandiset.published_count()
-    user_count = User.objects.count()
+    user_count = User.objects.filter(metadata__status=UserMetadata.Status.APPROVED).count()
     size = Asset.total_size()
     return Response(
         {