From aa9f258f0e20edfdad83e6e1b6f9b88d40ff43b1 Mon Sep 17 00:00:00 2001 From: Batuhan Taskaya Date: Tue, 2 Feb 2021 12:27:19 +0300 Subject: [PATCH] ssh: show total-size when uploading/downloading on ssh remote (#5383) (cherry picked from commit 9ad1e0e15925ec3ca0d741d64bdf65d293a83f58) --- dvc/tree/ssh/connection.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/dvc/tree/ssh/connection.py b/dvc/tree/ssh/connection.py index c313eafe02..e3b236f5bc 100644 --- a/dvc/tree/ssh/connection.py +++ b/dvc/tree/ssh/connection.py @@ -189,6 +189,7 @@ def download(self, src, dest, no_progress_bar=False, progress_title=None): desc=progress_title or os.path.basename(src), disable=no_progress_bar, bytes=True, + total=self.getsize(src), ) as pbar: self.sftp.get(src, dest, callback=pbar.update_to) @@ -221,7 +222,10 @@ def upload(self, src, dest, no_progress_bar=False, progress_title=None): progress_title = posixpath.basename(dest) with Tqdm( - desc=progress_title, disable=no_progress_bar, bytes=True + desc=progress_title, + disable=no_progress_bar, + bytes=True, + total=os.path.getsize(src), ) as pbar: self.sftp.put(src, tmp_file, callback=pbar.update_to)