mirror of
https://github.com/pacnpal/thrilltrack-explorer.git
synced 2025-12-22 04:51:13 -05:00
Refactor code structure and remove redundant changes
This commit is contained in:
pacnpal
419
django-backend/apps/users/models.py
Normal file
419
django-backend/apps/users/models.py
Normal file
@@ -0,0 +1,419 @@
|
||||
"""
|
||||
User models for ThrillWiki.
|
||||
Custom user model with OAuth and MFA support.
|
||||
"""
|
||||
|
||||
import uuid
|
||||
from django.contrib.auth.models import AbstractUser
|
||||
from django.db import models
|
||||
from apps.core.models import BaseModel
|
||||
|
||||
|
||||
class User(AbstractUser):
|
||||
"""
|
||||
Custom user model with UUID primary key and additional fields.
|
||||
|
||||
Supports:
|
||||
- Email-based authentication
|
||||
- OAuth (Google, Discord)
|
||||
- Two-factor authentication (TOTP)
|
||||
- User reputation and moderation
|
||||
"""
|
||||
|
||||
# Override id to use UUID
|
||||
id = models.UUIDField(
|
||||
primary_key=True,
|
||||
default=uuid.uuid4,
|
||||
editable=False
|
||||
)
|
||||
|
||||
# Email as primary identifier
|
||||
email = models.EmailField(
|
||||
unique=True,
|
||||
help_text="Email address for authentication"
|
||||
)
|
||||
|
||||
# OAuth fields
|
||||
oauth_provider = models.CharField(
|
||||
max_length=50,
|
||||
blank=True,
|
||||
choices=[
|
||||
('', 'None'),
|
||||
('google', 'Google'),
|
||||
('discord', 'Discord'),
|
||||
],
|
||||
help_text="OAuth provider used for authentication"
|
||||
)
|
||||
oauth_sub = models.CharField(
|
||||
max_length=255,
|
||||
blank=True,
|
||||
help_text="OAuth subject identifier from provider"
|
||||
)
|
||||
|
||||
# MFA fields
|
||||
mfa_enabled = models.BooleanField(
|
||||
default=False,
|
||||
help_text="Whether two-factor authentication is enabled"
|
||||
)
|
||||
|
||||
# Profile fields
|
||||
avatar_url = models.URLField(
|
||||
blank=True,
|
||||
help_text="URL to user's avatar image"
|
||||
)
|
||||
bio = models.TextField(
|
||||
blank=True,
|
||||
max_length=500,
|
||||
help_text="User biography"
|
||||
)
|
||||
|
||||
# Moderation fields
|
||||
banned = models.BooleanField(
|
||||
default=False,
|
||||
db_index=True,
|
||||
help_text="Whether this user is banned"
|
||||
)
|
||||
ban_reason = models.TextField(
|
||||
blank=True,
|
||||
help_text="Reason for ban"
|
||||
)
|
||||
banned_at = models.DateTimeField(
|
||||
null=True,
|
||||
blank=True,
|
||||
help_text="When the user was banned"
|
||||
)
|
||||
banned_by = models.ForeignKey(
|
||||
'self',
|
||||
on_delete=models.SET_NULL,
|
||||
null=True,
|
||||
blank=True,
|
||||
related_name='users_banned',
|
||||
help_text="Moderator who banned this user"
|
||||
)
|
||||
|
||||
# Reputation system
|
||||
reputation_score = models.IntegerField(
|
||||
default=0,
|
||||
help_text="User reputation score based on contributions"
|
||||
)
|
||||
|
||||
# Timestamps (inherited from AbstractUser)
|
||||
# date_joined, last_login
|
||||
|
||||
# Use email for authentication
|
||||
USERNAME_FIELD = 'email'
|
||||
REQUIRED_FIELDS = ['username']
|
||||
|
||||
class Meta:
|
||||
db_table = 'users'
|
||||
ordering = ['-date_joined']
|
||||
indexes = [
|
||||
models.Index(fields=['email']),
|
||||
models.Index(fields=['banned']),
|
||||
]
|
||||
|
||||
def __str__(self):
|
||||
return self.email
|
||||
|
||||
def ban(self, reason, banned_by=None):
|
||||
"""Ban this user"""
|
||||
from django.utils import timezone
|
||||
self.banned = True
|
||||
self.ban_reason = reason
|
||||
self.banned_at = timezone.now()
|
||||
self.banned_by = banned_by
|
||||
self.save(update_fields=['banned', 'ban_reason', 'banned_at', 'banned_by'])
|
||||
|
||||
def unban(self):
|
||||
"""Unban this user"""
|
||||
self.banned = False
|
||||
self.ban_reason = ''
|
||||
self.banned_at = None
|
||||
self.banned_by = None
|
||||
self.save(update_fields=['banned', 'ban_reason', 'banned_at', 'banned_by'])
|
||||
|
||||
@property
|
||||
def display_name(self):
|
||||
"""Return the user's display name (full name or username)"""
|
||||
if self.first_name or self.last_name:
|
||||
return f"{self.first_name} {self.last_name}".strip()
|
||||
return self.username or self.email.split('@')[0]
|
||||
|
||||
|
||||
class UserRole(BaseModel):
|
||||
"""
|
||||
User role assignments for permission management.
|
||||
|
||||
Roles:
|
||||
- user: Standard user (default)
|
||||
- moderator: Can approve submissions and moderate content
|
||||
- admin: Full access to admin features
|
||||
"""
|
||||
|
||||
ROLE_CHOICES = [
|
||||
('user', 'User'),
|
||||
('moderator', 'Moderator'),
|
||||
('admin', 'Admin'),
|
||||
]
|
||||
|
||||
user = models.OneToOneField(
|
||||
User,
|
||||
on_delete=models.CASCADE,
|
||||
related_name='role'
|
||||
)
|
||||
role = models.CharField(
|
||||
max_length=20,
|
||||
choices=ROLE_CHOICES,
|
||||
default='user',
|
||||
db_index=True
|
||||
)
|
||||
granted_at = models.DateTimeField(auto_now_add=True)
|
||||
granted_by = models.ForeignKey(
|
||||
User,
|
||||
on_delete=models.SET_NULL,
|
||||
null=True,
|
||||
blank=True,
|
||||
related_name='roles_granted'
|
||||
)
|
||||
|
||||
class Meta:
|
||||
db_table = 'user_roles'
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.user.email} - {self.role}"
|
||||
|
||||
@property
|
||||
def is_moderator(self):
|
||||
"""Check if user is a moderator or admin"""
|
||||
return self.role in ['moderator', 'admin']
|
||||
|
||||
@property
|
||||
def is_admin(self):
|
||||
"""Check if user is an admin"""
|
||||
return self.role == 'admin'
|
||||
|
||||
|
||||
class UserProfile(BaseModel):
|
||||
"""
|
||||
Extended user profile information.
|
||||
|
||||
Stores additional user preferences and settings.
|
||||
"""
|
||||
|
||||
user = models.OneToOneField(
|
||||
User,
|
||||
on_delete=models.CASCADE,
|
||||
related_name='profile'
|
||||
)
|
||||
|
||||
# Preferences
|
||||
email_notifications = models.BooleanField(
|
||||
default=True,
|
||||
help_text="Receive email notifications"
|
||||
)
|
||||
email_on_submission_approved = models.BooleanField(
|
||||
default=True,
|
||||
help_text="Email when submissions are approved"
|
||||
)
|
||||
email_on_submission_rejected = models.BooleanField(
|
||||
default=True,
|
||||
help_text="Email when submissions are rejected"
|
||||
)
|
||||
|
||||
# Privacy
|
||||
profile_public = models.BooleanField(
|
||||
default=True,
|
||||
help_text="Make profile publicly visible"
|
||||
)
|
||||
show_email = models.BooleanField(
|
||||
default=False,
|
||||
help_text="Show email on public profile"
|
||||
)
|
||||
|
||||
# Statistics
|
||||
total_submissions = models.IntegerField(
|
||||
default=0,
|
||||
help_text="Total number of submissions made"
|
||||
)
|
||||
approved_submissions = models.IntegerField(
|
||||
default=0,
|
||||
help_text="Number of approved submissions"
|
||||
)
|
||||
|
||||
class Meta:
|
||||
db_table = 'user_profiles'
|
||||
|
||||
def __str__(self):
|
||||
return f"Profile for {self.user.email}"
|
||||
|
||||
def update_submission_stats(self):
|
||||
"""Update submission statistics"""
|
||||
from apps.moderation.models import ContentSubmission
|
||||
self.total_submissions = ContentSubmission.objects.filter(user=self.user).count()
|
||||
self.approved_submissions = ContentSubmission.objects.filter(
|
||||
user=self.user,
|
||||
status='approved'
|
||||
).count()
|
||||
self.save(update_fields=['total_submissions', 'approved_submissions'])
|
||||
|
||||
|
||||
class UserRideCredit(BaseModel):
|
||||
"""
|
||||
Track which rides users have ridden (ride credits/coaster counting).
|
||||
|
||||
Users can log which rides they've been on and track their first ride date.
|
||||
"""
|
||||
|
||||
user = models.ForeignKey(
|
||||
User,
|
||||
on_delete=models.CASCADE,
|
||||
related_name='ride_credits'
|
||||
)
|
||||
ride = models.ForeignKey(
|
||||
'entities.Ride',
|
||||
on_delete=models.CASCADE,
|
||||
related_name='user_credits'
|
||||
)
|
||||
|
||||
# First ride date
|
||||
first_ride_date = models.DateField(
|
||||
null=True,
|
||||
blank=True,
|
||||
help_text="Date of first ride"
|
||||
)
|
||||
|
||||
# Ride count for this specific ride
|
||||
ride_count = models.PositiveIntegerField(
|
||||
default=1,
|
||||
help_text="Number of times user has ridden this ride"
|
||||
)
|
||||
|
||||
# Notes about the ride experience
|
||||
notes = models.TextField(
|
||||
blank=True,
|
||||
help_text="User notes about this ride"
|
||||
)
|
||||
|
||||
class Meta:
|
||||
db_table = 'user_ride_credits'
|
||||
unique_together = [['user', 'ride']]
|
||||
ordering = ['-first_ride_date', '-created']
|
||||
indexes = [
|
||||
models.Index(fields=['user', 'first_ride_date']),
|
||||
models.Index(fields=['ride']),
|
||||
]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.user.username} - {self.ride.name}"
|
||||
|
||||
@property
|
||||
def park(self):
|
||||
"""Get the park this ride is at"""
|
||||
return self.ride.park
|
||||
|
||||
|
||||
class UserTopList(BaseModel):
|
||||
"""
|
||||
User-created ranked lists (top parks, top rides, top coasters, etc.).
|
||||
|
||||
Users can create and share their personal rankings of parks, rides, and other entities.
|
||||
"""
|
||||
|
||||
LIST_TYPE_CHOICES = [
|
||||
('parks', 'Parks'),
|
||||
('rides', 'Rides'),
|
||||
('coasters', 'Coasters'),
|
||||
]
|
||||
|
||||
user = models.ForeignKey(
|
||||
User,
|
||||
on_delete=models.CASCADE,
|
||||
related_name='top_lists'
|
||||
)
|
||||
|
||||
# List metadata
|
||||
list_type = models.CharField(
|
||||
max_length=20,
|
||||
choices=LIST_TYPE_CHOICES,
|
||||
db_index=True,
|
||||
help_text="Type of entities in this list"
|
||||
)
|
||||
title = models.CharField(
|
||||
max_length=200,
|
||||
help_text="Title of the list"
|
||||
)
|
||||
description = models.TextField(
|
||||
blank=True,
|
||||
help_text="Description of the list"
|
||||
)
|
||||
|
||||
# Privacy
|
||||
is_public = models.BooleanField(
|
||||
default=True,
|
||||
db_index=True,
|
||||
help_text="Whether this list is publicly visible"
|
||||
)
|
||||
|
||||
class Meta:
|
||||
db_table = 'user_top_lists'
|
||||
ordering = ['-created']
|
||||
indexes = [
|
||||
models.Index(fields=['user', 'list_type']),
|
||||
models.Index(fields=['is_public', 'created']),
|
||||
]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.user.username} - {self.title}"
|
||||
|
||||
@property
|
||||
def item_count(self):
|
||||
"""Get the number of items in this list"""
|
||||
return self.items.count()
|
||||
|
||||
|
||||
class UserTopListItem(BaseModel):
|
||||
"""
|
||||
Individual items in a user's top list with position and notes.
|
||||
"""
|
||||
|
||||
top_list = models.ForeignKey(
|
||||
UserTopList,
|
||||
on_delete=models.CASCADE,
|
||||
related_name='items'
|
||||
)
|
||||
|
||||
# Generic relation to park or ride
|
||||
from django.contrib.contenttypes.fields import GenericForeignKey
|
||||
from django.contrib.contenttypes.models import ContentType
|
||||
|
||||
content_type = models.ForeignKey(
|
||||
ContentType,
|
||||
on_delete=models.CASCADE,
|
||||
limit_choices_to={'model__in': ('park', 'ride')}
|
||||
)
|
||||
object_id = models.PositiveIntegerField()
|
||||
content_object = GenericForeignKey('content_type', 'object_id')
|
||||
|
||||
# Position in list (1 = top)
|
||||
position = models.PositiveIntegerField(
|
||||
help_text="Position in the list (1 = top)"
|
||||
)
|
||||
|
||||
# Optional notes about this specific item
|
||||
notes = models.TextField(
|
||||
blank=True,
|
||||
help_text="User notes about why this item is ranked here"
|
||||
)
|
||||
|
||||
class Meta:
|
||||
db_table = 'user_top_list_items'
|
||||
ordering = ['position']
|
||||
unique_together = [['top_list', 'position']]
|
||||
indexes = [
|
||||
models.Index(fields=['top_list', 'position']),
|
||||
models.Index(fields=['content_type', 'object_id']),
|
||||
]
|
||||
|
||||
def __str__(self):
|
||||
entity_name = str(self.content_object) if self.content_object else f"ID {self.object_id}"
|
||||
return f"#{self.position}: {entity_name}"
|
||||
Reference in New Issue
Block a user