From 9223da6b903015fb3393b041134cc672dea2027e Mon Sep 17 00:00:00 2001
From: Tanner Collin <git@tannercollin.com>
Date: Tue, 21 Apr 2020 21:12:50 +0000
Subject: [PATCH] Add member counts to /charts/ API

---
 apiserver/apiserver/api/serializers.py | 6 ++++++
 apiserver/apiserver/api/views.py       | 6 ++++++
 apiserver/apiserver/urls.py            | 1 +
 3 files changed, 13 insertions(+)

diff --git a/apiserver/apiserver/api/serializers.py b/apiserver/apiserver/api/serializers.py
index b962dcd..fca9eba 100644
--- a/apiserver/apiserver/api/serializers.py
+++ b/apiserver/apiserver/api/serializers.py
@@ -417,6 +417,12 @@ class MyPasswordChangeSerializer(PasswordChangeSerializer):
         super().save()
 
 
+class MemberCountSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = models.StatsMemberCount
+        fields = '__all__'
+
+
 class HistoryChangeSerializer(serializers.ModelSerializer):
     class Meta:
         model = models.HistoryChange
diff --git a/apiserver/apiserver/api/views.py b/apiserver/apiserver/api/views.py
index 421a62f..c4b5dba 100644
--- a/apiserver/apiserver/api/views.py
+++ b/apiserver/apiserver/api/views.py
@@ -385,6 +385,12 @@ class StatsViewSet(viewsets.ViewSet, List):
             raise exceptions.ValidationError(dict(data='This field is required.'))
 
 
+class MemberCountViewSet(Base, List):
+    pagination_class = None
+    queryset = models.StatsMemberCount.objects.all()
+    serializer_class = serializers.MemberCountSerializer
+
+
 class BackupView(views.APIView):
     def get(self, request):
         auth_token = request.META.get('HTTP_AUTHORIZATION', '')
diff --git a/apiserver/apiserver/urls.py b/apiserver/apiserver/urls.py
index 4407a03..675f1a3 100644
--- a/apiserver/apiserver/urls.py
+++ b/apiserver/apiserver/urls.py
@@ -19,6 +19,7 @@ router.register(r'history', views.HistoryViewSet, basename='history')
 router.register(r'sessions', views.SessionViewSet, basename='session')
 router.register(r'training', views.TrainingViewSet, basename='training')
 router.register(r'transactions', views.TransactionViewSet, basename='transaction')
+router.register(r'charts/membercount', views.MemberCountViewSet, basename='membercount')
 #router.register(r'me', views.FullMemberView, basename='fullmember')
 #router.register(r'registration', views.RegistrationViewSet, basename='register')