From 47fd81a5910eab3483f79d03eedf9307bc81252f Mon Sep 17 00:00:00 2001 From: l3wdfut4pwr Date: Tue, 17 Mar 2026 14:11:45 +0200 Subject: simple registration prototype --- app/models/collections.py | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 app/models/collections.py (limited to 'app/models/collections.py') diff --git a/app/models/collections.py b/app/models/collections.py new file mode 100644 index 0000000..7dc5bb7 --- /dev/null +++ b/app/models/collections.py @@ -0,0 +1,40 @@ +from __future__ import annotations + +from datetime import datetime +from typing import TYPE_CHECKING + +from sqlalchemy import Boolean, DateTime, ForeignKey, Integer, String, Table +from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship + +Base = DeclarativeBase() + +if TYPE_CHECKING: + from .image import Image + +collection_images = Table( + "collection_images", + Base.metadata, + mapped_column( + "collection_id", Integer, ForeignKey("collections.id"), primary_key=True + ), + mapped_column("image_id", Integer, ForeignKey("images.id"), primary_key=True), +) + + +class Collection(Base): + __tablename__ = "collections" + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False) + name: Mapped[str] = mapped_column(String(100), nullable=False) + description: Mapped[str | None] = mapped_column(String(500), nullable=True) + is_private: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) + created_at: Mapped[datetime] = mapped_column( + DateTime(timezone=True), default=datetime.utcnow + ) + + images: Mapped[list["Image"]] = relationship( + "Image", + secondary=collection_images, + back_populates="collections", + ) -- cgit v1.3-3-g829e