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

tidy(bulk_download): clean up comments

Browse Source
This commit is contained in:
psychedelicious 2024-02-20 17:47:04 +11:00
parent e51867756a
commit 7f8f182a00
2 changed files with 2 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/services/bulk_download/bulk_download_base.py
View File

@ -27,7 +27,7 @@ class BulkDownloadBase(ABC):
self, image_names: Optional[list[str]], board_id: Optional[str], bulk_download_item_id: Optional[str]
) -> None:
"""
Starts a a bulk download job.
Create a zip file containing the images specified by the given image names or board id.
:param image_names: A list of image names to include in the zip file.
:param board_id: The ID of the board. If provided, all images associated with the board will be included in the zip file.
@ -54,15 +54,7 @@ class BulkDownloadBase(ABC):
@abstractmethod
def stop(self, *args, **kwargs) -> None:
"""
Stops the BulkDownloadService and cleans up all the remnants.
This method is responsible for stopping the BulkDownloadService and performing any necessary cleanup
operations to remove any remnants or resources associated with the service.
:param *args: Variable length argument list.
:param **kwargs: Arbitrary keyword arguments.
"""
"""Stops the BulkDownloadService and cleans up."""
@abstractmethod
def delete(self, bulk_download_item_name: str) -> None:

23
invokeai/app/services/bulk_download/bulk_download_default.py
View File

@ -30,9 +30,6 @@ class BulkDownloadService(BulkDownloadBase):
self.__event_bus = invoker.services.events
def __init__(self):
"""
Initialize the downloader object.
"""
self.__temp_directory = TemporaryDirectory()
self.__bulk_downloads_folder = Path(self.__temp_directory.name) / "bulk_downloads"
self.__bulk_downloads_folder.mkdir(parents=True, exist_ok=True)
@ -40,13 +37,6 @@ class BulkDownloadService(BulkDownloadBase):
def handler(
self, image_names: Optional[list[str]], board_id: Optional[str], bulk_download_item_id: Optional[str]
) -> None:
"""
Create a zip file containing the images specified by the given image names or board id.
param: image_names: A list of image names to include in the zip file.
param: board_id: The ID of the board. If provided, all images associated with the board will be included in the zip file.
"""
bulk_download_id: str = DEFAULT_BULK_DOWNLOAD_ID
bulk_download_item_id = uuid_string() if bulk_download_item_id is None else bulk_download_item_id
bulk_download_item_name = bulk_download_item_id + ".zip"
@ -162,26 +152,13 @@ class BulkDownloadService(BulkDownloadBase):
)
def stop(self, *args, **kwargs):
"""Stop the bulk download service and delete the files in the bulk download folder."""
# Get all the files in the bulk downloads folder, only .zip files
self.__temp_directory.cleanup()
def delete(self, bulk_download_item_name: str) -> None:
"""
Delete the bulk download file.
:param bulk_download_item_name: The name of the bulk download item.
"""
path = self.get_path(bulk_download_item_name)
Path(path).unlink()
def get_path(self, bulk_download_item_name: str) -> str:
"""
Get the path to the bulk download file.
:param bulk_download_item_name: The name of the bulk download item.
:return: The path to the bulk download file.
"""
path = str(self.__bulk_downloads_folder / bulk_download_item_name)
if not self._is_valid_path(path):
raise BulkDownloadTargetException()