feat(cli): implement mv command

Author: william-silversmith

taskqueue/taskqueue.py

@@ -245,9 +245,10 @@ def delete(self, task_id, total=None, tally=False):
     total = totalfn(task_id, total)
 
     def deltask(tid):
-      self.api.delete(totaskid(tid))
+      num_deleted = self.api.delete(totaskid(tid))
       if tally:
         self.api.tally()
+      return num_deleted
 
     schedule_jobs(
       fns=( partial(deltask, tid) for tid in task_id ),

taskqueue_cli/taskqueue_cli.py

@@ -1,6 +1,7 @@
 import os
 
 import click
+from tqdm import tqdm
 
 from taskqueue import TaskQueue, __version__, QueueEmptyError
 from taskqueue.lib import toabs
@@ -102,17 +103,20 @@ def mv(src, dest):
   src = normalize_path(src)
   dest = normalize_path(dest)
 
-  tqd = TaskQueue(dest)
-  tqs = TaskQueue(src)
+  tqd = TaskQueue(dest, progress=False)
+  tqs = TaskQueue(src, progress=False)
 
-  while True:
-    try:
-      tasks = tqs.lease(num_tasks=10, seconds=10)
-    except QueueEmptyError:
-      break
+  total = tqs.enqueued
+  with tqdm(total=total, desc="Moving") as pbar:
+    while True:
+      try:
+        tasks = tqs.lease(num_tasks=10, seconds=10)
+      except QueueEmptyError:
+        break
 
-    tqd.insert(tasks)
-    tqs.delete(tasks)
+      tqd.insert(tasks)
+      tqs.delete(tasks)
+      pbar.update(len(tasks))