40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from datetime import datetime
|
|
from sqlalchemy import DateTime, ForeignKey, Integer, Boolean
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from typing import TYPE_CHECKING
|
|
|
|
from app.core.database import Base
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.user import User
|
|
from app.models.shop import ShopItem
|
|
|
|
|
|
class UserInventory(Base):
|
|
__tablename__ = "user_inventory"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
item_id: Mapped[int] = mapped_column(ForeignKey("shop_items.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
quantity: Mapped[int] = mapped_column(Integer, default=1)
|
|
equipped: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
purchased_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
expires_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
|
|
# Relationships
|
|
user: Mapped["User"] = relationship(
|
|
"User",
|
|
back_populates="inventory"
|
|
)
|
|
item: Mapped["ShopItem"] = relationship(
|
|
"ShopItem",
|
|
back_populates="inventory_items"
|
|
)
|
|
|
|
@property
|
|
def is_expired(self) -> bool:
|
|
"""Check if item has expired"""
|
|
if self.expires_at is None:
|
|
return False
|
|
return datetime.utcnow() > self.expires_at
|