feat(buckets.py): added limit option to remove command (#121)

workflow/cli/buckets.py

@@ -38,13 +38,17 @@ def buckets():
 )
 @click.option("-p", "--parent", type=str, required=False, help="Filter by parent.")
 @click.option("-f", "--force", is_flag=True, help="Do not prompt for confirmation")
+@click.option(
+    "-l", "--limit", type=int, required=False, help="Limit of Work objects to remove."
+)
 def remove(
     bucket: str,
     status: Optional[str] = None,
     event: Optional[Tuple[int]] = None,
     tag: Optional[Tuple[str]] = None,
     parent: Optional[str] = None,
     force: bool = False,
+    limit: Optional[int] = 100,
 ):
     """Remove work[s] from the buckets.
 
@@ -55,8 +59,9 @@ def remove(
         tag (Optional[Tuple[str]], optional): Filter by tag. Defaults to None.
         parent (Optional[str], optional): Filter by parent. Defaults to None.
         force (bool, optional): Do not prompt for confirmation. Defaults to False.
+        limit (int, optional): Limit of Work objects to remove. Defaults to False.
     """
-    http = HTTPContext()
+    http = HTTPContext(backends=["buckets"])
     events: Optional[List[int]] = None
     tags: Optional[List[str]] = None
     if event:
@@ -70,6 +75,7 @@ def remove(
         tags=tags,
         parent=parent,
         force=force,
+        limit=limit,
     )
 
 

workflow/http/buckets.py

@@ -148,6 +148,7 @@ def delete_many(
         tags: Optional[List[str]] = None,
         parent: Optional[str] = None,
         force: bool = False,
+        limit: Optional[int] = 100
     ) -> bool:
         """Delete works belonging to a pipeline from the buckets backend.
 
@@ -161,6 +162,7 @@ def delete_many(
             event (Optional[List[int]]): The event to delete works with.
             force (bool, optional): Whether to force the deletion without requiring
                 user confirmation. Defaults to False.
+            limit (int, optional): Limit of Work objects to remove. Defaults to False.
 
         Returns:
             bool: Whether any works were deleted.
@@ -171,7 +173,7 @@ def delete_many(
         query.update({"tags": {"$in": tags}} if tags else {})
         query.update({"config.parent": parent} if parent else {})
         projection = {"id": True}
-        result = self.view(query, projection)
+        result = self.view(query, projection, limit=limit)
         ids: List[str] = []
         if result:
             ids = [work["id"] for work in result]