notatki

server/user_models.py

@@ -2,6 +2,7 @@ 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()
@@ -23,3 +24,58 @@ class User(UserMixin, user_db.Model):
 
     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")