Remove the sudo prompt message from the stdout stream when using sudo

dwighthubbard · Jul 24, 2013 · ba01265 · ba01265
1 parent e58a1d7
commit ba01265
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/sshmap/defaults.py b/sshmap/defaults.py
@@ -57,3 +57,11 @@
     "port": "6667",
     "use_ssl": False,
 }
+
+sudo_message = [
+    'We trust you have received the usual lecture from the local System',
+    'Administrator. It usually boils down to these three things:',
+    '#1) Respect the privacy of others.',
+    '#2) Think before you type.',
+    '#3) With great power comes great responsibility.'
+]
diff --git a/sshmap/sshmap.py b/sshmap/sshmap.py
@@ -391,7 +391,8 @@ def run_command(host, command="uname -a", username=None, password=None,
             for el in result.err:
                 if check_prompt:
                     if password in el or 'assword:' in el or \
-                            '[sudo] password' in el or el.strip() == '':
+                            '[sudo] password' in el or el.strip() == '' or \
+                            el.strip() in defaults.sudo_message:
                         skip = True
                     else:
                         check_prompt = False