small changes

This commit is contained in:
2025-12-07 21:39:49 +03:00
parent fb49bac5c3
commit 2a04f71e0c
2 changed files with 16 additions and 7 deletions
-1
View File
@@ -145,5 +145,4 @@ async def toolkit_request(
stocksData["toolboxes"][toolboxTitle]["count"] += stock["quantity"] stocksData["toolboxes"][toolboxTitle]["count"] += stock["quantity"]
response["status"] = "ok" response["status"] = "ok"
response["data"] = stocksData response["data"] = stocksData
logger.info(response)
return response return response
+16 -6
View File
@@ -32,15 +32,25 @@ class ToolkitHandler:
if "image" in toolkitData: if "image" in toolkitData:
imageData = toolkitData.pop("image") imageData = toolkitData.pop("image")
mainImage = imageData.get("main") mainImage = imageData.get("main")
imageFileName = handleToolkitImage(mainImage, title) if isinstance(mainImage, str) and mainImage.startswith(
if imageFileName: "static/images/"
imageDict["main"] = imageFileName ):
imageDict["main"] = mainImage
else:
imageFileName = handleToolkitImage(mainImage, title)
if imageFileName:
imageDict["main"] = imageFileName
additionalImages = imageData.get("additional", []) additionalImages = imageData.get("additional", [])
if len(additionalImages) > 0: if len(additionalImages) > 0:
for image in additionalImages: for image in additionalImages:
imageFileName = handleToolkitImage(image, title) if isinstance(image, str) and image.startswith(
if imageFileName: "static/images/"
imageDict["additional"].append(imageFileName) ):
imageDict["additional"].append(image)
else:
imageFileName = handleToolkitImage(image, title)
if imageFileName:
imageDict["additional"].append(imageFileName)
toolkitData["image"] = imageDict toolkitData["image"] = imageDict
newToolkit = await Toolkit(**toolkitData).save() newToolkit = await Toolkit(**toolkitData).save()
except Exception as e: except Exception as e: