yggdrasil/server/user_models.py

from datetime import datetime, timezone

from flask_login import UserMixin
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.orm import relationship
from werkzeug.security import check_password_hash, generate_password_hash

user_db = SQLAlchemy()


class User(UserMixin, user_db.Model):
    __tablename__ = "user"

    id = user_db.Column(user_db.Integer, primary_key=True)
    username = user_db.Column(user_db.String(80), nullable=False, unique=True)
    password_hash = user_db.Column(user_db.String(255), nullable=False)
    is_admin = user_db.Column(user_db.Boolean, nullable=False, default=False)
    created_at = user_db.Column(
        user_db.DateTime, default=lambda: datetime.now(timezone.utc)
    )

    def set_password(self, password: str) -> None:
        self.password_hash = generate_password_hash(password)

    def check_password(self, password: str) -> bool:
        return check_password_hash(self.password_hash, password)


class Note(user_db.Model):
    __tablename__ = "note"

    id = user_db.Column(user_db.Integer, primary_key=True)
    user_id = user_db.Column(
        user_db.Integer, user_db.ForeignKey("user.id"), nullable=False, index=True
    )
    title = user_db.Column(
        user_db.String(200), nullable=False, server_default="Untitled"
    )
    content = user_db.Column(user_db.Text, nullable=False)
    created_at = user_db.Column(
        user_db.DateTime, nullable=False, default=lambda: datetime.now(timezone.utc)
    )
    updated_at = user_db.Column(
        user_db.DateTime,
        nullable=False,
        default=lambda: datetime.now(timezone.utc),
        onupdate=lambda: datetime.now(timezone.utc),
    )

    items = relationship(
        "NoteItem",
        back_populates="note",
        cascade="all, delete-orphan",
    )

    @property
    def ordered_items(self):
        """Unchecked items first, checked items last; stable within each group."""
        return sorted(self.items, key=lambda item: (item.is_done, item.id or 0))

    def touch(self) -> None:
        self.updated_at = datetime.now(timezone.utc)


class NoteItem(user_db.Model):
    __tablename__ = "note_item"

    id = user_db.Column(user_db.Integer, primary_key=True)
    note_id = user_db.Column(
        user_db.Integer,
        user_db.ForeignKey("note.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )
    text = user_db.Column(user_db.String(255), nullable=False)
    is_done = user_db.Column(user_db.Boolean, nullable=False, default=False)
    created_at = user_db.Column(
        user_db.DateTime, nullable=False, default=lambda: datetime.now(timezone.utc)
    )

    note = relationship("Note", back_populates="items")