Improve out_of_stock_item SQLAlchemy model compliance #3

Clochette-AbsINThe · Oct 25, 2022 · 502b7ae · 502b7ae
1 parent dedd5db
commit 502b7ae
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/app/models/out_of_stock_item.py b/backend/app/models/out_of_stock_item.py
@@ -1,12 +1,15 @@
-from sqlalchemy import Boolean, Column, Integer, String
+from sqlalchemy import Boolean, Column, Enum, Integer, String
 from sqlalchemy.orm import relationship
 
+from app.core.types import IconName
 from app.db.base_class import Base
 
 
 class OutOfStockItem(Base):
     id = Column(Integer, primary_key=True, nullable=False)
+    buy_or_sell = Column(Boolean, nullable=False) # True means buy, False means sell
     name = Column(String, nullable=False)
-    sale = Column(Boolean, nullable=False)  # True is selling, False is buying
+    icon = Column(Enum(IconName), nullable=False)
+    sell_price = Column(Integer, nullable=True)
 
     outofstocks = relationship("OutOfStock", back_populates="item")