sima-cli 0.0.29__py3-none-any.whl → 0.0.31__py3-none-any.whl

sima_cli/__version__.py

@@ -1,2 +1,2 @@
 # sima_cli/__version__.py
-__version__ = "0.0.29"
+__version__ = "0.0.31"

sima_cli/auth/basic_auth.py

@@ -78,55 +78,116 @@ def _fetch_and_store_csrf_token(session: requests.Session) -> str:
 
 
 def login_external():
-    """Interactive login workflow with CSRF token, cookie caching, and dummy session validation."""
+    """Interactive login workflow with CSRF token, cookie caching, and TOTP handling."""
     for attempt in range(1, 4):
         session = requests.Session()
         session.headers.update(HEADERS)
 
         _load_cookie_jar(session)
-        csrf_token = _load_csrf_token()
-
-        if not csrf_token:
-            csrf_token = _fetch_and_store_csrf_token(session)
-
+        csrf_token = _load_csrf_token() or _fetch_and_store_csrf_token(session)
         if not csrf_token:
             click.echo("❌ CSRF token is missing or invalid.")
             continue
-
         session.headers["X-CSRF-Token"] = csrf_token
 
         if _is_session_valid(session):
             click.echo("🚀 You are already logged in.")
             return session
 
-        # Prompt user login
+        # Fresh login prompt
         _delete_auth_files()
         click.echo(f"🔐 Sima.ai Developer Portal Login Attempt {attempt}/3")
         username = click.prompt("Email or Username")
         password = getpass.getpass("Password: ")
 
-        login_data = {
+        # Base payload (no TOTP yet)
+        base_data = {
             "login": username,
             "password": password,
-            "second_factor_method": "1"
+            "second_factor_method": "1",  # TOTP
         }
 
-        try:
-            resp = session.post(LOGIN_URL, data=login_data)
-            name = resp.json().get('users')[0]['name'] if 'users' in resp.json() else ''
-            if resp.status_code != 200:
-                click.echo(f"⚠️ Login request returned status {resp.status_code}")
-                continue
-        except Exception as e:
-            click.echo(f"❌ Login request failed: {e}")
-            continue
-
-        if _is_session_valid(session):
+        def _post_login(payload: dict):
+            """POST and return (status_code, json or None, text) with robust error handling."""
+            try:
+                resp = session.post(LOGIN_URL, data=payload, timeout=30)
+            except Exception as e:
+                return None, None, f"request failed: {e}"
+            j = None
+            try:
+                j = resp.json()
+            except Exception:
+                pass
+            return resp.status_code, j, (j or resp.text)
+
+        # First try without TOTP (server may ask for it)
+        status, j, raw = _post_login(base_data)
+
+        # Helper: decide if success
+        def _success():
+            # Prefer server 'ok': True, but also double-check the session cookie validity
+            if j and j.get("ok") is True:
+                return True
+            return _is_session_valid(session)
+
+        # If immediate success
+        if status == 200 and _success():
             _save_cookie_jar(session)
-            click.echo(f"✅ Login successful. Welcome to Sima Developer Portal, {name}!")
+            welcome = (j.get("users", [{}])[0].get("name") if isinstance(j, dict) else "") or ""
+            click.echo(f"✅ Login successful. Welcome to Sima Developer Portal{', ' + welcome if welcome else ''}!")
             return session
+
+        # See if TOTP is required/invalid; then prompt and retry up to 3 times
+        def _needs_totp(payload_json):
+            if not isinstance(payload_json, dict):
+                return False
+            if payload_json.get("totp_enabled") is True:
+                return True
+            reason = payload_json.get("reason") or payload_json.get("error")
+            return str(reason) in {"invalid_second_factor", "second_factor_required"}
+
+        if _needs_totp(j):
+            # Try up to 3 TOTP attempts within this login attempt
+            for totp_try in range(1, 4):
+                totp = click.prompt(f"🔢 Enter TOTP code (attempt {totp_try}/3)", hide_input=True)
+                data = dict(base_data)
+                data["second_factor_token"] = totp
+
+                status, j2, raw2 = _post_login(data)
+                if status == 200 and (j2 and j2.get("ok") is True or _is_session_valid(session)):
+                    _save_cookie_jar(session)
+                    welcome = (j2.get("users", [{}])[0].get("name") if isinstance(j2, dict) else "") or ""
+                    click.echo(f"✅ Login successful. Welcome to Sima Developer Portal{', ' + welcome if welcome else ''}!")
+                    return session
+
+                # If still invalid 2FA, let user try again; otherwise break to outer loop
+                msg = ""
+                if isinstance(j2, dict):
+                    reason = j2.get("reason") or j2.get("error") or ""
+                    msg = f" ({reason})" if reason else ""
+                if isinstance(j2, dict) and str(j2.get("reason")) in {"invalid_second_factor"}:
+                    click.echo(f"❌ Invalid authentication code. Please try again.{msg}")
+                    continue
+                else:
+                    click.echo(f"❌ Login failed with TOTP{msg}.")
+                    break  # go to next overall attempt
+
+            # exhausted TOTP tries
+            click.echo("❌ TOTP verification failed after 3 attempts.")
+            continue  # next overall attempt
+
+        # Not a TOTP case; report error and continue
+        err_detail = ""
+        if isinstance(j, dict):
+            err_detail = j.get("error") or j.get("message") or ""
+            reason = j.get("reason")
+            if reason and reason != err_detail:
+                err_detail = f"{err_detail} ({reason})" if err_detail else reason
         else:
-            click.echo("❌ Login failed.")
+            err_detail = str(raw)[:200]
+
+        click.echo(f"❌ Login failed. {err_detail or 'Please check your credentials and try again.'}")
 
     click.echo("❌ Login failed after 3 attempts.")
     raise SystemExit(1)
+

sima_cli/cli.py

@@ -14,6 +14,7 @@ from sima_cli.serial.serial import connect_serial
 from sima_cli.storage.nvme import nvme_format, nvme_remount
 from sima_cli.storage.sdcard import sdcard_format
 from sima_cli.network.network import network_menu
+from sima_cli.utils.pkg_update_check import check_for_update
 
 # Entry point for the CLI tool using Click's command group decorator
 @click.group()
@@ -26,6 +27,7 @@ def main(ctx, internal):
     Global Options:
       --internal  Use internal Artifactory resources (can also be set via env variable SIMA_CLI_INTERNAL=1)
     """
+    check_for_update('sima-cli')
     ctx.ensure_object(dict)
 
     os.makedirs(os.path.dirname(CONFIG_PATH), exist_ok=True)

sima_cli/install/optiview.py

@@ -6,19 +6,65 @@ import tarfile
 import os
 import subprocess
 from pathlib import Path
+import tempfile
+import stat
 from sima_cli.download import download_file_from_url
 from sima_cli.utils.env import is_sima_board
 from sima_cli.utils.config_loader import load_resource_config
 
+
+def install_optiview_devkit():
+    """
+    Install optiview under /data/optiview with system site packages visible,
+    and add an alias to call it via sudo from .bashrc.
+    """
+    optiview_dir = "/data/optiview"
+    venv_dir = f"{optiview_dir}/.venv"
+
+    if is_sima_board():
+        click.echo("🛠  Detected SiMa DevKit. Cleaning up existing installation...")
+
+        # Ensure base folder exists
+        subprocess.run(["sudo", "mkdir", "-p", optiview_dir], check=True)
+
+        # Remove any old installation inside venv
+        subprocess.run(["sudo", "pip3", "uninstall", "-y", "optiview"], check=False)
+
+        click.echo("📦 Creating virtual environment with system site packages...")
+        subprocess.run([
+            "sudo", "python3", "-m", "venv", "--system-site-packages", venv_dir
+        ], check=True)
+
+        click.echo("📦 Installing Optiview via pip inside venv...")
+        subprocess.run([
+            "sudo", f"{venv_dir}/bin/pip", "install", "optiview"
+        ], check=True)
+
+        # Add alias to ~/.bashrc for sudo launch
+        alias_cmd = f"alias optiview='sudo {venv_dir}/bin/optiview'"
+        bashrc_path = os.path.expanduser("~/.bashrc")
+
+        # Ensure idempotent append
+        with open(bashrc_path, "a+") as f:
+            f.seek(0)
+            if alias_cmd not in f.read():
+                f.write(f"\n# Optiview alias\n{alias_cmd}\n")
+                click.echo(f"🔗 Added alias to {bashrc_path}")
+            else:
+                click.echo(f"ℹ️ Alias already exists in {bashrc_path}")
+
+        click.echo("✅ Optiview installed successfully on DevKit.")
+        click.echo("ℹ️  Restart your shell or run 'source ~/.bashrc' to use the alias.")
+
+        return True
+
+    return False
+
 def install_optiview():
     try:
         # Special path for SiMa DevKit
         if is_sima_board():
-            click.echo("🛠 Detected SiMa DevKit. Cleaning up existing installation...")
-            subprocess.run(["sudo", "pip3", "uninstall", "-y", "optiview"], check=True)
-            click.echo("📦 Installing Optiview via pip...")
-            subprocess.run(["sudo", "pip3", "install", "optiview"], check=True)
-            click.echo("✅ Optiview installed successfully on DevKit.")
+            install_optiview_devkit()
             return
 
         cfg = load_resource_config()

sima_cli/update/cleanlog.py

@@ -0,0 +1,26 @@
+import re
+from typing import Iterable, Pattern, List
+
+# Common noisy bits you mentioned; add your own here.
+DEFAULT_NOISE_PATTERNS = [
+    r"Information:\s+You may need to update /etc/fstab\.",   # fdisk/parted hint
+    r"resize2fs\s+\d+\.\d+\.\d+\s*\([^)]+\)",                # e2fsprogs banner
+    # add more as needed...
+]
+
+_ANSI_RE = re.compile(r"\x1b\[[0-9;]*[A-Za-z]")
+
+def strip_ansi(s: str) -> str:
+    return _ANSI_RE.sub("", s)
+
+class LineSquelcher:
+    """Return False for lines we want to hide."""
+    def __init__(self,
+                 patterns: Iterable[str] = DEFAULT_NOISE_PATTERNS,
+                 ignore_case: bool = True):
+        flags = re.IGNORECASE if ignore_case else 0
+        self._rx: List[Pattern[str]] = [re.compile(p, flags) for p in patterns]
+
+    def allow(self, line: str) -> bool:
+        clean = strip_ansi(line)
+        return not any(rx.search(clean) for rx in self._rx)

sima_cli/update/local.py

@@ -2,13 +2,22 @@ import os
 from typing import Tuple
 import pty
 import click
+import re
 
+from typing import Optional
 from sima_cli.utils.env import is_board_running_full_image, get_exact_devkit_type
+from sima_cli.update.cleanlog import LineSquelcher
 
-def _run_local_cmd(command: str, passwd: str) -> bool:
+
+def _run_local_cmd(
+    command: str,
+    passwd: str,
+    squelcher: Optional[LineSquelcher] = None,
+    show_summary: bool = True,
+) -> bool:
     """
-    Run a local command using a pseudo-terminal (pty) to force live output flushing,
-    and optionally pass a sudo password.
+    Run a local command using a PTY for live output.
+    Filters out noisy lines using LineSquelcher (if provided or default).
     """
     click.echo(f"🖥️  Running: {command}")
 
@@ -16,6 +25,10 @@ def _run_local_cmd(command: str, passwd: str) -> bool:
     if needs_sudo:
         command = f"sudo -S {command[len('sudo '):]}"
 
+    squelcher = squelcher or LineSquelcher()
+    suppressed = 0
+    buf = ""  # carry partial lines between reads
+
     try:
         pid, fd = pty.fork()
 
@@ -24,19 +37,43 @@ def _run_local_cmd(command: str, passwd: str) -> bool:
             os.execvp("sh", ["sh", "-c", command])
         else:
             if needs_sudo:
+                # Send the sudo password immediately (stdin is the PTY)
                 os.write(fd, (passwd + "\n").encode())
 
             while True:
                 try:
-                    output = os.read(fd, 1024).decode()
-                    if not output:
+                    chunk = os.read(fd, 4096)
+                    if not chunk:
                         break
-                    for line in output.splitlines():
-                        click.echo(line)
+
+                    # Decode & normalize progress lines: turn lone \r into \n
+                    text = chunk.decode("utf-8", errors="replace")
+                    text = text.replace("\r\n", "\n").replace("\r", "\n")
+
+                    buf += text
+                    *lines, buf = buf.split("\n") 
+
+                    for line in lines:
+                        if squelcher.allow(line):
+                            click.echo(line)
+                        else:
+                            suppressed += 1
+
                 except OSError:
                     break
 
+            # Flush any remaining partial line
+            if buf:
+                if squelcher.allow(buf):
+                    click.echo(buf)
+                else:
+                    suppressed += 1
+
             _, status = os.waitpid(pid, 0)
+
+            if show_summary and suppressed:
+                click.echo(f"🔇 suppressed {suppressed} noisy line(s)")
+
             return os.WIFEXITED(status) and os.WEXITSTATUS(status) == 0
 
     except Exception as e:
@@ -75,6 +112,31 @@ def get_local_board_info() -> Tuple[str, str, bool]:
     return board_type, build_version, fdt_name, is_board_running_full_image()
 
 
+def get_boot_mmc(mounts_path="/proc/mounts", cmdline_path="/proc/cmdline"):
+    """
+    Figure out which eMMC the device was booted from - local version
+    """    
+    try:
+        with open(mounts_path) as f:
+            for line in f:
+                dev, mnt = line.split()[:2]
+                if mnt == "/":
+                    m = re.search(r'(mmcblk\d+)', dev)
+                    if m:
+                        return m.group(1)
+    except OSError:
+        pass
+
+    try:
+        with open(cmdline_path) as f:
+            m = re.search(r'\broot=(?:/dev/)?(mmcblk\d+)', f.read())
+            if m:
+                return m.group(1)
+    except OSError:
+        pass
+
+    return None
+
 def push_and_update_local_board(troot_path: str, palette_path: str, passwd: str, flavor: str):
     """
     Perform local firmware update using swupdate commands.
@@ -83,6 +145,11 @@ def push_and_update_local_board(troot_path: str, palette_path: str, passwd: str,
     click.echo("📦 Starting local firmware update...")
 
     try:
+        blk = get_boot_mmc()
+        if blk != None:
+            fix_gpt_cmd = f'sudo printf "fix\n" | sudo parted ---pretend-input-tty /dev/{blk} print'
+            _run_local_cmd(fix_gpt_cmd, passwd)
+
         # Run tRoot update
         if troot_path != None:
             click.echo("⚙️  Flashing tRoot image...")

sima_cli/update/query.py

@@ -39,6 +39,10 @@ def _list_available_firmware_versions_internal(board: str, match_keyword: str =
     }
 
     response = requests.post(aql_url, data=aql_query, headers=headers)
+
+    if response.status_code == 401:
+        print('❌ You are not authorized to access Artifactory, use `sima-cli -i login` with your Artifactory identity token to authenticate, then try the command again.')
+
     if response.status_code != 200:
         return None
 

sima_cli/update/remote.py

@@ -7,7 +7,9 @@ import itertools
 import threading
 import sys
 import select
-from typing import Tuple
+from typing import Tuple, Optional
+
+from sima_cli.update.cleanlog import LineSquelcher
 
 DEFAULT_USER = "sima"
 DEFAULT_PASSWORD = "edgeai"
@@ -97,7 +99,7 @@ def get_remote_board_info(ip: str, passwd: str = DEFAULT_PASSWORD) -> Tuple[str,
 
         for line in fdt_output.splitlines():
             if line.startswith("fdt_name"):
-                fdt_name = line.split("=", 1)[-1].strip()
+                fdt_name = line.split("=", 1)[-1].strip().replace('.dtb', '')
 
         return board_type, build_version, fdt_name, full_image
 
@@ -113,7 +115,9 @@ def _scp_file(sftp, local_path: str, remote_path: str):
     sftp.put(local_path, remote_path)
     click.echo("✅ Upload complete")
 
-def run_remote_command(ssh, command: str, password: str = DEFAULT_PASSWORD):
+
+def run_remote_command(ssh, command: str, password: str = DEFAULT_PASSWORD,
+                       squelcher: Optional[LineSquelcher] = None):
     """
     Run a remote command over SSH and stream its output live to the console.
     If the command starts with 'sudo', pipe in the password.
@@ -123,40 +127,56 @@ def run_remote_command(ssh, command: str, password: str = DEFAULT_PASSWORD):
         command (str): The command to run on the remote host.
         password (str): Password to use if the command requires sudo.
     """
-    click.echo(f"🚀 Running on remote: {command}")
+    squelcher = squelcher or LineSquelcher()  # use defaults unless you pass a custom one
 
+    click.echo(f"🚀 Running on remote: {command}")
     needs_sudo = command.strip().startswith("sudo")
     if needs_sudo:
-        # Use -S to allow password from stdin
         command = f"sudo -S {command[len('sudo '):]}"
 
     stdin, stdout, stderr = ssh.exec_command(command, get_pty=True)
-
     if needs_sudo:
-        # Send password immediately, followed by newline
         stdin.write(password + "\n")
         stdin.flush()
 
+    suppressed = 0
     while not stdout.channel.exit_status_ready():
         rl, _, _ = select.select([stdout.channel], [], [], 0.5)
         if rl:
             if stdout.channel.recv_ready():
                 output = stdout.channel.recv(4096).decode("utf-8", errors="replace")
                 for line in output.splitlines():
-                    click.echo(f"↦ {line}")
+                    if squelcher.allow(line):
+                        click.echo(f"↦ {line}")
+                    else:
+                        suppressed += 1
             if stdout.channel.recv_stderr_ready():
                 err_output = stdout.channel.recv_stderr(4096).decode("utf-8", errors="replace")
                 for line in err_output.splitlines():
-                    click.echo(f"⚠️ {line}")
+                    if squelcher.allow(line):
+                        click.echo(f"⚠️ {line}")
+                    else:
+                        suppressed += 1
 
     # Final remaining output
     remaining = stdout.read().decode("utf-8", errors="replace")
     for line in remaining.splitlines():
-        click.echo(f"↦ {line}")
+        if squelcher.allow(line):
+            click.echo(f"↦ {line}")
+        else:
+            suppressed += 1
 
     remaining_err = stderr.read().decode("utf-8", errors="replace")
     for line in remaining_err.splitlines():
-        click.echo(f"⚠️ {line}")
+        if squelcher.allow(line):
+            click.echo(f"⚠️ {line}")
+        else:
+            suppressed += 1
+
+    # Optional: surface how much noise we hid (comment out if you want it totally silent)
+    if suppressed:
+        click.echo(f"🔇 suppressed {suppressed} noisy line(s)")
+
 
 def init_ssh_session(ip: str, password: str = DEFAULT_PASSWORD):
     ssh = paramiko.SSHClient()
@@ -182,6 +202,54 @@ def reboot_remote_board(ip: str, passwd: str):
         click.echo(f"⚠️  Unable to connect to the remote board")
 
 
+def run_remote_command_capture(ssh, command: str, password: str = DEFAULT_PASSWORD):
+    """
+    Run a remote command over SSH and return (exit_status, stdout_str, stderr_str).
+    Does not stream output to the console.
+    If the command starts with 'sudo', it will send the password via stdin.
+    """
+    needs_sudo = command.strip().startswith("sudo")
+    if needs_sudo:
+        command = f"sudo -S {command[len('sudo '):]}"
+
+    stdin, stdout, stderr = ssh.exec_command(command, get_pty=needs_sudo)
+    if needs_sudo:
+        stdin.write(password + "\n")
+        stdin.flush()
+
+    out = stdout.read().decode("utf-8", errors="replace")
+    err = stderr.read().decode("utf-8", errors="replace")
+    code = stdout.channel.recv_exit_status()
+    return code, out, err
+
+
+def get_remote_boot_mmc(ssh, password: str = DEFAULT_PASSWORD) -> str | None:
+    """
+    Determine the remote boot device: 'mmcblk0', 'mmcblk1', or None.
+
+    Strategy (remote):
+      1) Look at the actual device mounted as '/' in /proc/mounts.
+      2) Fallback to parsing /proc/cmdline (root=...).
+    """
+    # Minimal, BusyBox-friendly shell: grep/awk only, no fancy sed EREs.
+    remote_script = r"""
+mmc="$(awk '$2=="/"{print $1}' /proc/mounts | grep -oE 'mmcblk[0-9]+' | head -n1)"
+if [ -z "$mmc" ]; then
+  root_tok="$(sed -n 's/.*root=\([^ ]*\).*/\1/p' /proc/cmdline)"
+  mmc="$(printf '%s\n' "$root_tok" | grep -oE 'mmcblk[0-9]+' | head -n1)"
+fi
+[ -n "$mmc" ] && printf '%s\n' "$mmc"
+"""
+
+    code, out, err = run_remote_command_capture(ssh, f"sh -c {quote_shell(remote_script)}", password=password)
+    mmc = out.strip()
+    return mmc if mmc else None
+
+def quote_shell(s: str) -> str:
+    """Safely single-quote a string for sh -c."""
+    # Turn: abc'def -> 'abc'"'"'def'
+    return "'" + s.replace("'", "'\"'\"'") + "'"
+
 def copy_file_to_remote_board(ip: str, file_path: str, remote_dir: str, passwd: str):
     """
     Copy a file to the remote board over SSH.
@@ -220,6 +288,12 @@ def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, pa
         remote_dir = "/tmp"
         palette_name = os.path.basename(palette_path)
 
+        boot_mmc = get_remote_boot_mmc(ssh, passwd)
+        if boot_mmc != None:
+            click.echo(f'✅ Checking partition table GPT record for {boot_mmc}...')
+            fix_gpt_cmd = f'sudo printf "fix\n" | sudo parted ---pretend-input-tty /dev/{boot_mmc} print'
+            run_remote_command(ssh, fix_gpt_cmd)
+
         # Upload tRoot image
         if troot_path is not None:
             troot_name = os.path.basename(troot_path)
@@ -231,9 +305,10 @@ def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, pa
                 ssh,
                 f"sudo swupdate -H simaai-image-troot:1.0 -i /tmp/{troot_name}", password=passwd
             )
-            click.echo("✅ tRoot update complete, the board needs to be rebooted to proceed to the next phase of update.")
-            click.confirm("⚠️  Have you rebooted the board?", default=True, abort=True)
-            _wait_for_ssh(ip, timeout=120)
+            # Disabled the following code per agreement with QA, this reboot step is not required for tRoot update
+            # click.echo("✅ tRoot update complete, the board needs to be rebooted to proceed to the next phase of update.")
+            # click.confirm("⚠️  Have you rebooted the board?", default=True, abort=True)
+            # _wait_for_ssh(ip, timeout=120)
         else:
             click.echo("⚠️  tRoot update skipped because the requested image doesn't contain troot image.")
 

sima_cli/update/updater.py

@@ -137,7 +137,7 @@ def _pick_from_available_versions(board: str, version_or_url: str, internal: boo
             )
             raise SystemExit(1)
     except:
-        click.echo("Unable to determine available versions")
+        click.echo("❌ Unable to determine available versions")
         exit(0)
 
 def _sanitize_url_to_filename(url: str) -> str:
@@ -511,9 +511,9 @@ def perform_update(version_or_url: str, ip: str = None, internal: bool = False,
         flavor = _confirm_flavor_switching(full_image=full_image, flavor=flavor)
 
         if board in ['davinci', 'modalix']:
-            click.echo(f"🔧 Target board: {board} {fdt_name}, board currently running: {version}, full_image: {full_image}")
+            click.echo(f"🔧 Target board: {board} : {fdt_name}, board currently running: {version}, full_image: {full_image}")
             
-            if flavor == 'full' and fdt_name != 'modalix-som.dtb':
+            if flavor == 'full' and fdt_name != 'modalix-som':
                 click.echo(f"❌ You've requested updating {fdt_name} to full image, this is only supported for the Modalix DevKit")
                 return
 
@@ -539,9 +539,13 @@ def perform_update(version_or_url: str, ip: str = None, internal: bool = False,
                 if env_type == "host" and env_subtype == 'linux':
                     # Always update the remote device first then update the host driver, otherwise the host would 
                     # not be able to connect to the board
-                    click.echo("👉 Updating PCIe host driver and downloading firmware...")        
                     script_path = _update_remote(extracted_paths, ip, board, passwd, reboot_and_wait=False, flavor=flavor)
-                    _update_host(script_path, board, ip, passwd)
+                    click.echo("👉 sima-cli detected you are updating the board on a Linux host...")        
+                    if click.confirm("👉  Do you want to update the host PCIe driver now? If you do not intend to use a Sima PCIe card ever on this machine, enter N", default=False):
+                        click.echo("👉  Updating PCIe host driver and downloading firmware...")
+                        _update_host(script_path, board, ip, passwd)
+                    else:
+                        click.echo("⚠️  Skipping Linux host driver update.")
                 elif env_type == "board":
                     _update_board(extracted_paths, board, passwd, flavor=flavor)
                 elif env_type == "sdk":

sima_cli/utils/pkg_update_check.py

@@ -0,0 +1,84 @@
+import importlib.metadata
+import urllib.request
+import subprocess
+import json
+import socket
+import click
+import sys
+import shlex
+import shutil
+import glob
+import os
+
+def cleanup_pip_leftovers():
+    """Remove ~-prefixed leftover dirs in site-packages."""
+    for path in sys.path:
+        if path.endswith("site-packages") and os.path.isdir(path):
+            junk_dirs = glob.glob(os.path.join(path, "~*"))
+            for d in junk_dirs:
+                try:
+                    shutil.rmtree(d, ignore_errors=True)
+                except Exception as e:
+                    click.secho(f"⚠️ Failed to remove {d}: {e}", fg="yellow")
+
+def update_package(package_name: str):
+    """Suggest manual update on Windows; auto-update elsewhere."""
+    pip_cmd = f"{shlex.quote(sys.executable)} -m pip install --upgrade {package_name}"
+
+    if sys.platform.startswith("win"):
+        click.secho("⚠️  Automatic self-update is not supported on Windows while the CLI is running.", fg="yellow", bold=True)
+        safe_cmd = pip_cmd.replace("'", "")
+        click.echo(f"Please run the following command in a new terminal:\n\n    {safe_cmd}\n")
+        return
+
+    try:
+        subprocess.run(shlex.split(pip_cmd), check=True)
+        cleanup_pip_leftovers()
+        click.secho(f"✅ {package_name} updated successfully.", fg="green", bold=True)
+    except subprocess.CalledProcessError as e:
+        click.secho(f"❌ Failed to update {package_name}: {e}", fg="red", bold=True)
+
+def has_internet(timeout: float = 1.0) -> bool:
+    """
+    Quick check for internet connectivity by connecting to a known DNS server.
+    Uses IP to avoid DNS lookup delays.
+    """
+    try:
+        socket.setdefaulttimeout(timeout)
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.connect(("1.1.1.1", 53))
+        sock.close()
+        return True
+    except OSError:
+        return False
+
+def check_for_update(package_name: str, timeout: float = 2.0):
+
+    if os.environ.get("SIMA_CLI_CHECK_FOR_UPDATE", "1") != "1":
+        print(f'⚠️  You have disabled update check with SIMA_CLI_CHECK_FOR_UPDATE environment variable, skipping sima-cli update check..')
+        return
+    
+    try:
+        current_version = importlib.metadata.version(package_name)
+    except importlib.metadata.PackageNotFoundError:
+        print(f'❌ package not found {package_name}')
+        return
+
+    if not has_internet(timeout=0.2):
+        print(f'⚠️  Offline mode, skipping sima-cli update check..')
+        return
+
+    try:
+        with urllib.request.urlopen(f"https://pypi.org/pypi/{package_name}/json", timeout=timeout) as resp:
+            latest_version = json.load(resp)["info"]["version"]
+    except Exception:
+        return  # PyPI unreachable or network error; skip
+
+    if current_version != latest_version:
+        click.secho(f"🔔 Update available: {current_version} → {latest_version}", fg="green", bold=True)
+        click.secho(f"🔔 If you don't want to automatically check for updates, set SIMA_CLI_CHECK_FOR_UPDATE environment variable to 0")
+        if click.confirm(f"🔔 Do you want to update {package_name} now?", default=True):
+            update_package(package_name)
+            exit(0)
+    else:
+        print('✅ sima-cli is up-to-date')

{sima_cli-0.0.29.dist-info → sima_cli-0.0.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sima-cli
-Version: 0.0.29
+Version: 0.0.31
 Summary: CLI tool for SiMa Developer Portal to download models, firmware, and apps.
 Home-page: https://developer.sima.ai/
 Author: SiMa.ai

{sima_cli-0.0.29.dist-info → sima_cli-0.0.31.dist-info}/RECORD

@@ -1,11 +1,11 @@
 sima_cli/__init__.py,sha256=Nb2jSg9-CX1XvSc1c21U9qQ3atINxphuNkNfmR-9P3o,332
 sima_cli/__main__.py,sha256=ehzD6AZ7zGytC2gLSvaJatxeD0jJdaEvNJvwYeGsWOg,69
-sima_cli/__version__.py,sha256=2CZTSSWitwWN8SjkJ0aMqI9jsY-Mo2BPRi6u1w3Ympk,49
-sima_cli/cli.py,sha256=GYmQ7_XObl9VgFwuWWkWDo-_Y_Vn6lM53F7mKiYGubI,17126
+sima_cli/__version__.py,sha256=5Ig2ah-U3Zaaf5Y1Kp0steH9WKPAxGvzGGgVTVSg_tY,49
+sima_cli/cli.py,sha256=uCZ-9mKBGyGtfx3p97WZkdMjXL2KJlqihfZm8VsWiqw,17220
 sima_cli/app_zoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/app_zoo/app.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sima_cli/auth/basic_auth.py,sha256=pcocI6v496vC7v5MLYPZq3AEgD-2DdkzNFZiKsbx3eQ,4290
+sima_cli/auth/basic_auth.py,sha256=mEmPrj32TVu1s34xR_UrJlIKHA3xBh98i_FzIZvAWag,7364
 sima_cli/auth/login.py,sha256=yCYXWgrfbP4jSTZ3hITfxlgHkdVQVzsd8hQKpqaqCKs,3780
 sima_cli/data/resources_internal.yaml,sha256=zlQD4cSnZK86bLtTWuvEudZTARKiuIKmB--Jv4ajL8o,200
 sima_cli/data/resources_public.yaml,sha256=U7hmUomGeQ2ULdo1BU2OQHr0PyKBamIdK9qrutDlX8o,201
@@ -16,7 +16,7 @@ sima_cli/install/hostdriver.py,sha256=kAWDLebs60mbWIyTbUxmNrChcKW1uD5r7FtWNSUVUE
 sima_cli/install/metadata_info.py,sha256=wmMqwzGfXbuilkqaxRVrFOzOtTOiONkmPCyA2oDAQpA,2168
 sima_cli/install/metadata_installer.py,sha256=UPXxXL5gH0iotX8WCUgEbySbYeIHE1UwsfpZvACjZQs,18928
 sima_cli/install/metadata_validator.py,sha256=7954rp9vFRNnqmIMvCVTjq40kUIEbGXzfc8HmQmChe0,5221
-sima_cli/install/optiview.py,sha256=i5eWVor-9MScEfrQm3Ty9OP4VpSsCgWvNh7AvYdZu7s,3365
+sima_cli/install/optiview.py,sha256=r4DYdQDTUbZVCR87hl5T21gsjZrhqpU8hWnYxKmUJ_k,4790
 sima_cli/install/palette.py,sha256=uRznoHa4Mv9ZXHp6AoqknfC3RxpYNKi9Ins756Cyifk,3930
 sima_cli/mla/meminfo.py,sha256=ndc8kQJmWGEIdvNh6iIhATGdrkqM2pbddr_eHxaPNfg,1466
 sima_cli/model_zoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -31,11 +31,12 @@ sima_cli/storage/sdcard.py,sha256=-WULjdV31-n8v5OOqfxR77qBbIK4hJnrD3xWxUVMoGI,63
 sima_cli/update/__init__.py,sha256=0P-z-rSaev40IhfJXytK3AFWv2_sdQU4Ry6ei2sEus0,66
 sima_cli/update/bmaptool.py,sha256=KrhUGShBwY4Wzz50QiuMYAxxPgEy1nz5C68G-0a4qF4,4988
 sima_cli/update/bootimg.py,sha256=Eg8ZSp8LMZXbOMxX4ZPCjFOg3YEufmsVfojKrRc3fug,13631
-sima_cli/update/local.py,sha256=no3PDChERbBcyjeNVAMR4dH4OaMoRUv8hpym-aoFdhQ,3597
+sima_cli/update/cleanlog.py,sha256=-V6eDl3MdsvDmCfkKUJTqkXJ_WnLJE01uxS7z96b15g,909
+sima_cli/update/local.py,sha256=yOMvOu9nrODEzYZBrxUpdmlfqmkahkDk9nAEuG4RyAg,5588
 sima_cli/update/netboot.py,sha256=hsJQLq4HVwFFkaWjA54VZdkMGDhO0RmylciS78qAfrM,19663
-sima_cli/update/query.py,sha256=b6Su7OlBGooIDcpb3_xH55tqkzznWcd_Fg1MkaNR874,4680
-sima_cli/update/remote.py,sha256=ieZpKN2PCMu90Q72PnN66DZYyqpvBjQYHiDIjvdS5BY,11475
-sima_cli/update/updater.py,sha256=-NXrGbPaxsj69NjdG_1GTB6g2bZMzk8sHJEg4NaKSgQ,23807
+sima_cli/update/query.py,sha256=6RgvQfQT1_EtBGcibvVcz003dRKOq17NaGgL2mhaBbY,4891
+sima_cli/update/remote.py,sha256=dAMIGpHqpf7VBps9JPe4hfHD_qyi1tG6ZW8E_qXQQiQ,14446
+sima_cli/update/updater.py,sha256=Bi-9apDXJlnQg_sUQ46rQPXD6zbx4uE_JhXbfJIGBxE,24213
 sima_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/utils/artifactory.py,sha256=6YyVpzVm8ATy7NEwT9nkWx-wptkXrvG7Wl_zDT6jmLs,2390
 sima_cli/utils/config.py,sha256=wE-cPQqY_gOqaP8t01xsRHD9tBUGk9MgBUm2GYYxI3E,1616
@@ -44,7 +45,8 @@ sima_cli/utils/disk.py,sha256=66Kr631yhc_ny19up2aijfycWfD35AeLQOJgUsuH2hY,446
 sima_cli/utils/env.py,sha256=IP5HrH0lE7RMSiBeXcEt5GCLMT5p-QQroG-uGzl5XFU,8181
 sima_cli/utils/net.py,sha256=WVntA4CqipkNrrkA4tBVRadJft_pMcGYh4Re5xk3rqo,971
 sima_cli/utils/network.py,sha256=UvqxbqbWUczGFyO-t1SybG7Q-x9kjUVRNIn_D6APzy8,1252
-sima_cli-0.0.29.dist-info/licenses/LICENSE,sha256=a260OFuV4SsMZ6sQCkoYbtws_4o2deFtbnT9kg7Rfd4,1082
+sima_cli/utils/pkg_update_check.py,sha256=IAV_NAOsBDL_lYNYMRYfdZWuVq-rJ_zzHjJJZ7UQaoc,3274
+sima_cli-0.0.31.dist-info/licenses/LICENSE,sha256=a260OFuV4SsMZ6sQCkoYbtws_4o2deFtbnT9kg7Rfd4,1082
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_app_zoo.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_auth.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -53,8 +55,8 @@ tests/test_download.py,sha256=t87DwxlHs26_ws9rpcHGwr_OrcRPd3hz6Zmm0vRee2U,4465
 tests/test_firmware.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_model_zoo.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sima_cli-0.0.29.dist-info/METADATA,sha256=1MSfBv5taVOiFdadoLN9hElB33o0aKhHRgU-EgnH-e8,3705
-sima_cli-0.0.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sima_cli-0.0.29.dist-info/entry_points.txt,sha256=xRYrDq1nCs6R8wEdB3c1kKuimxEjWJkHuCzArQPT0Xk,47
-sima_cli-0.0.29.dist-info/top_level.txt,sha256=FtrbAUdHNohtEPteOblArxQNwoX9_t8qJQd59fagDlc,15
-sima_cli-0.0.29.dist-info/RECORD,,
+sima_cli-0.0.31.dist-info/METADATA,sha256=Q18Y7rOEEqe3ejP7IMqt5H6F3KipuWjUk15o_UwSIpg,3705
+sima_cli-0.0.31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sima_cli-0.0.31.dist-info/entry_points.txt,sha256=xRYrDq1nCs6R8wEdB3c1kKuimxEjWJkHuCzArQPT0Xk,47
+sima_cli-0.0.31.dist-info/top_level.txt,sha256=FtrbAUdHNohtEPteOblArxQNwoX9_t8qJQd59fagDlc,15
+sima_cli-0.0.31.dist-info/RECORD,,