Support upstreams with showall

Author: BartoszBlizniak

cloudsmith_cli/cli/commands/upstream.py

@@ -31,7 +31,7 @@
 ]
 
 
-def print_upstreams(upstreams, upstream_fmt):
+def print_upstreams(upstreams, upstream_fmt, page_info=None, show_all=False):
     """Print upstreams as a table or output in another format."""
 
     def build_row(u):
@@ -109,9 +109,14 @@ def build_row(u):
     utils.pretty_print_table(headers, rows)
     click.echo()
 
-    num_results = len(rows)
+    num_results = len(upstreams)
     list_suffix = "upstream%s" % ("" if num_results == 1 else "s")
-    utils.pretty_print_list_info(num_results=num_results, suffix=list_suffix)
+    utils.pretty_print_list_info(
+        num_results=num_results,
+        page_info=None if show_all else page_info,
+        suffix=list_suffix,
+        show_all=show_all,
+    )
 
 
 @main.group(cls=command.AliasGroup, name="upstream", aliases=[])
@@ -158,7 +163,7 @@ def build_upstream_list_command(upstream_fmt):
         "owner_repo", metavar="OWNER/REPO", callback=validators.validate_owner_repo
     )
     @click.pass_context
-    def func(ctx, opts, owner_repo, page, page_size):
+    def func(ctx, opts, owner_repo, page, page_size, show_all):
         owner, repo = owner_repo
 
         # Use stderr for messages if the output is something else (e.g.  # JSON)
@@ -169,20 +174,40 @@ def func(ctx, opts, owner_repo, page, page_size):
         context_msg = "Failed to get upstreams!"
         with handle_api_exceptions(ctx, opts=opts, context_msg=context_msg):
             with maybe_spinner(opts):
-                upstreams, page_info = api.list_upstreams(
-                    owner=owner,
-                    repo=repo,
-                    upstream_format=upstream_fmt,
-                    page=page,
-                    page_size=page_size,
-                )
+                if show_all:
+                    upstreams = []
+                    current_page = 1
+                    while True:
+                        page_upstreams, page_info = api.list_upstreams(
+                            owner=owner,
+                            repo=repo,
+                            upstream_format=upstream_fmt,
+                            page=current_page,
+                            page_size=page_size,
+                        )
+                        upstreams.extend(page_upstreams)
+                        if (
+                            len(page_upstreams) < page_size
+                            or current_page >= page_info.page_total
+                        ):
+                            break
+                        current_page += 1
+                    page_info.count = len(upstreams)
+                else:
+                    upstreams, page_info = api.list_upstreams(
+                        owner=owner,
+                        repo=repo,
+                        upstream_format=upstream_fmt,
+                        page=page,
+                        page_size=page_size,
+                    )
 
         click.secho("OK", fg="green", err=use_stderr)
 
         if utils.maybe_print_as_json(opts, upstreams, page_info):
             return
 
-        print_upstreams(upstreams, upstream_fmt)
+        print_upstreams(upstreams, upstream_fmt, page_info, show_all)
 
     func.__doc__ = f"""
         List {upstream_fmt} upstreams for a repository.