From bf0a8c8256c3e9345c24c93cdcf77bcd3fd54f98 Mon Sep 17 00:00:00 2001 From: Max Meinhold Date: Thu, 20 May 2021 17:23:49 -0400 Subject: [PATCH] Update directory thumbnails when deleting images --- gallery/__init__.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/gallery/__init__.py b/gallery/__init__.py index 3f835c1..773bd90 100644 --- a/gallery/__init__.py +++ b/gallery/__init__.py @@ -458,6 +458,13 @@ def delete_file(file_id: int, auth_dict: Optional[Dict[str, Any]] = None): current_tags = Tag.query.filter(Tag.file_id == file_id).all() for tag in current_tags: db.session.delete(tag) + + # Remove image from directory thumbnails + dirs = Directory.query.filter(or_(Directory.thumbnail_uuid == file_model.thumbnail_uuid, \ + Directory.thumbnail_uuid == file_model.thumbnail_uuid[:-4])) + for d in dirs: + d.thumbnail_uuid = refresh_directory_thumbnail(d) + db.session.delete(file_model) db.session.flush() db.session.commit()