Merge pull request #53 from 9001/cleanup-args

provide all args for core.cleanup
WoeUSB · Oct 17, 2021 · 4ddbd21 · 4ddbd21
2 parents 7b8f23d + 73d85d2
commit 4ddbd21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/WoeUSB/core.py b/WoeUSB/core.py
@@ -108,7 +108,7 @@ def init(from_cli=True, install_mode=None, source_media=None, target_media=None,
         return [source_fs_mountpoint, target_fs_mountpoint, temp_directory, install_mode, source_media, target_media,
                 workaround_bios_boot_flag, target_filesystem_type, filesystem_label, verbose, debug, parser]
     else:
-        return [source_fs_mountpoint, target_fs_mountpoint, temp_directory]
+        return [source_fs_mountpoint, target_fs_mountpoint, temp_directory, target_media]
 
 
 def main(source_fs_mountpoint, target_fs_mountpoint, source_media, target_media, install_mode, temp_directory,

diff --git a/WoeUSB/gui.py b/WoeUSB/gui.py
@@ -393,7 +393,7 @@ def __init__(self, source, target, boot_flag, filesystem):
         self.filesystem = filesystem
 
     def run(self):
-        source_fs_mountpoint, target_fs_mountpoint, temp_directory = core.init(
+        source_fs_mountpoint, target_fs_mountpoint, temp_directory, target_media = core.init(
             from_cli=False,
             install_mode="device",
             source_media=self.source,
@@ -405,7 +405,7 @@ def run(self):
         except SystemExit:
             pass
 
-        core.cleanup(source_fs_mountpoint, target_fs_mountpoint, temp_directory)
+        core.cleanup(source_fs_mountpoint, target_fs_mountpoint, temp_directory, target_media)
 
 
 def run():