From 59eb9295169f2c7500f6aeb45b45184a95e4fc6a Mon Sep 17 00:00:00 2001
From: Alexander Schepanovski <suor.web@gmail.com>
Date: Tue, 23 Jul 2019 16:20:58 +0300
Subject: [PATCH] ssh: do not ask for same password repeatedly

---
 dvc/remote/ssh/__init__.py | 25 ++++++++++++++++++-------
 1 file changed, 18 insertions(+), 7 deletions(-)

diff --git a/dvc/remote/ssh/__init__.py b/dvc/remote/ssh/__init__.py
index 2ed8b06e8..22a7da5aa 100644
--- a/dvc/remote/ssh/__init__.py
+++ b/dvc/remote/ssh/__init__.py
@@ -6,6 +6,7 @@ import logging
 import itertools
 import errno
 from concurrent.futures import ThreadPoolExecutor
+import threading
 
 try:
     import paramiko
@@ -26,6 +27,10 @@ from .connection import SSHConnection
 logger = logging.getLogger(__name__)
 
 
+saved_passwords = {}
+saved_passwords_lock = threading.Lock()
+
+
 class RemoteSSH(RemoteBASE):
     scheme = Schemes.SSH
     REQUIRES = {"paramiko": paramiko}
@@ -115,13 +120,19 @@ class RemoteSSH(RemoteBASE):
         host, user, port = path_info.host, path_info.user, path_info.port
 
         # NOTE: we use the same password regardless of the server :(
-        if self.ask_password and not self.password:
-            self.password = prompt.password(
-                "Enter a private key passphrase or a password for "
-                "host '{host}' port '{port}' user '{user}'".format(
-                    host=host, port=port, user=user
-                )
-            )
+        if self.ask_password and self.password is None:
+            with saved_passwords_lock:
+                server_key = (host, user, port)
+                password = saved_passwords.get(server_key)
+
+                if password is None:
+                    saved_passwords[server_key] = password = prompt.password(
+                        "Enter a private key passphrase or a password for "
+                        "host '{host}' port '{port}' user '{user}'".format(
+                            host=host, port=port, user=user
+                        )
+                    )
+                self.password = password
 
         return get_connection(
             SSHConnection,