Initial commit

backend/accounts/models.py

@@ -0,0 +1,87 @@
+from django.db import models
+from django.contrib.auth.models import AbstractUser
+from django.urls import reverse
+from django.db.models.signals import post_migrate
+from django.dispatch import receiver
+from django_resized import ResizedImageField
+import os
+
+
+class CustomUser(AbstractUser):
+    # first_name inherited from base user class
+    # last_name inherited from base user class
+    # 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
+
+    avatar = models.ImageField(
+        null=True, upload_to='avatars/')
+
+    def avatar_url(self):
+        # Assuming your media root is set to 'media/'
+        return f'/api/v1/media/avatars/{self.avatar.name}'
+
+    @property
+    def full_name(self):
+        return f"{self.first_name} {self.last_name}"
+
+    @property
+    def admin_url(self):
+        return reverse('admin:users_customuser_change', args=(self.pk,))
+
+    def get_prep_value(self, value):
+        # Get the original filename without the random string
+        original_filename = self.avatar.field.storage.name(self.avatar.path)
+        return original_filename
+    pass
+
+
+@receiver(post_migrate)
+def create_superuser(sender, **kwargs):
+    if sender.name == 'accounts':
+        # Add test users here if needed
+        # They will automatically be created after migrating the db
+        users = [
+            # Superadmin Account
+            {
+                'username': os.getenv('DJANGO_ADMIN_USERNAME'),
+                'email': os.getenv('DJANGO_ADMIN_EMAIL'),
+                'password': os.getenv('DJANGO_ADMIN_PASSWORD'),
+                'is_staff': True,
+                'is_superuser': True,
+                'first_name': 'Super',
+                'last_name': 'Admin'
+            },
+            # Debug User
+            {
+                'username': 'debug-user',
+                'email': os.getenv('DJANGO_ADMIN_EMAIL'),
+                'password': os.getenv('DJANGO_ADMIN_PASSWORD'),
+                'is_staff': False,
+                'is_superuser': False,
+                'first_name': "Test",
+                'last_name': "User"
+            },
+        ]
+
+        for user in users:
+            if not CustomUser.objects.filter(username=user['username']).exists():
+                if (user['is_superuser']):
+                    USER = CustomUser.objects.create_superuser(
+                        username=user['username'],
+                        password=user['password'],
+                        email=user['email'],
+                    )
+                    print('Created Superuser:', user['username'])
+                else:
+                    USER = CustomUser.objects.create_user(
+                        username=user['username'],
+                        password=user['password'],
+                        email=user['email'],
+                    )
+                    print('Created User:', user['username'])
+                USER.first_name = user['first_name']
+                USER.last_name = user['last_name']
+                USER.is_active = True
+                USER.save()