1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
from __future__ import annotations
from datetime import datetime
from sqlalchemy import (
Boolean,
Column,
DateTime,
ForeignKey,
Integer,
String,
Table,
func,
)
from sqlalchemy.orm import Mapped, mapped_column, relationship
from app.utils.db import Base
# from .collections import Collection
image_tags = Table(
"image_tags",
Base.metadata,
Column("image_id", Integer, ForeignKey("images.id"), primary_key=True),
Column("tag_id", Integer, ForeignKey("tags.id"), primary_key=True),
)
class Tag(Base):
__tablename__ = "tags"
id: Mapped[int] = mapped_column(Integer, primary_key=True)
name: Mapped[str] = mapped_column(String(50), unique=True, nullable=False)
category: Mapped[str | None] = mapped_column(String(50), nullable=True)
images: Mapped[list["Image"]] = relationship(
"Image",
secondary=image_tags,
back_populates="tags",
)
class Image(Base):
__tablename__ = "images"
id: Mapped[int] = mapped_column(Integer, primary_key=True)
image_file: Mapped[str] = mapped_column(String(255), nullable=False)
original_file: Mapped[str] = mapped_column(String(255), nullable=False)
user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False)
publication_date: Mapped[datetime] = mapped_column(
DateTime(timezone=True), server_default=func.now()
)
source: Mapped[str] = mapped_column(String(255), nullable=False)
views: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
likes: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
is_verified: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
is_rejected: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
tags: Mapped[list[Tag]] = relationship(
"Tag",
secondary=image_tags,
back_populates="images",
)
"""
collections: Mapped[list["Collection"]] = relationship(
"Collection",
secondary="collection_images",
back_populates="images",
)
"""
|
