import datetime from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Text, Float from sqlalchemy.orm import relationship from .database import Base class User(Base): __tablename__ = "users" id = Column(Integer, primary_key=True, index=True) username = Column(String, unique=True, index=True) role = Column(String, default="user") # 'admin' or 'user' audit_logs = relationship("AuditLog", back_populates="user") class Item(Base): __tablename__ = "items" id = Column(Integer, primary_key=True, index=True) barcode = Column(String, unique=True, index=True) name = Column(String, index=True) category = Column(String, index=True) quantity = Column(Float, default=0.0) min_quantity = Column(Float, default=1.0) image_url = Column(String, nullable=True) # Store labels template extracted data simply as JSON text for now labels_data = Column(Text, nullable=True) class AuditLog(Base): __tablename__ = "audit_logs" id = Column(Integer, primary_key=True, index=True) timestamp = Column(DateTime, default=datetime.datetime.utcnow) user_id = Column(Integer, ForeignKey("users.id")) action = Column(String) # e.g., 'CHECK_IN', 'CHECK_OUT', 'CREATE_ITEM' target_item_id = Column(Integer, nullable=True) quantity_change = Column(Float, nullable=True) user = relationship("User", back_populates="audit_logs") class Intervention(Base): __tablename__ = "interventions" id = Column(Integer, primary_key=True, index=True) name = Column(String) status = Column(String, default="ACTIVE") created_at = Column(DateTime, default=datetime.datetime.utcnow) items = relationship("InterventionItem", back_populates="intervention") class InterventionItem(Base): __tablename__ = "intervention_items" id = Column(Integer, primary_key=True, index=True) intervention_id = Column(Integer, ForeignKey("interventions.id")) item_id = Column(Integer, ForeignKey("items.id")) required_quantity = Column(Float, default=1.0) checked_out_quantity = Column(Float, default=0.0) intervention = relationship("Intervention", back_populates="items")