scripts: west_commands: sdk: Display a download progress bar.

Improved the display using tqdm to show the download progress.

Signed-off-by: TOKITA Hiroshi <[email protected]>

Author: soburi

scripts/requirements-base.txt

@@ -23,6 +23,7 @@ pylink-square
 pyserial
 requests
 semver
+tqdm
 
 # for ram/rom reports
 anytree

scripts/west_commands/sdk.py

@@ -4,6 +4,7 @@
 
 import argparse
 import hashlib
+import io
 import os
 import patoolib
 import platform
@@ -14,6 +15,7 @@
 import subprocess
 import tempfile
 import textwrap
+import tqdm
 import zcmake
 from pathlib import Path
 
@@ -313,10 +315,41 @@ def download_and_extract(self, base_dir, dir_name, target_release, req_headers):
                 total_length = int(resp.headers["Content-Length"])
                 count = 0
 
+                class WestLogAdapter(io.StringIO):
+                    buf = []
+                    ctx = None
+
+                    def __init__(self, ctx):
+                        super(__class__, self).__init__()
+                        self.ctx = ctx
+
+                    def write(self, buf):
+                        try:
+                            self.ctx.inf(buf, colorize=False, end='')
+                        except TypeError:
+                            # West v1.1.0 and earlier versions do not support the end parameter.
+                            self.ctx.inf(buf, colorize=False)
+
+                    def flush(self):
+                        pass
+
+                tbar = tqdm.tqdm(
+                    total=total_length,
+                    file=WestLogAdapter(self),
+                    desc=Path(file.name).name,
+                    unit_scale=True,
+                    ncols=shutil.get_terminal_size().columns,
+                    unit="",
+                    bar_format="{desc}: {percentage:3.0f}%|{bar}|   {n_fmt} {rate_fmt}   [{elapsed}]        ",
+                )
+
                 for chunk in resp.iter_content(chunk_size=8192):
+                    tbar.update(len(chunk))
                     file.write(chunk)
                     count = count + len(chunk)
-                    self.inf(f"\r {count}/{total_length}", end="")
+
+                tbar.close()
+
                 self.inf()
                 self.inf(f"Downloaded: {file.name}")
                 file.close()