57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from sqlalchemy import Column, Integer, String, Float, Boolean, ForeignKey, DateTime, Text
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.sql import func
|
|
|
|
Base = declarative_base()
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String(50), nullable=True)
|
|
phone = Column(String(20), nullable=True)
|
|
address = Column(Text, nullable=True)
|
|
created_at = Column(DateTime, default=func.now())
|
|
|
|
def __repr__(self):
|
|
return f"<User(id={self.id}, name={self.name})>"
|
|
|
|
class Prize(Base):
|
|
__tablename__ = "prizes"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String(100))
|
|
description = Column(Text, nullable=True)
|
|
probability = Column(Float, default=0)
|
|
available_quantity = Column(Integer, default=0)
|
|
is_active = Column(Boolean, default=True)
|
|
|
|
def __repr__(self):
|
|
return f"<Prize(id={self.id}, name={self.name}, probability={self.probability})>"
|
|
|
|
class Card(Base):
|
|
__tablename__ = "cards"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
card_id = Column(String(50), unique=True)
|
|
card_type = Column(String(20))
|
|
is_collected = Column(Boolean, default=False)
|
|
collected_by = Column(Integer, ForeignKey("users.id"), nullable=True)
|
|
collected_at = Column(DateTime, nullable=True)
|
|
|
|
def __repr__(self):
|
|
return f"<Card(id={self.id}, card_id={self.card_id}, collected={self.is_collected})>"
|
|
|
|
class UserPrize(Base):
|
|
__tablename__ = "user_prizes"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"))
|
|
prize_id = Column(Integer, ForeignKey("prizes.id"))
|
|
awarded_at = Column(DateTime, default=func.now())
|
|
is_shipped = Column(Boolean, default=False)
|
|
shipped_at = Column(DateTime, nullable=True)
|
|
|
|
def __repr__(self):
|
|
return f"<UserPrize(id={self.id}, user_id={self.user_id}, prize_id={self.prize_id})>"
|