diff options
author | Miguel Jacq <mig@mig5.net> | 2021-05-04 16:42:23 +1000 |
---|---|---|
committer | Miguel Jacq <mig@mig5.net> | 2021-05-04 16:42:23 +1000 |
commit | c6ccd4de579332187232f0e4bfeb33fe531f76d8 (patch) | |
tree | 6fdba9e7b7080f4f75ed6a8c9678c612891b64fd | |
parent | d8801ff061ef38c339bc106b2b334c9e7989ea0f (diff) | |
download | onionshare-c6ccd4de579332187232f0e4bfeb33fe531f76d8.tar.gz onionshare-c6ccd4de579332187232f0e4bfeb33fe531f76d8.zip |
Remove unnecessary cleanup_filenames appending in the CompressThread. Ensure we also remove the temp dir that a Zip file is made within, by setting the temp dir in ZipWriter as a variable
-rw-r--r-- | cli/onionshare_cli/web/share_mode.py | 4 | ||||
-rw-r--r-- | desktop/src/onionshare/tab/mode/share_mode/threads.py | 3 |
2 files changed, 3 insertions, 4 deletions
diff --git a/cli/onionshare_cli/web/share_mode.py b/cli/onionshare_cli/web/share_mode.py index c5007d7f..95aec1ba 100644 --- a/cli/onionshare_cli/web/share_mode.py +++ b/cli/onionshare_cli/web/share_mode.py @@ -525,6 +525,7 @@ class ShareModeWeb(SendBaseModeWeb): # Make sure the zip file gets cleaned up when onionshare stops self.web.cleanup_filenames.append(self.zip_writer.zip_filename) + self.web.cleanup_filenames.append(self.zip_writer.zip_temp_dir) self.is_zipped = True @@ -545,8 +546,9 @@ class ZipWriter(object): if zip_filename: self.zip_filename = zip_filename else: + self.zip_temp_dir = tempfile.mkdtemp() self.zip_filename = ( - f"{tempfile.mkdtemp()}/onionshare_{self.common.random_string(4, 6)}.zip" + f"{self.zip_temp_dir}/onionshare_{self.common.random_string(4, 6)}.zip" ) self.z = zipfile.ZipFile(self.zip_filename, "w", allowZip64=True) diff --git a/desktop/src/onionshare/tab/mode/share_mode/threads.py b/desktop/src/onionshare/tab/mode/share_mode/threads.py index 5be94e88..839d30ea 100644 --- a/desktop/src/onionshare/tab/mode/share_mode/threads.py +++ b/desktop/src/onionshare/tab/mode/share_mode/threads.py @@ -47,9 +47,6 @@ class CompressThread(QtCore.QThread): self.mode.filenames, processed_size_callback=self.set_processed_size ) self.success.emit() - self.mode.web.cleanup_filenames += ( - self.mode.web.share_mode.cleanup_filenames - ) except OSError as e: self.error.emit(e.strerror) |