Skip to content

Commit 884e5fd

Browse files
authored
Fix module disable (#254)
1 parent f12416b commit 884e5fd

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

acquire/acquire.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1900,13 +1900,16 @@ def acquire_target(target: Target, args: argparse.Namespace, output_ts: str | No
19001900
print_acquire_warning(target)
19011901

19021902
modules_selected = {}
1903+
modules_disabled = []
19031904
modules_successful = []
19041905
modules_failed = {}
19051906
for name, mod in MODULES.items():
19061907
name_slug = name.lower()
19071908
# check if module was set in the arguments provided
1908-
if getattr(args, name_slug):
1909+
if (mod_arg := getattr(args, name_slug)) is True:
19091910
modules_selected[name] = mod
1911+
elif mod_arg is False:
1912+
modules_disabled.append(name)
19101913

19111914
profile = args.profile
19121915

@@ -1929,6 +1932,10 @@ def acquire_target(target: Target, args: argparse.Namespace, output_ts: str | No
19291932
)
19301933
modules_selected.update(volatile_modules)
19311934

1935+
# Filter modules that are explicitly disabled
1936+
for name in modules_disabled:
1937+
modules_selected.pop(name, None)
1938+
19321939
if not modules_selected:
19331940
log.warning("NO modules selected!")
19341941
else:

0 commit comments

Comments
 (0)