начало БД

db/schemas/toolkit.py

@@ -0,0 +1,45 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, Text
+from sqlalchemy.dialects.postgresql import JSONB
+from sqlalchemy.orm import relationship
+from db import Base
+import utils
+
+
+class Toolkit(Base):
+    __tablename__ = "toolkits"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String, unique=True, index=True)
+    description = Column(Text)
+    category_id = Column(Integer, ForeignKey("categories.id", ondelete="CASCADE"))
+    category_data = relationship(
+        "Category", cascade="all, delete-orphan", lazy="joined", uselist=False
+    )
+    image = Column(
+        JSONB, default={"main": "images/tools/default.png", "additional": []}
+    )
+    quantity_min = Column(Integer, nullable=True)
+    quantity_min_extra = Column(Integer, nullable=True)
+    external_link = Column(String, nullable=True)
+    created_at = Column(DateTime, default=datetime.now)
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
+    refilled_at = Column(DateTime, default=datetime.now)
+    moved_at = Column(DateTime, default=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(id: int, **kwargs):
+        from db import CRUD
+
+        return await CRUD.update(Toolkit, id, **kwargs)