"""FluentGerman.ai — Database models.""" import enum from datetime import datetime, timezone from sqlalchemy import Boolean, DateTime, Enum, ForeignKey, String, Text from sqlalchemy.orm import Mapped, mapped_column, relationship from app.database import Base class InstructionType(str, enum.Enum): GLOBAL = "global" PERSONAL = "personal" HOMEWORK = "homework" class User(Base): __tablename__ = "users" id: Mapped[int] = mapped_column(primary_key=True) username: Mapped[str] = mapped_column(String(50), unique=True, index=True) email: Mapped[str] = mapped_column(String(255), unique=True, index=True) hashed_password: Mapped[str] = mapped_column(String(255)) is_admin: Mapped[bool] = mapped_column(Boolean, default=False) is_active: Mapped[bool] = mapped_column(Boolean, default=True) created_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), default=lambda: datetime.now(timezone.utc) ) instructions: Mapped[list["Instruction"]] = relationship( back_populates="user", cascade="all, delete-orphan" ) class Instruction(Base): __tablename__ = "instructions" id: Mapped[int] = mapped_column(primary_key=True) user_id: Mapped[int | None] = mapped_column( ForeignKey("users.id", ondelete="CASCADE"), nullable=True, index=True ) title: Mapped[str] = mapped_column(String(200)) content: Mapped[str] = mapped_column(Text) type: Mapped[InstructionType] = mapped_column( Enum(InstructionType), default=InstructionType.PERSONAL ) created_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), default=lambda: datetime.now(timezone.utc) ) user: Mapped[User | None] = relationship(back_populates="instructions")