import time from typing import Optional, TYPE_CHECKING from sqlalchemy import ForeignKey from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.sql.sqltypes import * from files.classes import Base if TYPE_CHECKING: from files.classes import Post, User class Subscription(Base): __tablename__ = "subscriptions" user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), primary_key=True) post_id: Mapped[int] = mapped_column(ForeignKey("posts.id"), primary_key=True) created_utc: Mapped[Optional[int]] user: Mapped["User"] = relationship(uselist=False) post: Mapped["Post"] = relationship(uselist=False) def __init__(self, *args, **kwargs): if "created_utc" not in kwargs: kwargs["created_utc"] = int(time.time()) super().__init__(*args, **kwargs) def __repr__(self): return f"<{self.__class__.__name__}(id={self.id})>"