Skip to content

Commit 6a60c69

Browse files
authored
Use existing file collection code for GUI integration
1 parent ff9d15d commit 6a60c69

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

acquire/acquire.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2209,12 +2209,19 @@ def main() -> None:
22092209
flavour = platform.system()
22102210
acquire_gui = GUI(flavour=flavour, upload_available=args.auto_upload)
22112211

2212-
args.output, args.auto_upload, cancel = acquire_gui.wait_for_start(args)
2212+
args.output, files, args.auto_upload, cancel = acquire_gui.wait_for_start(args)
22132213
if cancel:
22142214
log.info("Acquire cancelled")
22152215
exit_success(args.config.get("arguments"))
22162216
# From here onwards, the GUI will be locked and cannot be closed because we're acquiring
22172217

2218+
if files:
2219+
if args.output:
2220+
args.file = files
2221+
else:
2222+
args.upload = files
2223+
args.auto_upload = False
2224+
22182225
plugins_to_load = [("cloud", MinIO)]
22192226
upload_plugins = UploaderRegistry("acquire.plugins", plugins_to_load)
22202227

@@ -2223,6 +2230,10 @@ def main() -> None:
22232230
if args.upload:
22242231
try:
22252232
upload_files(args.upload, args.upload_plugin, args.no_proxy)
2233+
2234+
if acquire_gui:
2235+
acquire_gui.finish()
2236+
acquire_gui.wait_for_quit()
22262237
except Exception:
22272238
acquire_gui.message("Failed to upload files")
22282239
log.exception("")

0 commit comments

Comments
 (0)