Added user model

equipment_tracker/accounts/admin.py

@@ -0,0 +1,11 @@
+from django import forms
+from django.contrib import admin
+from django.contrib.auth.admin import UserAdmin
+from .models import CustomUser
+
+
+class CustomUserAdmin(UserAdmin):
+    model = CustomUser
+
+
+admin.site.register(CustomUser, CustomUserAdmin)

equipment_tracker/accounts/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class AccountsConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'accounts'

equipment_tracker/accounts/migrations/0001_initial.py

@@ -0,0 +1,46 @@
+# Generated by Django 4.2.7 on 2023-11-09 11:42
+
+import accounts.models
+import django.contrib.auth.models
+import django.contrib.auth.validators
+from django.db import migrations, models
+import django.utils.timezone
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('auth', '0012_alter_user_first_name_max_length'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CustomUser',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('password', models.CharField(max_length=128, verbose_name='password')),
+                ('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
+                ('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
+                ('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
+                ('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
+                ('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
+                ('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
+                ('first_name', models.CharField(max_length=100)),
+                ('last_name', models.CharField(max_length=100)),
+                ('is_active', models.BooleanField(default=False)),
+                ('avatar', models.ImageField(null=True, upload_to=accounts.models.CustomUser._get_upload_to)),
+                ('groups', models.ManyToManyField(blank=True, help_text='The groups this user belongs to. A user will get all permissions granted to each of their groups.', related_name='user_set', related_query_name='user', to='auth.group', verbose_name='groups')),
+                ('user_permissions', models.ManyToManyField(blank=True, help_text='Specific permissions for this user.', related_name='user_set', related_query_name='user', to='auth.permission', verbose_name='user permissions')),
+            ],
+            options={
+                'verbose_name': 'user',
+                'verbose_name_plural': 'users',
+                'abstract': False,
+            },
+            managers=[
+                ('objects', django.contrib.auth.models.UserManager()),
+            ],
+        ),
+    ]

equipment_tracker/accounts/models.py

@@ -0,0 +1,82 @@
+from django.db import models
+from django.contrib.auth.models import AbstractUser
+from django.db import models
+from django.db.models.signals import post_migrate
+from django.dispatch import receiver
+import os
+from uuid import uuid4
+
+
+class CustomUser(AbstractUser):
+    # Function for avatar uploads
+    def _get_upload_to(instance, filename):
+        base_filename, file_extension = os.path.splitext(filename)
+        # Get the student ID number
+        ext = base_filename.split('.')[-1]
+        filename = '{}.{}'.format(uuid4().hex, ext)
+
+        student_id = str(instance.student_id_number)
+        new_filename = f"{student_id}_{filename}_{file_extension}"
+        return os.path.join('avatars', new_filename)
+
+    # Delete old avatar file if new one is uploaded
+    def save(self, *args, **kwargs):
+        try:
+            # is the object in the database yet?
+            this = CustomUser.objects.get(id=self.id)
+            if this.avatar != self.avatar:
+                this.avatar.delete(save=False)
+        except:
+            pass  # when new photo then we do nothing, normal case
+        super(CustomUser, self).save(*args, **kwargs)
+
+    first_name = models.CharField(max_length=100)
+    last_name = models.CharField(max_length=100)
+    # Email inherited from base user class
+    # Username inherited from base user class
+    # Password inherited from base user class
+    # is_admin inherited from base user class
+    is_active = models.BooleanField(default=False)
+    avatar = models.ImageField(upload_to=_get_upload_to, null=True)
+
+    @property
+    def full_name(self):
+        return f"{self.first_name} {self.last_name}"
+    pass
+
+
+@receiver(post_migrate)
+def create_superuser(sender, **kwargs):
+    if sender.name == 'accounts':
+        User = CustomUser
+        username = os.getenv('DJANGO_ADMIN_USERNAME')
+        email = os.getenv('DJANGO_ADMIN_EMAIL')
+        password = os.getenv('DJANGO_ADMIN_PASSWORD')
+        first_name = 'Admin'
+        last_name = 'Admin'
+
+        if not User.objects.filter(username=username).exists():
+            # Create the superuser with is_active set to False
+            superuser = User.objects.create_superuser(
+                username=username, email=email, password=password, first_name=first_name, last_name=last_name)
+
+            # Activate the superuser
+            superuser.is_active = True
+            print('Created admin account')
+            superuser.save()
+
+        username = 'usertest1'
+        email = os.getenv('DJANGO_ADMIN_EMAIL')
+        password = os.getenv('DJANGO_ADMIN_PASSWORD')
+        first_name = 'Test'
+        last_name = 'User'
+
+        if not User.objects.filter(username=username).exists():
+            # Create the superuser with is_active set to False
+            user = User.objects.create_user(
+                username=username, email=email, password=password, first_name=first_name, last_name=last_name)
+
+            # Activate the user
+            user.is_active = True
+            print('Created debug user account')
+            user.save()

equipment_tracker/accounts/serializers.py

@@ -0,0 +1,45 @@
+from djoser.serializers import UserCreateSerializer as BaseUserRegistrationSerializer
+from djoser.serializers import UserSerializer as BaseUserSerializer
+from django.core import exceptions as django_exceptions
+from rest_framework import serializers
+from accounts.models import CustomUser
+from rest_framework.settings import api_settings
+from django.contrib.auth.password_validation import validate_password
+from django.utils.encoding import smart_str
+from drf_spectacular.utils import extend_schema_field
+from drf_spectacular.types import OpenApiTypes
+from drf_extra_fields.fields import Base64ImageField
+
+# There can be multiple subject instances with the same name, only differing in course, year level, and semester. We filter them here
+
+
+class CustomUserSerializer(BaseUserSerializer):
+    avatar = Base64ImageField()
+
+    class Meta(BaseUserSerializer.Meta):
+        model = CustomUser
+        fields = ('username', 'email', 'avatar', 'first_name', 'last_name',)
+
+
+class UserRegistrationSerializer(serializers.ModelSerializer):
+    email = serializers.EmailField(required=True)
+    password = serializers.CharField(
+        write_only=True, style={'input_type': 'password', 'placeholder': 'Password'})
+
+    class Meta:
+        model = CustomUser    # Use your custom user model here
+        fields = ('username', 'email', 'password', 'avatar',
+                  'first_name', 'last_name')
+
+    def validate(self, attrs):
+        user = self.Meta.model(**attrs)
+        password = attrs.get("password")
+        try:
+            validate_password(password, user)
+        except django_exceptions.ValidationError as e:
+            serializer_error = serializers.as_serializer_error(e)
+            raise serializers.ValidationError(
+                {"password": serializer_error[api_settings.NON_FIELD_ERRORS_KEY]}
+            )
+
+        return super().validate(attrs)

equipment_tracker/accounts/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

equipment_tracker/accounts/urls.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+from django.urls import path, include
+
+urlpatterns = [
+    path('', include('djoser.urls')),
+    path('', include('djoser.urls.jwt')),
+]

equipment_tracker/accounts/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.