29 lines
939 B
Python
29 lines
939 B
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import DateTime, ForeignKey, Integer, UniqueConstraint, func
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base
|
|
|
|
|
|
class Favorite(Base):
|
|
__tablename__ = "favorites"
|
|
__table_args__ = (UniqueConstraint("user_id", "spot_id", name="uq_user_spot"),)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(
|
|
Integer, ForeignKey("users.id"), nullable=False
|
|
)
|
|
spot_id: Mapped[int] = mapped_column(
|
|
Integer, ForeignKey("spots.id"), nullable=False
|
|
)
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now()
|
|
)
|
|
|
|
user = relationship("User", lazy="selectin")
|
|
spot = relationship("Spot", lazy="selectin")
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<Favorite user={self.user_id} spot={self.spot_id}>"
|