склади и инструмент готовы

db/handlers/toolkit.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from utils import logger, saveImage, safeFilename, deleteImage
 from db import CRUD
 from db.schemas.toolkit import Toolkit
@@ -27,7 +28,7 @@ class ToolkitHandler:
             return {}
 
         try:
-            imageDict = {"main": "images/tools/default.png", "additional": []}
+            imageDict = {"main": "static/images/tools/default.png", "additional": []}
             if "image" in toolkitData:
                 imageData = toolkitData.pop("image")
                 mainImage = imageData.get("main")
@@ -54,6 +55,21 @@ class ToolkitHandler:
         await ServiceRecordsHandler.add(user_id, {"Добавлен инструмент": toolkitData})
         return newToolkit
 
+    async def updateMovindDate(toolkitId: int):
+        editedToolkit = await ToolkitHandler.edit(toolkitId, moved_at=datetime.now())
+        if not editedToolkit:
+            logger.error("Инструмент не обновлен")
+            return False
+        return True
+
+    async def updateRefillDate(toolkitId: int):
+        logger.info(f"Обновление даты пополнения инструмента {toolkitId}...")
+        editedToolkit = await ToolkitHandler.edit(toolkitId, refilled_at=datetime.now())
+        if not editedToolkit:
+            logger.error("Инструмент не обновлен")
+            return False
+        return True
+
     async def edit(toolkitId: int, **kwargs):
         query = select(Toolkit).where(Toolkit.id == toolkitId)
         toolkit = await CRUD.read(query)
@@ -98,6 +114,7 @@ class ToolkitHandler:
 
                 kwargs["image"] = imageDict
             user_id = kwargs.pop("user_id", None)
+            logger.debug(f"Обновление инструмента {toolkit.title}...")
             editedToolkit = await toolkit.edit(**kwargs)
         except Exception as e:
             logger.error(f"Ошибка обновления инструмента: {str(e)}")
@@ -208,6 +225,13 @@ class ToolkitHandler:
                     "Радиус": "0.4",
                     "Ещё что-то": "Ещё столько-то",
                 },
+                "image": {
+                    "main": "static/images/tools/default.png",
+                    "additional": [
+                        "static/images/users/default.png",
+                        "static/images/logo.png",
+                    ],
+                },
                 "category_id": categories["Токарка"],
                 "quantity_min": 20,
                 "quantity_min_extra": 10,