Use ssh_forwardagent as a key in a Host dict

nornir-automation · Jun 27, 2018 · d870431 · d870431
1 parent c0ea53a
commit d870431
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/nornir/plugins/tasks/commands/remote_command.py b/nornir/plugins/tasks/commands/remote_command.py
@@ -24,7 +24,7 @@ def remote_command(task, command):
 
     chan = client.get_transport().open_session()
 
-    if task.host.ssh_forwardagent:
+    if task.host["ssh_forwardagent"]:
         AgentRequestHandler(chan)
 
     chan.exec_command(command)

diff --git a/nornir/plugins/tasks/connections/paramiko_connection.py b/nornir/plugins/tasks/connections/paramiko_connection.py
@@ -35,9 +35,10 @@ def paramiko_connection(task=None):
     if "proxycommand" in user_config:
         parameters["sock"] = paramiko.ProxyCommand(user_config["proxycommand"])
 
+    task.host["ssh_forwardagent"] = False
     if "forwardagent" in user_config:
         if user_config["forwardagent"] == "yes":
-            task.host.ssh_forwardagent = True
+            task.host["ssh_forwardagent"] = True
 
     # TODO configurable
     #  if ssh_key_file: