From 712a6c533f6ceab8723b857e5c5db258454065fb Mon Sep 17 00:00:00 2001
From: Tanner Collin <git@tannercollin.com>
Date: Sun, 19 Jan 2020 08:59:16 +0000
Subject: [PATCH] Retally membership months on any transaction change

---
 apiserver/apiserver/api/views.py | 12 +++++++++++-
 webclient/src/Classes.js         |  2 +-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/apiserver/apiserver/api/views.py b/apiserver/apiserver/api/views.py
index c32845a..84e0060 100644
--- a/apiserver/apiserver/api/views.py
+++ b/apiserver/apiserver/api/views.py
@@ -9,7 +9,7 @@ from rest_auth.registration.views import RegisterView
 from fuzzywuzzy import fuzz, process
 from collections import OrderedDict
 
-from . import models, serializers
+from . import models, serializers, utils
 
 class AllowMetadata(BasePermission):
     def has_permission(self, request, view):
@@ -197,8 +197,18 @@ class TransactionViewSet(Base, Create, Retrieve, Update):
     queryset = models.Transaction.objects.all()
     serializer_class = serializers.TransactionSerializer
 
+    def retally_membership(self):
+        member_id = self.request.data['member_id']
+        member = get_object_or_404(models.Member, id=member_id)
+        utils.tally_membership_months(member)
+
     def perform_create(self, serializer):
         serializer.save(recorder=self.request.user)
+        self.retally_membership()
+
+    def perform_update(self, serializer):
+        serializer.save()
+        self.retally_membership()
 
 
 class UserView(views.APIView):
diff --git a/webclient/src/Classes.js b/webclient/src/Classes.js
index d914992..068c064 100644
--- a/webclient/src/Classes.js
+++ b/webclient/src/Classes.js
@@ -40,7 +40,7 @@ function ClassTable(props) {
 						</Table.Row>
 					)
 				:
-					<p>None</p>
+					<Table.Row><Table.Cell>None</Table.Cell></Table.Row>
 				}
 			</Table.Body>
 		</Table>