Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(app): delete images instead of using send2trash

Browse Source 
Closes #6709
This commit is contained in:
psychedelicious 2024-08-03 20:17:55 +10:00
parent 571ba87e13
commit 08993c0d29
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/image_files/image_files_disk.py
View File

@ -5,7 +5,6 @@ from typing import Dict, Optional, Union
from PIL import Image, PngImagePlugin from PIL import Image, PngImagePlugin
from PIL.Image import Image as PILImageType from PIL.Image import Image as PILImageType
from send2trash import send2trash
from invokeai.app.services.image_files.image_files_base import ImageFileStorageBase from invokeai.app.services.image_files.image_files_base import ImageFileStorageBase
from invokeai.app.services.image_files.image_files_common import ( from invokeai.app.services.image_files.image_files_common import (
@ -103,7 +102,7 @@ class DiskImageFileStorage(ImageFileStorageBase):
image_path = self.get_path(image_name) image_path = self.get_path(image_name)
if image_path.exists(): if image_path.exists():
send2trash(image_path) image_path.unlink()
if image_path in self.__cache: if image_path in self.__cache:
del self.__cache[image_path] del self.__cache[image_path]
@ -111,7 +110,7 @@ class DiskImageFileStorage(ImageFileStorageBase):
thumbnail_path = self.get_path(thumbnail_name, True) thumbnail_path = self.get_path(thumbnail_name, True)
if thumbnail_path.exists(): if thumbnail_path.exists():
send2trash(thumbnail_path) thumbnail_path.unlink()
if thumbnail_path in self.__cache: if thumbnail_path in self.__cache:
del self.__cache[thumbnail_path] del self.__cache[thumbnail_path]
except Exception as e: except Exception as e: