начало БД

db/schemas/stock.py

@@ -0,0 +1,45 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, Float, ForeignKey, Integer
+from sqlalchemy.orm import relationship
+
+from db import Base
+import utils
+
+
+class Stock(Base):
+    __tablename__ = "stocks"
+
+    id = Column(Integer, primary_key=True, index=True)
+    toolkit_id = Column(
+        Integer, ForeignKey("toolkits.id", ondelete="CASCADE"), nullable=False
+    )
+    toolkit_data = relationship(
+        "Toolkit", cascade="all, delete-orphan", lazy="joined", uselist=False
+    )
+    toolbox_id = Column(
+        Integer, ForeignKey("toolboxes.id", ondelete="CASCADE"), nullable=False
+    )
+    toolbox_data = relationship(
+        "Toolbox", cascade="all, delete-orphan", lazy="joined", uselist=False
+    )
+    quantity = Column(Integer, nullable=False)
+    price = Column(Float, nullable=False)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def toDict(self):
+        return utils.toDict(self)
+
+    async def save(self):
+        from db import CRUD
+
+        return await CRUD.create(self, refresh=True)
+
+    async def edit(self, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(Stock, self.id, **kwargs)