Add --no-ssh awareness to autocompleter

cisagov · Aug 16, 2022 · 551c99a · 551c99a
1 parent c48c3c4
commit 551c99a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/awssh/autocompleter.py b/src/awssh/autocompleter.py
@@ -18,6 +18,7 @@
 HELP_OPTIONS = {"--help", "-h"}
 LOG_LEVEL_OPTIONS = {"--log-level"}
 LOG_LEVELS = {"debug", "info", "warning", "error", "critical"}
+NO_SSH_OPTIONS = {"--no-ssh", "-n"}
 REGION_OPTIONS = {"--region", "-r"}
 SSH_ARGS_OPTIONS = {"--ssh-args", "-s"}
 
@@ -141,6 +142,8 @@ def build_option_candidates(word_set: set[str]) -> set[str]:
         candidates |= CREDENTIALS_OPTIONS
     if not LOG_LEVEL_OPTIONS & word_set:
         candidates |= LOG_LEVEL_OPTIONS
+    if not NO_SSH_OPTIONS & word_set:
+        candidates |= NO_SSH_OPTIONS
     if not REGION_OPTIONS & word_set:
         candidates |= REGION_OPTIONS
     if not SSH_ARGS_OPTIONS & word_set: