49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
|
from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime, func
|
||
|
|
from sqlalchemy.orm import relationship
|
||
|
|
from .database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class User(Base):
|
||
|
|
__tablename__ = "users"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
|
|
username = Column(String(50), unique=True, nullable=False, index=True)
|
||
|
|
password_hash = Column(String(255), nullable=False)
|
||
|
|
role = Column(String(20), nullable=False, default="user") # "admin" or "user"
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
|
||
|
|
|
||
|
|
class Set(Base):
|
||
|
|
__tablename__ = "sets"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
|
|
code = Column(String(10), unique=True, nullable=False, index=True) # e.g. "CRI"
|
||
|
|
name = Column(String(200), nullable=False) # e.g. "Chaos Rising"
|
||
|
|
release_date = Column(String(20), nullable=True)
|
||
|
|
description = Column(Text, nullable=True)
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
|
||
|
|
cards = relationship("Card", back_populates="set", cascade="all, delete-orphan",
|
||
|
|
order_by="Card.number")
|
||
|
|
|
||
|
|
|
||
|
|
class Card(Base):
|
||
|
|
__tablename__ = "cards"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
|
|
set_id = Column(Integer, ForeignKey("sets.id", ondelete="CASCADE"), nullable=False, index=True)
|
||
|
|
number = Column(String(10), nullable=False) # e.g. "068"
|
||
|
|
name = Column(String(200), nullable=False) # e.g. "Garchomp EX"
|
||
|
|
type = Column(String(50), nullable=True) # e.g. "Colorless", "Fire"
|
||
|
|
rarity = Column(String(50), nullable=True) # e.g. "Rare", "Uncommon"
|
||
|
|
hp = Column(Integer, nullable=True)
|
||
|
|
stage = Column(String(50), nullable=True) # e.g. "Basic", "Stage 1", "Stage 2"
|
||
|
|
description = Column(Text, nullable=True)
|
||
|
|
image_url = Column(String(500), nullable=True)
|
||
|
|
|
||
|
|
set = relationship("Set", back_populates="cards")
|
||
|
|
|
||
|
|
@property
|
||
|
|
def full_code(self):
|
||
|
|
return f"{self.set.code}-{self.number}"
|