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})>"