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 %}