59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
from datetime import date
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import Column, Integer, String, Boolean, DateTime, Date
|
|
from sqlalchemy.dialects.postgresql import ENUM
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.sql import func
|
|
import enum
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class UserRole(str, enum.Enum):
|
|
USER = "user"
|
|
MODERATOR = "moderator"
|
|
ADMIN = "admin"
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
username = Column(String(50), unique=True, index=True, nullable=False)
|
|
email = Column(String(120), unique=True, index=True, nullable=False)
|
|
hashed_password = Column(String(255), nullable=False)
|
|
date_of_birth = Column(Date, nullable=True)
|
|
is_active = Column(Boolean, default=True)
|
|
role = Column(
|
|
ENUM(UserRole, name="user_role", create_type=False),
|
|
default=UserRole.USER,
|
|
nullable=False,
|
|
)
|
|
banned_until = Column(DateTime(timezone=True), nullable=True)
|
|
suspended_until = Column(DateTime(timezone=True), nullable=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
|
|
|
|
sessions = relationship("Session", back_populates="user", cascade="all, delete-orphan")
|
|
worlds = relationship("World", back_populates="owner")
|
|
assets = relationship("Asset", back_populates="owner")
|
|
reports_made = relationship("Report", foreign_keys="Report.reporter_id", back_populates="reporter")
|
|
reports_resolved = relationship("Report", foreign_keys="Report.resolved_by", back_populates="resolved_by_user")
|
|
moderation_actions_made = relationship("ModerationAction", foreign_keys="ModerationAction.moderator_id", back_populates="moderator")
|
|
moderation_actions_targeted = relationship("ModerationAction", foreign_keys="ModerationAction.target_user_id", back_populates="target_user")
|
|
appeals_made = relationship("Appeal", foreign_keys="Appeal.user_id", back_populates="user")
|
|
appeals_reviewed = relationship("Appeal", foreign_keys="Appeal.reviewed_by", back_populates="reviewed_by_user")
|
|
|
|
@property
|
|
def is_adult(self) -> bool:
|
|
"""Return True if the user is 18 or older based on date_of_birth."""
|
|
if self.date_of_birth is None:
|
|
return False
|
|
today = date.today()
|
|
age = today.year - self.date_of_birth.year
|
|
# Adjust if birthday hasn't occurred yet this year
|
|
if (today.month, today.day) < (self.date_of_birth.month, self.date_of_birth.day):
|
|
age -= 1
|
|
return age >= 18
|