from datetime import datetime from enum import Enum from sqlalchemy import String, Text, DateTime, ForeignKey, Integer 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 class CoinTransactionType(str, Enum): CHALLENGE_COMPLETE = "challenge_complete" PLAYTHROUGH_COMPLETE = "playthrough_complete" MARATHON_WIN = "marathon_win" MARATHON_PLACE = "marathon_place" COMMON_ENEMY_BONUS = "common_enemy_bonus" PURCHASE = "purchase" REFUND = "refund" ADMIN_GRANT = "admin_grant" ADMIN_DEDUCT = "admin_deduct" class CoinTransaction(Base): __tablename__ = "coin_transactions" id: Mapped[int] = mapped_column(primary_key=True) user_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True) amount: Mapped[int] = mapped_column(Integer, nullable=False) transaction_type: Mapped[str] = mapped_column(String(30), nullable=False) reference_type: Mapped[str | None] = mapped_column(String(30), nullable=True) reference_id: Mapped[int | None] = mapped_column(Integer, nullable=True) description: Mapped[str | None] = mapped_column(Text, nullable=True) created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow) # Relationships user: Mapped["User"] = relationship( "User", back_populates="coin_transactions" )