feat: Add --version CLI option and setting User-Agent HTTP header

Author: kukovecz

onekey_client/__init__.py

@@ -1,3 +1,5 @@
+from .client import APP_NAME as APP_NAME
+from .client import APP_VERSION as APP_VERSION
 from .client import Client as Client
 from .models import FirmwareMetadata as FirmwareMetadata
 from .models import Tenant as Tenant

onekey_client/cli/cli.py

@@ -3,14 +3,15 @@
 import click
 import httpx
 
-from onekey_client import Client
+from onekey_client import APP_NAME, APP_VERSION, Client
 
 from .ci import ci_result
 from .firmware_upload import upload_firmware
 from .misc import get_tenant_token, list_tenants
 
 
 @click.group()
+@click.version_option(version=APP_VERSION, prog_name=APP_NAME)
 @click.option(
     "--api-url",
     default="https://app.eu.onekey.com/api",

onekey_client/client.py

@@ -3,6 +3,7 @@
 import secrets
 import time
 from importlib import resources
+from importlib.metadata import version
 from pathlib import Path
 
 import httpx
@@ -17,6 +18,8 @@
 
 CLIENT_ID = "ONEKEY Python SDK"
 TOKEN_NAMESPACE = "https://www.onekey.com/"  # noqa: S105 (hardcoded credential)
+APP_NAME="onekey_client"
+APP_VERSION=version(APP_NAME)
 
 
 def _login_required(func):
@@ -63,17 +66,18 @@ def _setup_httpx_client(
         ca_bundle: Path | None = None,
         disable_tls_verify: bool | None = False,
     ):
+        headers = {"User-Agent": f"{APP_NAME}/{APP_VERSION}"}
         if disable_tls_verify:
-            return httpx.Client(base_url=api_url, verify=False)  # noqa: S501 (TLS certificate validation disabled)
+            return httpx.Client(base_url=api_url, headers=headers, verify=False)  # noqa: S501 (TLS certificate validation disabled)
 
         if ca_bundle is not None:
             ca = ca_bundle.expanduser()
             if not ca.exists():
                 raise errors.InvalidCABundle
 
-            return httpx.Client(base_url=api_url, verify=str(ca))
+            return httpx.Client(base_url=api_url, headers=headers, verify=str(ca))
         with resources.path(keys, "ca.pem") as ca:
-            return httpx.Client(base_url=api_url, verify=str(ca))
+            return httpx.Client(base_url=api_url, headers=headers, verify=str(ca))
 
     def _load_key(self, key_name: str, path: Path | None = None):
         if path is not None: