Fixed AutoAddPolicy

2018-04-23 18:14:42 +08:00 · 2018-04-23 18:14:42 +08:00 · 7b715bea17
parent 3165cf2399
commit 7b715bea17
1 changed files with 2 additions and 1 deletions
--- a/webssh/policy.py
+++ b/webssh/policy.py
@ -56,7 +56,8 @@ class AutoAddPolicy(paramiko.client.MissingHostKeyPolicy):
    lock = threading.Lock()

    def is_missing_host_key(self, client, hostname, key):
-        k = client._host_keys.lookup(hostname)
+        k = client._system_host_keys.lookup(hostname) or \
+                client._host_keys.lookup(hostname)
        if k is None:
            return True
        host_key = k.get(key.get_name(), None)