From 0d2166db3a4e9f0e7d4ef44673c0067da8d99648 Mon Sep 17 00:00:00 2001 From: g5fighter Date: Tue, 4 Jul 2023 14:16:57 +0200 Subject: [PATCH] Add temp file cleanup, and protect on saving problems --- Model/ImageWriter.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/Model/ImageWriter.py b/Model/ImageWriter.py index 9852523..0805605 100644 --- a/Model/ImageWriter.py +++ b/Model/ImageWriter.py @@ -1,4 +1,5 @@ from iptcinfo3 import IPTCInfo +import os class ImageWriter: def writeTagsFromPredictionsInImages(self, predictions): @@ -14,5 +15,10 @@ def writeTagsFromPredictionsInImages(self, predictions): for category in prediction.m_categories: if category not in iptcInfo['keywords']: iptcInfo['keywords'].append(category) - - iptcInfo.save() \ No newline at end of file + try: + iptcInfo.save() + tempFilename = filename + "~" + if os.path.exists(tempFilename): + os.remove(tempFilename) + except Exception as e: + print('Error in file "'+filename+'" : \n\t', e) \ No newline at end of file