Otherwise there will be an error when paramiko tries to write the
known_hosts file back

Signed-off-by: Ben Lipton <[email protected]>
---
 p2v-transfer/p2v_transfer.py |    5 ++++-
 1 files changed, 4 insertions(+), 1 deletions(-)

diff --git a/p2v-transfer/p2v_transfer.py b/p2v-transfer/p2v_transfer.py
index fdbb335..e98dc00 100755
--- a/p2v-transfer/p2v_transfer.py
+++ b/p2v-transfer/p2v_transfer.py
@@ -142,7 +142,10 @@ def EstablishConnection(user, host, key):
     # Load from the known_hosts file. Additional keys will be saved back there.
     client.load_host_keys(known_hosts_filename)
   except IOError:
-    pass
+    # Error is ok, file will be created as long as parent directory exists.
+    # So, make sure it exists:
+    ssh_dir = os.path.dirname(known_hosts_filename)
+    if not os.path.exists(ssh_dir):
+      os.makedirs(ssh_dir)
 
   try:
     client.connect(host, username=user, pkey=key,
-- 
1.7.3.1

Reply via email to