cli: add option --same-sudo-password

src/pyinfra_cli/cli.py

@@ -2,6 +2,7 @@
 import sys
 import warnings
 from fnmatch import fnmatch
+from getpass import getpass
 from os import chdir as os_chdir, getcwd, path
 from typing import Iterable, List, Tuple, Union
 
@@ -82,6 +83,12 @@ def _print_support(ctx, param, value):
     envvar="PYINFRA_YES",
     show_envvar=True,
 )
+@click.option(
+    "--same-sudo-password",
+    is_flag=True,
+    default=False,
+    help="All hosts have the same sudo password, so ask only once.",
+)
 @click.option(
     "--limit",
     help="Restrict the target hosts by name and group name.",
@@ -274,6 +281,7 @@ def _main(
     ssh_key,
     ssh_key_password: str,
     ssh_password: str,
+    same_sudo_password: bool,
     shell_executable,
     sudo: bool,
     sudo_user: str,
@@ -344,6 +352,9 @@ def _main(
         ssh_password,
     )
 
+    if same_sudo_password:
+        config.SUDO_PASSWORD = getpass("sudo password: ")
+
     if yes is False:
         _set_fail_prompts(state, config)
 

tests/test_cli/test_cli.py

@@ -170,6 +170,7 @@ def test_deploy_operation_direct(self):
                 ssh_key=None,
                 ssh_key_password=None,
                 ssh_password=None,
+                same_sudo_password=False,
                 sudo=False,
                 sudo_user=None,
                 use_sudo_password=False,