61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
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")
|