small changes
This commit is contained in:
+16
-6
@@ -32,15 +32,25 @@ class ToolkitHandler:
|
||||
if "image" in toolkitData:
|
||||
imageData = toolkitData.pop("image")
|
||||
mainImage = imageData.get("main")
|
||||
imageFileName = handleToolkitImage(mainImage, title)
|
||||
if imageFileName:
|
||||
imageDict["main"] = imageFileName
|
||||
if isinstance(mainImage, str) and mainImage.startswith(
|
||||
"static/images/"
|
||||
):
|
||||
imageDict["main"] = mainImage
|
||||
else:
|
||||
imageFileName = handleToolkitImage(mainImage, title)
|
||||
if imageFileName:
|
||||
imageDict["main"] = imageFileName
|
||||
additionalImages = imageData.get("additional", [])
|
||||
if len(additionalImages) > 0:
|
||||
for image in additionalImages:
|
||||
imageFileName = handleToolkitImage(image, title)
|
||||
if imageFileName:
|
||||
imageDict["additional"].append(imageFileName)
|
||||
if isinstance(image, str) and image.startswith(
|
||||
"static/images/"
|
||||
):
|
||||
imageDict["additional"].append(image)
|
||||
else:
|
||||
imageFileName = handleToolkitImage(image, title)
|
||||
if imageFileName:
|
||||
imageDict["additional"].append(imageFileName)
|
||||
toolkitData["image"] = imageDict
|
||||
newToolkit = await Toolkit(**toolkitData).save()
|
||||
except Exception as e:
|
||||
|
||||
Reference in New Issue
Block a user