datalab-python-sdk 0.6.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

{datalab_python_sdk-0.6.0.dist-info → datalab_python_sdk-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datalab-python-sdk
-Version: 0.6.0
+Version: 0.7.0
 Summary: SDK for the Datalab document intelligence API
 Project-URL: Homepage, https://www.datalab.to
 Project-URL: Documentation, https://documentation.datalab.to
@@ -41,6 +41,14 @@ pip install datalab-python-sdk
 export DATALAB_API_KEY="your_api_key_here"
 ```
 
+Or sign in from a terminal, which also works for coding agents:
+
+```bash
+datalab login
+```
+
+It shows a code and a link. Open the link, sign in or sign up, approve the code, and type the 6-digit code that we email you. The SDK saves the key to `~/.config/datalab/config.json`, readable only by your user, and uses it when `DATALAB_API_KEY` is not set. `datalab logout` deletes it.
+
 ### Basic Usage
 
 ```python

datalab_python_sdk-0.7.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+datalab_sdk/__init__.py,sha256=BLAqVuqjxhdjKabkDUePLSmGeVMo0BnWfVbLDt8haP4,2397
+datalab_sdk/auth.py,sha256=Z7eaKSxhDFt2uUbSIlU8yuOC2myDWXl-rY9JLuThg4A,4278
+datalab_sdk/cli.py,sha256=icUy91brz0J8hzvC1G-pBtZPszc5oRd4H3sEpYhKr7Y,54186
+datalab_sdk/client.py,sha256=aNPW4_gUdyW7UbLHI3YRzPA_Ae2dQmlQwCV38qsw0hA,125702
+datalab_sdk/exceptions.py,sha256=5-lzlfvB7FKGOhvgyacceTFRRUxton_JqcIyuB8k_mU,2625
+datalab_sdk/mimetypes.py,sha256=a4ULJmHxoR_RMsD0j51hx0t2Q7In_PzNm7UlWX07eEE,932
+datalab_sdk/models.py,sha256=OpiIAdOVcL2tdfBkzRNHZ8MjGjRiBSjaKtFVf_U1l8w,32494
+datalab_sdk/settings.py,sha256=B_B42QQADAz6ePXQTOqENZllcppFR4JpwonEk2vlzhw,693
+datalab_python_sdk-0.7.0.dist-info/METADATA,sha256=z3hdjWxjWWjORewPlI58rO_stgNdHKUkwvf0Y4JemRg,6963
+datalab_python_sdk-0.7.0.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+datalab_python_sdk-0.7.0.dist-info/entry_points.txt,sha256=rqWxO21UV57JxxyeF20iBTxHpzBYOA69hCHIqwdn-Ok,48
+datalab_python_sdk-0.7.0.dist-info/licenses/LICENSE,sha256=XGjT-Uz6t7UZzr3JCZfilsGcZEgMm_1aONcRunohxWg,1064
+datalab_python_sdk-0.7.0.dist-info/RECORD,,

{datalab_python_sdk-0.6.0.dist-info → datalab_python_sdk-0.7.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.32.4
+Generator: hatchling 1.33.0
 Root-Is-Purelib: true
 Tag: py3-none-any

datalab_sdk/auth.py

@@ -0,0 +1,130 @@
+"""Sign in from a terminal with a device code (`datalab login`), and keep the API key on disk.
+
+The flow (server side: /api/auth/device/code, /app/device, /api/auth/device/token):
+1. Ask the server for a code. Show the human the code and a URL, and open the browser.
+2. The human approves in the browser, where they are signed in as usual.
+3. Poll until the server returns a new API key, then save it to the config file.
+
+The key is saved to $XDG_CONFIG_HOME/datalab/config.json (default ~/.config/datalab/config.json)
+with mode 0600. DATALAB_API_KEY in the environment always wins over the saved key.
+"""
+
+from __future__ import annotations
+
+import json
+import os
+import stat
+import time
+import urllib.error
+import urllib.request
+import webbrowser
+from pathlib import Path
+from typing import Callable, Optional
+
+from datalab_sdk.exceptions import DatalabAPIError
+
+KEYS_PAGE = "https://www.datalab.to/app/keys"
+
+
+def config_path() -> Path:
+    base = os.environ.get("XDG_CONFIG_HOME") or os.path.join(Path.home(), ".config")
+    return Path(base) / "datalab" / "config.json"
+
+
+def load_saved_api_key() -> Optional[str]:
+    try:
+        return json.loads(config_path().read_text()).get("api_key") or None
+    except (OSError, ValueError):
+        return None
+
+
+def save_api_key(api_key: str, host: str, team_name: Optional[str] = None) -> Path:
+    path = config_path()
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(
+        json.dumps({"api_key": api_key, "host": host, "team_name": team_name}, indent=2)
+    )
+    os.chmod(path, stat.S_IRUSR | stat.S_IWUSR)
+    return path
+
+
+def delete_saved_api_key() -> bool:
+    try:
+        config_path().unlink()
+        return True
+    except FileNotFoundError:
+        return False
+
+
+def _post(url: str, payload: dict, headers: dict) -> tuple[int, dict]:
+    req = urllib.request.Request(
+        url,
+        data=json.dumps(payload).encode(),
+        headers={
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+            **headers,
+        },
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=30) as res:
+            return res.status, json.loads(res.read() or b"{}")
+    except urllib.error.HTTPError as e:
+        try:
+            return e.code, json.loads(e.read() or b"{}")
+        except ValueError:
+            return e.code, {}
+
+
+def device_login(
+    host: str = "https://www.datalab.to",
+    client_name: str = "Datalab CLI",
+    open_browser: bool = True,
+    extra_headers: Optional[dict] = None,
+    on_code: Optional[Callable[[dict], None]] = None,
+    sleep: Callable[[float], None] = time.sleep,
+) -> dict:
+    """Run the device flow and return {"api_key", "team_name", "key_name"}. Raises DatalabAPIError."""
+    host = host.rstrip("/")
+    headers = extra_headers or {}
+    status, code = _post(
+        f"{host}/api/auth/device/code", {"client_name": client_name}, headers
+    )
+    if status != 200:
+        raise DatalabAPIError(
+            f"Could not start the sign-in ({status}): {code.get('detail', code)}"
+        )
+    if on_code:
+        on_code(code)
+    if open_browser:
+        try:
+            webbrowser.open(code["verification_uri_complete"])
+        except Exception:
+            pass
+
+    interval = float(code.get("interval", 5))
+    deadline = time.monotonic() + float(code.get("expires_in", 600))
+    while time.monotonic() < deadline:
+        sleep(interval)
+        status, body = _post(
+            f"{host}/api/auth/device/token",
+            {"device_code": code["device_code"]},
+            headers,
+        )
+        if status == 200 and body.get("api_key"):
+            return body
+        error = body.get("error")
+        if error == "authorization_pending":
+            continue
+        if error == "slow_down":
+            interval += 5
+            continue
+        if error == "access_denied":
+            raise DatalabAPIError("The sign-in was denied in the browser.")
+        if error == "expired_token":
+            raise DatalabAPIError(
+                "The sign-in code expired. Run `datalab login` again."
+            )
+        raise DatalabAPIError(f"Sign-in failed ({status}): {body}")
+    raise DatalabAPIError("The sign-in code expired. Run `datalab login` again.")

datalab_sdk/cli.py

@@ -495,6 +495,55 @@ def cli():
     pass
 
 
+@click.command()
+@click.option(
+    "--host",
+    default=None,
+    help="API host (default: DATALAB_HOST or https://www.datalab.to)",
+)
+@click.option(
+    "--no-browser",
+    is_flag=True,
+    help="Print the sign-in URL instead of opening a browser",
+)
+def login(host: Optional[str], no_browser: bool):
+    """Sign in from this terminal and save an API key (device code)."""
+    from datalab_sdk.auth import device_login, save_api_key
+
+    host = (host or settings.DATALAB_HOST).rstrip("/")
+
+    def show(code: dict):
+        click.echo("To sign in, open this page and approve the code:")
+        click.echo(f"  {code['verification_uri_complete']}")
+        click.echo(f"  Code: {code['user_code']}")
+        click.echo("Then type the 6-digit code that we email you.")
+        click.echo("Waiting for approval...")
+
+    try:
+        result = device_login(host=host, open_browser=not no_browser, on_code=show)
+    except DatalabError as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    path = save_api_key(result["api_key"], host, result.get("team_name"))
+    click.echo(f"Signed in to team {result.get('team_name')}. API key saved to {path}.")
+    click.echo(
+        "DATALAB_API_KEY in your environment still takes priority over the saved key."
+    )
+
+
+@click.command()
+def logout():
+    """Delete the API key that `datalab login` saved. The key itself stays valid until you revoke it."""
+    from datalab_sdk.auth import config_path, delete_saved_api_key
+
+    if delete_saved_api_key():
+        click.echo(
+            f"Removed {config_path()}. Revoke the key at https://www.datalab.to/app/keys if you no longer need it."
+        )
+    else:
+        click.echo("No saved API key.")
+
+
 @click.command()
 @click.argument("path", type=click.Path(exists=True))
 @common_options
@@ -1584,6 +1633,8 @@ def _render_dag_simple(layers, children, step_map):
 
 
 # Add commands to CLI group
+cli.add_command(login)
+cli.add_command(logout)
 cli.add_command(convert)
 cli.add_command(extract)
 cli.add_command(segment)

datalab_sdk/client.py

@@ -201,9 +201,13 @@ class AsyncDatalabClient:
         """
         if api_key is None:
             api_key = settings.DATALAB_API_KEY
+        if api_key is None:
+            from datalab_sdk.auth import load_saved_api_key
+
+            api_key = load_saved_api_key()
         if api_key is None:
             raise DatalabAPIError(
-                "You must pass in an api_key or set DATALAB_API_KEY. "
+                "No API key. Run `datalab login`, set DATALAB_API_KEY, or pass api_key. "
                 "Get a free API key at https://www.datalab.to/app/keys"
             )
 

datalab_python_sdk-0.6.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-datalab_sdk/__init__.py,sha256=BLAqVuqjxhdjKabkDUePLSmGeVMo0BnWfVbLDt8haP4,2397
-datalab_sdk/cli.py,sha256=DAI7fzs_1cqEctp9N_rw9VPuiIFHgN75w41FG55HxrM,52427
-datalab_sdk/client.py,sha256=PvcNA7ff6yqIKvLkspG2XzSYMStYKlvAIkZRNDNblWg,125551
-datalab_sdk/exceptions.py,sha256=5-lzlfvB7FKGOhvgyacceTFRRUxton_JqcIyuB8k_mU,2625
-datalab_sdk/mimetypes.py,sha256=a4ULJmHxoR_RMsD0j51hx0t2Q7In_PzNm7UlWX07eEE,932
-datalab_sdk/models.py,sha256=OpiIAdOVcL2tdfBkzRNHZ8MjGjRiBSjaKtFVf_U1l8w,32494
-datalab_sdk/settings.py,sha256=B_B42QQADAz6ePXQTOqENZllcppFR4JpwonEk2vlzhw,693
-datalab_python_sdk-0.6.0.dist-info/METADATA,sha256=E2qTxFzvZarV8CeVwvGR7UW-EA6lUn_iLG9NlIT450s,6582
-datalab_python_sdk-0.6.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
-datalab_python_sdk-0.6.0.dist-info/entry_points.txt,sha256=rqWxO21UV57JxxyeF20iBTxHpzBYOA69hCHIqwdn-Ok,48
-datalab_python_sdk-0.6.0.dist-info/licenses/LICENSE,sha256=XGjT-Uz6t7UZzr3JCZfilsGcZEgMm_1aONcRunohxWg,1064
-datalab_python_sdk-0.6.0.dist-info/RECORD,,