Make sure host clients dict has key for host when running command

ParallelSSH · Feb 26, 2016 · de75c7b · de75c7b
1 parent 6603c13
commit de75c7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pssh/pssh_client.py b/pssh/pssh_client.py
@@ -336,7 +336,7 @@ def exec_command(self, *args, **kwargs):
 
     def _exec_command(self, host, *args, **kwargs):
         """Make SSHClient, run command on host"""
-        if not self.host_clients[host]:
+        if not host in self.host_clients or not self.host_clients[host]:
             self.host_clients[host] = SSHClient(host, user=self.user,
                                                 password=self.password,
                                                 port=self.port, pkey=self.pkey,