mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
worktree: use progress bars in push/fetch
This commit is contained in:
committed by
Peter Rowlands (변기호)
parent
2748492f6b
commit
af255a46b4
+20
-2
@@ -1,5 +1,7 @@
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
|
||||
from dvc.fs.callbacks import Callback
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from dvc.cloud import Remote
|
||||
from dvc.index import Index, IndexView
|
||||
@@ -57,7 +59,12 @@ def fetch_worktree(repo: "Repo", remote: "Remote") -> int:
|
||||
remote.path,
|
||||
*key,
|
||||
)
|
||||
return save(index)
|
||||
total = len(index)
|
||||
with Callback.as_tqdm_callback(
|
||||
unit="file", desc="Fetch", disable=total == 0
|
||||
) as cb:
|
||||
cb.set_size(total)
|
||||
return save(index, callback=cb)
|
||||
|
||||
|
||||
def push_worktree(repo: "Repo", remote: "Remote") -> int:
|
||||
@@ -66,7 +73,18 @@ def push_worktree(repo: "Repo", remote: "Remote") -> int:
|
||||
|
||||
view = worktree_view(repo.index, push=True)
|
||||
index = view.data["repo"]
|
||||
pushed = checkout(index, remote.path, remote.fs, latest_only=False)
|
||||
total = len(index)
|
||||
with Callback.as_tqdm_callback(
|
||||
unit="file", desc="Push", disable=total == 0
|
||||
) as cb:
|
||||
cb.set_size(total)
|
||||
pushed = checkout(
|
||||
index,
|
||||
remote.path,
|
||||
remote.fs,
|
||||
latest_only=False,
|
||||
callback=cb,
|
||||
)
|
||||
|
||||
for stage in view.stages:
|
||||
for out in stage.outs:
|
||||
|
||||
Reference in New Issue
Block a user