ai_giga_tcg/app/models/inventory_management.py

from sqlalchemy import Column, Integer, String, Float, DateTime, ForeignKey, Table
from sqlalchemy.orm import relationship
from app.db.database import Base

class PhysicalItem(Base):
    __tablename__ = "physical_items"
    
    id = Column(Integer, primary_key=True)
    item_type = Column(String)
    product_id = Column(Integer, ForeignKey("products.id"), nullable=False)
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)
    
    __mapper_args__ = {
        'polymorphic_on': item_type,
        'polymorphic_identity': 'physical_item'
    }

    # Relationships
    product = relationship("Product")
    inventory_item = relationship("InventoryItem", uselist=False, back_populates="physical_item")
    transaction_items = relationship("TransactionItem", back_populates="physical_item")

class SealedCase(PhysicalItem):
    __tablename__ = "sealed_cases"
    
    id = Column(Integer, ForeignKey('physical_items.id'), primary_key=True)
    
    __mapper_args__ = {
        'polymorphic_identity': 'sealed_case'
    }

    # Relationships
    boxes = relationship("SealedBox", back_populates="case")
    open_event = relationship("OpenEvent", uselist=False, back_populates="sealed_case")

class SealedBox(PhysicalItem):
    __tablename__ = "sealed_boxes"
    
    id = Column(Integer, ForeignKey('physical_items.id'), primary_key=True)
    case_id = Column(Integer, ForeignKey("sealed_cases.id"), nullable=True)
    
    __mapper_args__ = {
        'polymorphic_identity': 'sealed_box'
    }

    # Relationships
    case = relationship("SealedCase", back_populates="boxes")
    open_event = relationship("OpenEvent", uselist=False, back_populates="sealed_box")

class OpenBox(PhysicalItem):
    __tablename__ = "open_boxes"
    
    id = Column(Integer, ForeignKey('physical_items.id'), primary_key=True)
    open_event_id = Column(Integer, ForeignKey("open_events.id"))
    sealed_box_id = Column(Integer, ForeignKey("sealed_boxes.id"))
    
    __mapper_args__ = {
        'polymorphic_identity': 'open_box'
    }

    # Relationships
    open_event = relationship("OpenEvent", back_populates="resulting_boxes")
    sealed_box = relationship("SealedBox")
    cards = relationship("OpenCard", back_populates="box")

class OpenCard(PhysicalItem):
    __tablename__ = "open_cards"
    
    id = Column(Integer, ForeignKey('physical_items.id'), primary_key=True)
    open_event_id = Column(Integer, ForeignKey("open_events.id"))
    box_id = Column(Integer, ForeignKey("open_boxes.id"), nullable=True)
    
    __mapper_args__ = {
        'polymorphic_identity': 'open_card'
    }

    # Relationships
    open_event = relationship("OpenEvent", back_populates="resulting_cards")
    box = relationship("OpenBox", back_populates="cards")

class InventoryItem(Base):
    __tablename__ = "inventory_items"

    id = Column(Integer, primary_key=True, index=True)
    physical_item_id = Column(Integer, ForeignKey("physical_items.id"), unique=True)
    cost_basis = Column(Float)  # Current cost basis for this item
    parent_id = Column(Integer, ForeignKey("inventory_items.id"), nullable=True)  # For tracking hierarchy
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)

    # Relationships
    physical_item = relationship("PhysicalItem", back_populates="inventory_item")
    parent = relationship("InventoryItem", remote_side=[id])
    children = relationship("InventoryItem")

class TransactionItem(Base):
    __tablename__ = "transaction_items"

    id = Column(Integer, primary_key=True, index=True)
    transaction_id = Column(Integer, ForeignKey("transactions.id"))
    physical_item_id = Column(Integer, ForeignKey("physical_items.id"))
    unit_price = Column(Float, nullable=False)
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)

    # Relationships
    transaction = relationship("Transaction", back_populates="transaction_items")
    physical_item = relationship("PhysicalItem", back_populates="transaction_items")

class OpenEvent(Base):
    __tablename__ = "open_events"

    id = Column(Integer, primary_key=True, index=True)
    sealed_case_id = Column(Integer, ForeignKey("sealed_cases.id"), nullable=True)
    sealed_box_id = Column(Integer, ForeignKey("sealed_boxes.id"), nullable=True)
    open_date = Column(DateTime(timezone=True))
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)

    # Relationships
    sealed_case = relationship("SealedCase", back_populates="open_event")
    sealed_box = relationship("SealedBox", back_populates="open_event")
    resulting_boxes = relationship("OpenBox", back_populates="open_event")
    resulting_cards = relationship("OpenCard", back_populates="open_event")

class Vendor(Base):
    __tablename__ = "vendors"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, unique=True, index=True)
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)

class Customer(Base):
    __tablename__ = "customers"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, unique=True, index=True)
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))

class Transaction(Base):
    __tablename__ = "transactions"

    id = Column(Integer, primary_key=True, index=True)
    vendor_id = Column(Integer, ForeignKey("vendors.id"), nullable=True)
    customer_id = Column(Integer, ForeignKey("customers.id"), nullable=True)
    transaction_type = Column(String)  # 'purchase' or 'sale'
    transaction_date = Column(DateTime(timezone=True))
    transaction_total_amount = Column(Float)
    transaction_notes = Column(String)
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)

    # Relationships
    transaction_items = relationship("TransactionItem", back_populates="transaction")

class CostBasis(Base):
    __tablename__ = "cost_basis"

    id = Column(Integer, primary_key=True, index=True)
    transaction_item_id = Column(Integer, ForeignKey("transaction_items.id"))
    sealed_case_id = Column(Integer, ForeignKey("sealed_cases.id"), nullable=True)
    sealed_box_id = Column(Integer, ForeignKey("sealed_boxes.id"), nullable=True)
    open_box_id = Column(Integer, ForeignKey("open_boxes.id"), nullable=True)
    open_card_id = Column(Integer, ForeignKey("open_cards.id"), nullable=True)
    quantity = Column(Integer, nullable=False, default=1)
    unit_cost = Column(Float, nullable=False)
    created_at = Column(DateTime(timezone=True))
    updated_at = Column(DateTime(timezone=True))
    deleted_at = Column(DateTime(timezone=True), nullable=True)

    # Relationships
    transaction_item = relationship("TransactionItem")
    sealed_case = relationship("SealedCase")
    sealed_box = relationship("SealedBox")
    open_box = relationship("OpenBox")
    open_card = relationship("OpenCard")