From 92223b9b888951e1fd53a48cbc76ee4b8ceb74c0 Mon Sep 17 00:00:00 2001 From: Batuhan Taskaya Date: Tue, 2 Feb 2021 12:05:19 +0300 Subject: [PATCH] ssh: show total-size when uploading/downloading on ssh remote --- 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)