43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
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"
|
|
PROMO_CODE = "promo_code"
|
|
|
|
|
|
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"
|
|
)
|