remotivelabs-cli 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

cli/remotive.py

@@ -20,6 +20,7 @@ from cli.settings.migration.migrate_legacy_dirs import migrate_legacy_settings_d
 from cli.tools.tools import app as tools_app
 from cli.topology.cmd import app as topology_app
 from cli.typer import typer_utils
+from cli.utils.version_check import check_for_update
 
 err_console = Console(stderr=True)
 
@@ -46,7 +47,6 @@ def version_callback(value: bool) -> None:
     if value:
         my_version = version("remotivelabs-cli")
         typer.echo(my_version)
-        raise typer.Exit()
 
 
 def test_callback(value: int) -> None:
@@ -55,6 +55,10 @@ def test_callback(value: int) -> None:
         raise typer.Exit()
 
 
+def check_for_newer_version(settings: Settings) -> None:
+    check_for_update("remotivelabs-cli", version("remotivelabs-cli"), settings)
+
+
 def run_migrations(settings: Settings) -> None:
     """
     Run all migration scripts.
@@ -91,11 +95,12 @@ def main(
         None,
         "--version",
         callback=version_callback,
-        is_eager=False,
+        is_eager=True,
         help="Print current version",
     ),
 ) -> None:
     run_migrations(settings)
+    check_for_newer_version(settings)
     set_default_org_as_env(settings)
     # Do other global stuff, handle other global options here
 

cli/settings/core.py

@@ -7,20 +7,21 @@ from json import JSONDecodeError
 from pathlib import Path
 from typing import Optional, Tuple
 
+from pydantic import ValidationError
 from rich.console import Console
 
 from cli.errors import ErrorPrinter
 from cli.settings import config_file, token_file
 from cli.settings import token_file as tf
 from cli.settings.config_file import ConfigFile
+from cli.settings.state_file import StateFile
 from cli.settings.token_file import TokenFile
 
 err_console = Console(stderr=True)
 
-
 CONFIG_DIR_PATH = Path.home() / ".config" / "remotive"
 CLI_CONFIG_FILE_NAME = "config.json"
-
+CLI_INTERNAL_STATE_FILE_NAME = "app-state.json"
 
 TokenFileMetadata = Tuple[TokenFile, Path]
 
@@ -46,6 +47,10 @@ class Settings:
         self.config_file_path = self.config_dir / CLI_CONFIG_FILE_NAME
         if not self.config_file_path.exists():
             self._write_config_file(ConfigFile())
+        self.state_dir = self.config_dir / "state"
+        self.state_file_path = self.state_dir / CLI_INTERNAL_STATE_FILE_NAME
+        if not self.state_file_path.exists():
+            self.write_state_file(StateFile())
 
     def set_default_organisation(self, organisation: str) -> None:
         cli_config = self.get_cli_config()
@@ -166,6 +171,15 @@ class Settings:
 
         return file
 
+    def read_state_file(self) -> StateFile:
+        try:
+            return StateFile.loads(self.state_file_path.read_text(encoding="utf-8"))
+        except Exception:
+            return StateFile()
+
+    def write_state_file(self, state_file: StateFile) -> None:
+        self._write_file(self.state_file_path, state_file.dumps())
+
     def list_personal_tokens(self) -> list[TokenFile]:
         """
         List all personal tokens
@@ -234,7 +248,7 @@ class Settings:
             try:
                 self._read_token_file(path)
                 return True
-            except JSONDecodeError:
+            except (JSONDecodeError, ValidationError):
                 # TODO - this should be printed but printing it here causes it to be displayed to many times
                 # err_console.print(f"File is not valid json, skipping. {path}")
                 return False

cli/settings/state_file.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+import dataclasses
+import json
+from dataclasses import dataclass
+from datetime import datetime
+from typing import Optional
+
+from dacite import from_dict
+
+from cli.utils.time import parse_datetime
+
+
+@dataclass
+class StateFile:
+    version: str = "1.0"
+    last_update_check_time: Optional[str] = None
+
+    def dumps(self) -> str:
+        return json.dumps(dataclasses.asdict(self), default=str)
+
+    def should_perform_update_check(self) -> bool:
+        if self.last_update_check_time:
+            seconds = (datetime.now() - parse_datetime(self.last_update_check_time)).seconds
+            return (seconds / 3600) > 2
+        # This will solve the issue
+        return True
+
+    @staticmethod
+    def loads(data: str) -> StateFile:
+        d = json.loads(data)
+        return from_dict(StateFile, d)

cli/settings/token_file.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
-import dataclasses
-import json
 import re
-from dataclasses import dataclass
 from datetime import date, datetime
 from typing import Any, Literal
 
+from pydantic import BaseModel, EmailStr, Field, field_validator, model_validator
+
+from cli.utils.time import parse_date
+
 DEFAULT_EMAIL = "unknown@remotivecloud.com"
 PERSONAL_TOKEN_FILE_PREFIX = "personal-token-"
 SERVICE_ACCOUNT_TOKEN_FILE_PREFIX = "service-account-token-"
@@ -14,9 +15,9 @@ SERVICE_ACCOUNT_TOKEN_FILE_PREFIX = "service-account-token-"
 TokenType = Literal["authorized_user", "service_account"]
 
 
-def _parse_date(date_str: str) -> date:
-    normalized = date_str.replace("Z", "+00:00")
-    return datetime.fromisoformat(normalized).date()
+def _email_to_safe_filename(email: str) -> str:
+    """Replace any invalid character with an underscore"""
+    return re.sub(r'[<>:"/\\|?*]', "_", email)
 
 
 def _parse_token_type(token: str) -> TokenType:
@@ -27,75 +28,109 @@ def _parse_token_type(token: str) -> TokenType:
     raise ValueError(f"Unknown token type for token: {token}")
 
 
-def _from_dict(d: dict[str, Any]) -> TokenFile:
-    if "version" not in d:
-        token_type = _parse_token_type(d["token"])
-        return TokenFile(
-            version="1.0",
-            type=token_type,
-            name=d["name"],
-            token=d["token"],
-            created=_parse_date(d["created"]),
-            expires=_parse_date(d["expires"]),
-            account=TokenFileAccount(email=DEFAULT_EMAIL),
-        )
-
-    account_email = d.get("account", {}).get("email", DEFAULT_EMAIL)
-    return TokenFile(
-        version=d["version"],
-        type=d["type"],
-        name=d["name"],
-        token=d["token"],
-        created=_parse_date(d["created"]),
-        expires=_parse_date(d["expires"]),
-        account=TokenFileAccount(email=account_email),
-    )
+class TokenFileAccount(BaseModel):
+    email: EmailStr = DEFAULT_EMAIL
 
 
-def loads(data: str) -> TokenFile:
-    return _from_dict(json.loads(data))
+class TokenFile(BaseModel):
+    version: str = "1.0"
+    type: TokenType
+    name: str
+    token: str
+    created: date
+    expires: date
+    account: TokenFileAccount = Field(default_factory=TokenFileAccount)
 
+    @field_validator("created", "expires", mode="before")
+    @classmethod
+    def validate_parse_date(cls, value: str | date) -> date:
+        if isinstance(value, date):
+            return value
+        return parse_date(value)
 
-def dumps(token: TokenFile) -> str:
-    return json.dumps(dataclasses.asdict(token), default=str)
+    @model_validator(mode="before")
+    @classmethod
+    def init_with_defaults(cls, json_data: Any) -> Any:
+        """
+        Try to migrate old formats and missing fields as best we can.
 
+        NOTE: If we ever need to add a new version (like 2.0), we should add explicit classes for each version (e.g. TokenFileV1,
+        TokenFileV2, etc.), each with their own fields. This will allow us to migrate to new versions without breaking
+        backwards compatibility.
+        """
+        if not isinstance(json_data, dict):
+            return json_data
 
-@dataclass
-class TokenFileAccount:
-    email: str
+        if "version" not in json_data:
+            json_data["version"] = "1.0"
 
+        if "type" not in json_data and "token" in json_data:
+            json_data["type"] = _parse_token_type(json_data["token"])
 
-@dataclass
-class TokenFile:
-    version: str
-    type: TokenType
-    name: str
-    token: str
-    created: date
-    expires: date
-    account: TokenFileAccount
+        if "account" not in json_data:
+            json_data["account"] = {"email": DEFAULT_EMAIL}
+        elif isinstance(json_data["account"], str):
+            json_data["account"] = {"email": json_data["account"]}
 
-    def get_token_file_name(self) -> str:
-        def email_to_safe_filename(email: str) -> str:
-            # Replace any invalid character with an underscore
-            return re.sub(r'[<>:"/\\|?*]', "_", email)
+        return json_data
 
-        # From now, user will never be None when adding a token so in this case token_file.user is never None
-        email = email_to_safe_filename(self.account.email) if self.account is not None else "unknown"
+    def get_token_file_name(self) -> str:
+        """
+        Returns the name of the token file using the proper file name format.
+        """
+        email = _email_to_safe_filename(self.account.email) if self.account is not None else "unknown"
         if self.type == "authorized_user":
             return f"{PERSONAL_TOKEN_FILE_PREFIX}{self.name}-{email}.json"
         return f"{SERVICE_ACCOUNT_TOKEN_FILE_PREFIX}{self.name}-{email}.json"
 
     def is_expired(self) -> bool:
+        """
+        Returns True if the token is expired, False otherwise.
+        """
         return datetime.today().date() > self.expires
 
     def expires_in_days(self) -> int:
+        """
+        Returns the number of days until the token expires.
+        """
         return (self.expires - datetime.today().date()).days
 
-    @staticmethod
-    def from_json_str(data: str) -> TokenFile:
-        return loads(data)
+    @classmethod
+    def from_json_str(cls, data: str) -> TokenFile:
+        """
+        Creates a TokenFile from a JSON string.
+        """
+        return cls.model_validate_json(data)
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> TokenFile:
+        """
+        Creates a TokenFile from a dictionary.
+        """
+        return cls.model_validate(data)
+
+    def to_json_str(self) -> str:
+        """
+        Returns the JSON string representation of the TokenFile.
+        """
+        return self.model_dump_json()
+
+    def to_dict(self) -> dict[str, Any]:
+        """
+        Returns the dictionary representation of the TokenFile.
+        """
+        return self.model_dump()
+
+
+def loads(data: str) -> TokenFile:
+    """
+    Creates a TokenFile from a JSON string.
+    """
+    return TokenFile.from_json_str(data)
+
 
-    @staticmethod
-    def from_dict(data: dict[str, Any]) -> TokenFile:
-        return _from_dict(data)
+def dumps(token_file: TokenFile) -> str:
+    """
+    Returns the JSON string representation of the TokenFile.
+    """
+    return token_file.to_json_str()

cli/typer/typer_utils.py

@@ -17,7 +17,7 @@ console = Console()
 def create_typer(**kwargs: Any) -> typer.Typer:
     """Create a Typer instance with default settings."""
     # return typer.Typer(no_args_is_help=True, **kwargs)
-    return typer.Typer(cls=OrderCommands, no_args_is_help=True, **kwargs)
+    return typer.Typer(cls=OrderCommands, no_args_is_help=True, invoke_without_command=True, **kwargs)
 
 
 def print_padded(label: str, right_text: str, length: int = 30) -> None:

cli/utils/time.py

@@ -0,0 +1,11 @@
+from datetime import date, datetime
+
+
+def parse_date(date_str: str) -> date:
+    return parse_datetime(date_str).date()
+
+
+def parse_datetime(date_str: str) -> datetime:
+    """Required for pre 3.11"""
+    normalized = date_str.replace("Z", "+00:00")
+    return datetime.fromisoformat(normalized)

cli/utils/version_check.py

@@ -0,0 +1,112 @@
+from __future__ import annotations
+
+import datetime
+import json
+import os
+import urllib.request
+from datetime import timedelta
+from importlib import metadata as importlib_metadata
+
+from packaging.version import InvalidVersion, Version
+
+from cli.errors import ErrorPrinter
+from cli.settings import Settings
+
+
+def _pypi_latest(
+    project: str, *, include_prereleases: bool, timeout: float = 2.5, user_agent: str | None = None
+) -> tuple[str | None, str | None]:
+    """Return (latest_version, project_url) from PyPI, skipping yanked files."""
+    url = f"https://pypi.org/pypi/{project}/json"
+    headers = {"Accept": "application/json"}
+    if user_agent:
+        headers["User-Agent"] = user_agent
+    req = urllib.request.Request(url, headers=headers)
+
+    try:
+        with urllib.request.urlopen(req, timeout=timeout) as resp:
+            data = json.load(resp)
+    except Exception:
+        return None, None  # network/404/etc.
+
+    releases = data.get("releases") or {}
+    candidates: list[Version] = []
+    for s, files in releases.items():
+        try:
+            v = Version(s)
+        except InvalidVersion:
+            continue
+        the_files = files or []
+        if any(f.get("yanked", False) for f in the_files):
+            continue
+        if (v.is_prerelease or v.is_devrelease) and not include_prereleases:
+            continue
+        candidates.append(v)
+
+    if not candidates:
+        return None, None
+
+    latest = str(max(candidates))
+    info = data.get("info") or {}
+    proj_url = info.get("project_url") or info.get("package_url") or f"https://pypi.org/project/{project}/"
+    return latest, proj_url
+
+
+def _installed_version(distribution_name: str, fallback: str | None = None) -> str | None:
+    try:
+        return importlib_metadata.version(distribution_name)
+    except importlib_metadata.PackageNotFoundError:
+        return fallback
+
+
+def check_for_update(project: str, current_version: str, settings: Settings) -> None:
+    # Make it possible to disable update check, i.e in CI
+    if os.environ.get("PYTHON_DISABLE_UPDATE_CHECK"):
+        return
+
+    # Determine current version
+    cur = current_version or _installed_version(project)
+    if not cur:
+        return  # unknown version → skip silently
+
+    state = settings.read_state_file()
+
+    if not state.last_update_check_time:
+        if os.environ.get("RUNS_IN_DOCKER"):
+            # To prevent that we always check update in docker due to ephemeral disks we write an "old" check if the state
+            # is missing. If no disk is mounted we will never get the update check but if its mounted properly we will get
+            # it on the second attempt. This is good enough
+            state.last_update_check_time = (datetime.datetime.now() - timedelta(hours=10)).isoformat()
+            settings.write_state_file(state)
+            return
+
+    elif not state.should_perform_update_check():
+        return
+
+    # We end up here if last_update_check_time is None or should_perform_update_check is true
+
+    include_prereleases = Version(cur).is_prerelease or Version(cur).is_devrelease
+
+    latest, proj_url = _pypi_latest(
+        project, include_prereleases=include_prereleases, user_agent=f"{project}/{cur} (+https://pypi.org/project/{project}/)"
+    )
+    if latest:
+        if Version(latest) > Version(cur):
+            _print_update_info(
+                cur,
+                latest,
+            )
+    state.last_update_check_time = datetime.datetime.now().isoformat()
+    settings.write_state_file(state)
+
+
+def _print_update_info(cur: str, latest: str) -> None:
+    instructions = (
+        "upgrade with: docker pull remotivelabs/remotivelabs-cli"
+        if os.environ.get("RUNS_IN_DOCKER")
+        else "upgrade with: pipx install -U remotivelabs-cli"
+    )
+
+    ErrorPrinter.print_hint(
+        f"Update available: remotivelabs-cli {cur} → {latest} , ({instructions}) we always recommend to use latest version"
+    )

{remotivelabs_cli-0.2.1.dist-info → remotivelabs_cli-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: remotivelabs-cli
-Version: 0.2.1
+Version: 0.2.2
 Summary: CLI for operating RemotiveCloud and RemotiveBroker
 Author: Johan Rask
 Author-email: johan.rask@remotivelabs.com
@@ -13,9 +13,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: click (<8.2.0)
 Requires-Dist: dacite (>=1.9.2,<2.0.0)
+Requires-Dist: email-validator (>=2.2.0,<3.0.0)
 Requires-Dist: grpc-stubs (>=1.53.0.5)
 Requires-Dist: mypy-protobuf (>=3.0.0)
 Requires-Dist: plotext (>=5.2,<6.0)
+Requires-Dist: pydantic (>=2.11.7,<3.0.0)
 Requires-Dist: pyjwt (>=2.6,<3.0)
 Requires-Dist: python-can (>=4.3.1)
 Requires-Dist: python-socketio (>=4.6.1)

{remotivelabs_cli-0.2.1.dist-info → remotivelabs_cli-0.2.2.dist-info}/RECORD

@@ -37,28 +37,31 @@ cli/connect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/connect/connect.py,sha256=SH2DNTTVLu2dNpk6xIah1-KJZAqrK_7Skt8RKp8Mjh8,4231
 cli/connect/protopie/protopie.py,sha256=ElmrGaV0ivb85wo0gLzCAXZhmSmIDASaCVlF1iQblLI,6532
 cli/errors.py,sha256=djODw6sdMJXzOsuAUOP3N13nfmm1sIP3Pe6tllGdozM,1657
-cli/remotive.py,sha256=FkXJCq0mw_bak75u6aukleIld-jjgn4hjuBQ28Lcz1U,3831
+cli/remotive.py,sha256=xfeekzG6tojXsWZdiGN5ceKCCd8xVNf8v9EUAY2Gnjc,4033
 cli/settings/__init__.py,sha256=t1qkaGrJ4xx8WMHlmBTbQ1VdJL4YOcz8VFfRkGa2_jQ,711
 cli/settings/config_file.py,sha256=6sdHUtZSUIgubwpfwEEn7GarTK1M_iQhtRJZzFDdP5o,2784
-cli/settings/core.py,sha256=FqME9ghV6bBNDrZnQCm3jxz-7T8M04dSFumYK1-QOhs,10911
+cli/settings/core.py,sha256=IJ62CzPrrvwO46zmvsjKIn6VD4oR9VG5IX29ctd2RO4,11611
 cli/settings/migration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/settings/migration/migrate_all_token_files.py,sha256=xoVvAqn_tGskEW148uf3xZx1mpJKUnERMTcBo0nkCnI,3010
 cli/settings/migration/migrate_config_file.py,sha256=hw4EpRwJz1zUNxfCOk0PvMuZjAlaGy4m_rDbMsHZO_w,2047
 cli/settings/migration/migrate_legacy_dirs.py,sha256=N0t2io3bT_ub8BcVPw1CeQ4eeexRUiu3jXq3DL018OE,1819
 cli/settings/migration/migrate_token_file.py,sha256=Fp7Z_lNqSdoWY05TYwFW2QH8q9QhmB2TYSok6hV1Mic,1530
 cli/settings/migration/migration_tools.py,sha256=P72tuw6-aS_Kd0qn-0ZecplsYxMTu0LTXM5sMSNTVEM,1378
-cli/settings/token_file.py,sha256=llfP1GohXtLVIDy3UyNB8FneGRuCesnce9uasv-7XKM,2953
+cli/settings/state_file.py,sha256=ujTOJgCts-gpM-66EhRYXwO803HAFcmia2Pf_nYGldc,837
+cli/settings/token_file.py,sha256=Po3Vwu5cdT5ZgLO3_ZLEX13_57coqHz1PPu2SQ-202o,4177
 cli/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/tools/can/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/tools/can/can.py,sha256=TtP5w8vb0QG4ObNhkWIDpRMdNelirFffoc_lFZy8ePM,2260
 cli/tools/tools.py,sha256=jhLfrFDqkmWV3eBAzNwBf6WgDGrz7sOhgVCia36Twn8,232
 cli/topology/cmd.py,sha256=SQ5wi7KDoh4iR2Ed7gyfGLNj6UE0K6UkksmBMSD2XAk,3981
 cli/typer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cli/typer/typer_utils.py,sha256=8SkvG9aKkfK9fTRsLD9pOBtWn9XSwtOXWg2RAk9FhOI,708
+cli/typer/typer_utils.py,sha256=TaJuK1EtE9Gv3DfmoyHPTNKmhiAimuQCHKxQjnUZ7bs,737
 cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cli/utils/rest_helper.py,sha256=De-1Z75p-zjA4hZrylVxWn2wqdPB2gvyvA-ixm_dRFo,14141
-remotivelabs_cli-0.2.1.dist-info/LICENSE,sha256=qDPP_yfuv1fF-u7EfexN-cN3M8aFgGVndGhGLovLKz0,608
-remotivelabs_cli-0.2.1.dist-info/METADATA,sha256=gJSJmVNokErSbGtxifX0Z2uMMFdXy77NC4LLD6Owx8A,1428
-remotivelabs_cli-0.2.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-remotivelabs_cli-0.2.1.dist-info/entry_points.txt,sha256=lvDhPgagLqW_KTnLPCwKSqfYlEp-1uYVosRiPjsVj10,45
-remotivelabs_cli-0.2.1.dist-info/RECORD,,
+cli/utils/time.py,sha256=TEKcNZ-pQoJ7cZ6hQmVD0sTRwRm2rBy51-MuDNdO4S4,296
+cli/utils/version_check.py,sha256=sAkTzNRlgGn4Hto_56J0KHLe-3zK7he7gbUTjrtY1lc,3940
+remotivelabs_cli-0.2.2.dist-info/LICENSE,sha256=qDPP_yfuv1fF-u7EfexN-cN3M8aFgGVndGhGLovLKz0,608
+remotivelabs_cli-0.2.2.dist-info/METADATA,sha256=Dyz6ht0Psz5TCHsxJXxsKIWckmwKEPVsO3amT4GkDng,1518
+remotivelabs_cli-0.2.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+remotivelabs_cli-0.2.2.dist-info/entry_points.txt,sha256=lvDhPgagLqW_KTnLPCwKSqfYlEp-1uYVosRiPjsVj10,45
+remotivelabs_cli-0.2.2.dist-info/RECORD,,