diff --git a/tin/apps/users/migrations/0003_user_dark_mode.py b/tin/apps/users/migrations/0003_user_dark_mode.py
index 80174d87..dba18dff 100644
--- a/tin/apps/users/migrations/0003_user_dark_mode.py
+++ b/tin/apps/users/migrations/0003_user_dark_mode.py
@@ -1,5 +1,6 @@
-# Generated by Django 4.2.15 on 2024-09-03 19:43
+# Generated by Django 4.2.15 on 2024-09-04 12:28
+import django.core.validators
from django.db import migrations, models
@@ -13,6 +14,6 @@ class Migration(migrations.Migration):
migrations.AddField(
model_name='user',
name='dark_mode',
- field=models.IntegerField(default=0),
+ field=models.PositiveIntegerField(default=0, validators=[django.core.validators.MaxValueValidator(1)]),
),
]
diff --git a/tin/apps/users/models.py b/tin/apps/users/models.py
index 3ed14353..4701aa0a 100644
--- a/tin/apps/users/models.py
+++ b/tin/apps/users/models.py
@@ -5,6 +5,7 @@
import requests
from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin
from django.contrib.auth.models import UserManager as DjangoUserManager
+from django.core.validators import MaxValueValidator
from django.db import models
from django.utils import timezone
from social_django.utils import load_strategy
@@ -30,7 +31,8 @@ class User(AbstractBaseUser, PermissionsMixin):
is_teacher = models.BooleanField(default=False)
is_student = models.BooleanField(default=False)
date_joined = models.DateTimeField(default=timezone.now)
- dark_mode = models.IntegerField(default=0)
+ # 0 = Light mode, 1 = Dark Mode
+ dark_mode = models.PositiveIntegerField(default=0, validators=[MaxValueValidator(1)])
USERNAME_FIELD = "username"
REQUIRED_FIELDS = ["email"]
diff --git a/tin/apps/users/views.py b/tin/apps/users/views.py
index 21568bbf..d539075d 100644
--- a/tin/apps/users/views.py
+++ b/tin/apps/users/views.py
@@ -8,6 +8,7 @@
@login_required
def change_theme(request):
+ """Sets the color theme"""
if request.method == "POST":
form = ThemeForm(request.POST)
if form.is_valid():
diff --git a/tin/templates/base.html b/tin/templates/base.html
index 99e89521..d33f30b7 100644
--- a/tin/templates/base.html
+++ b/tin/templates/base.html
@@ -8,7 +8,7 @@
{% include "meta.html" %}
- {% if request.user.dark_mode == 1 %}
+ {% if request.user.dark_mode %}
{% endif %}