From dc784058e0aff88cacb964e71ffae5e1d6e880e4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafael=20L=C3=A1szl=C3=B3?= <rlacko99@gmail.com>
Date: Fri, 12 Mar 2021 21:22:42 +0100
Subject: [PATCH] extra bits

---
 src/account/models.py      | 1 +
 src/account/serializers.py | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/account/models.py b/src/account/models.py
index 8dab9f0..6ab3bd1 100644
--- a/src/account/models.py
+++ b/src/account/models.py
@@ -46,6 +46,7 @@ class Profile(models.Model):
     groups = models.ManyToManyField(
         GroupChoice, related_name='profiles', blank=True)
     role = models.CharField(max_length=10, choices=ROLES, default='Applicant')
+    extra_bits = models.IntegerField(default=0, blank=False)
 
     @property
     def events_visited(self):
diff --git a/src/account/serializers.py b/src/account/serializers.py
index 895d845..fceda00 100644
--- a/src/account/serializers.py
+++ b/src/account/serializers.py
@@ -69,8 +69,8 @@ class ProfileSerializer_User(serializers.ModelSerializer):
 
     def get_bits(self, obj):
         if obj.homework_bits is None:
-            return obj.events_visited
-        return obj.homework_bits + obj.events_visited
+            return obj.events_visited + obj.extra_bits
+        return obj.homework_bits + obj.events_visited + obj.extra_bits
 
 
 class ProfileSerializer_Staff(serializers.ModelSerializer):
-- 
GitLab