almost

src/ytdl_sub/downloaders/downloader.py

@@ -439,7 +439,7 @@ class Downloader(DownloadArchiver, Generic[DownloaderOptionsT], ABC):
                 if path.endswith(".info.json")
             ]
             for info_json_file in info_json_files:
-                os.remove(info_json_file)
+                FileHandler.delete(info_json_file)
 
     def _extract_entry_info_with_retry(self, entry: Entry) -> Entry:
         download_entry_dict = self.extract_info_with_retry(

src/ytdl_sub/utils/ffmpeg.py

@@ -140,6 +140,7 @@ def set_ffmpeg_metadata_chapters(
         metadata_file.write("\n".join(lines))
         metadata_file.flush()
 
+    try:
         FFMPEG.run(
             [
                 "-i",
@@ -157,7 +158,8 @@ def set_ffmpeg_metadata_chapters(
             ]
         )
         FileHandler.move(tmp_file_path, file_path)
-    os.remove(metadata_file.name)
+    finally:
+        FileHandler.delete(metadata_file.name)
 
 
 def add_ffmpeg_metadata_key_values(file_path: str, key_values: Dict[str, str]) -> None:

src/ytdl_sub/utils/thumbnail.py

@@ -67,6 +67,7 @@ def convert_url_thumbnail(thumbnail_url: str, output_thumbnail_path: str) -> Opt
         with tempfile.NamedTemporaryFile(delete=False) as thumbnail:
             thumbnail.write(file.read())
 
+        try:
             os.makedirs(os.path.dirname(output_thumbnail_path), exist_ok=True)
 
             tmp_output_path = FFMPEG.tmp_file_path(
@@ -76,8 +77,8 @@ def convert_url_thumbnail(thumbnail_url: str, output_thumbnail_path: str) -> Opt
 
             # Have FileHandler handle the move to a potential cross-device
             FileHandler.move(tmp_output_path, output_thumbnail_path)
+        finally:
             FileHandler.delete(tmp_output_path)
-
+            FileHandler.delete(thumbnail.name)
-        os.remove(thumbnail.name)
 
     return True

tests/conftest.py

@@ -1,6 +1,7 @@
 import contextlib
 import json
 import logging
+import os
 import tempfile
 from pathlib import Path
 from typing import Any
@@ -11,6 +12,7 @@ from unittest.mock import patch
 
 import pytest
 
+from ytdl_sub.utils.file_handler import FileHandler
 from ytdl_sub.utils.logger import Logger
 
 
@@ -84,9 +86,12 @@ def preset_dict_to_subscription_yaml_generator() -> Callable:
     @contextlib.contextmanager
     def _preset_dict_to_subscription_yaml_generator(subscription_name: str, preset_dict: Dict):
         subscription_dict = {subscription_name: preset_dict}
-        with tempfile.NamedTemporaryFile(suffix=".yaml") as tmp_file:
+        with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as tmp_file:
             tmp_file.write(json.dumps(subscription_dict).encode("utf-8"))
-            tmp_file.flush()
+
+        try:
             yield tmp_file.name
+        finally:
+            FileHandler.delete(tmp_file.name)
 
     return _preset_dict_to_subscription_yaml_generator

tests/e2e/conftest.py

@@ -15,7 +15,7 @@ from ytdl_sub.cli.main import main
 from ytdl_sub.config.config_file import ConfigFile
 from ytdl_sub.subscriptions.subscription import Subscription
 from ytdl_sub.subscriptions.subscription_download import SubscriptionDownload
-from ytdl_sub.utils.file_handler import FileHandlerTransactionLog
+from ytdl_sub.utils.file_handler import FileHandlerTransactionLog, FileHandler
 from ytdl_sub.utils.logger import Logger
 from ytdl_sub.utils.yaml import load_yaml
 
@@ -67,10 +67,13 @@ def music_video_config(music_video_config_path, working_directory) -> ConfigFile
 
 @pytest.fixture()
 def music_video_config_for_cli(music_video_config) -> str:
-    with tempfile.NamedTemporaryFile(suffix=".yaml") as tmp_file:
+    with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as tmp_file:
         tmp_file.write(json.dumps(music_video_config._value).encode("utf-8"))
-        tmp_file.flush()
+
+    try:
         yield tmp_file.name
+    finally:
+        FileHandler.delete(tmp_file.name)
 
 
 @pytest.fixture()
@@ -98,10 +101,13 @@ def timestamps_file_path():
         "00:01:01 Part 5\n",
     ]
 
-    with tempfile.NamedTemporaryFile(mode="w", encoding="utf-8", suffix=".txt") as tmp:
+    with tempfile.NamedTemporaryFile(mode="w", encoding="utf-8", suffix=".txt", delete=False) as tmp:
         tmp.writelines(timestamps)
-        tmp.seek(0)
+
+    try:
         yield tmp.name
+    finally:
+        FileHandler.delete(tmp.name)
 
 
 def mock_run_from_cli(args: str) -> List[Tuple[Subscription, FileHandlerTransactionLog]]:

tests/unit/utils/test_yaml.py

@@ -6,6 +6,7 @@ import pytest
 
 from ytdl_sub.utils.exceptions import FileNotFoundException
 from ytdl_sub.utils.exceptions import InvalidYamlException
+from ytdl_sub.utils.file_handler import FileHandler
 from ytdl_sub.utils.yaml import load_yaml
 
 
@@ -28,8 +29,10 @@ def bad_yaml_file_path(bad_yaml) -> str:
         tmp_file.write(bad_yaml.encode("utf-8"))
         tmp_file.flush()
 
+    try:
         yield tmp_file.name
-    os.remove(tmp_file.name)
+    finally:
+        FileHandler.delete(tmp_file.name)
 
 
 def test_load_yaml_file_not_found():