danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[BUGFIX] Clear working directory before starting any downloads

Browse source
This commit is contained in:
Jesse Bannon 2022-11-30 14:08:58 -08:00
parent 96e741c71d
commit df88ae99ba
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/ytdl_sub/subscriptions/subscription_download.py
View file

@ -95,9 +95,10 @@ class SubscriptionDownload(BaseSubscription, ABC):
entry=entry,
)
def _delete_working_directory(self, is_error: bool) -> None:
def _delete_working_directory(self, is_error: bool = False) -> None:
_ = is_error
shutil.rmtree(self.working_directory)
if os.path.isdir(self.working_directory):
shutil.rmtree(self.working_directory)
@contextlib.contextmanager
def _prepare_working_directory(self):
@ -105,6 +106,7 @@ class SubscriptionDownload(BaseSubscription, ABC):
Context manager to create all directories to the working directory. Deletes the entire
working directory when cleaning up.
"""
self._delete_working_directory()
os.makedirs(self.working_directory, exist_ok=True)
try:
@ -113,7 +115,7 @@ class SubscriptionDownload(BaseSubscription, ABC):
self._delete_working_directory(is_error=True)
raise exc
else:
self._delete_working_directory(is_error=False)
self._delete_working_directory()
@contextlib.contextmanager
def _maintain_archive_file(self):