dataerai-sdk 0.2.0b1

Python SDK for the Dataerai transfer daemon

Dataerai Python SDK

Python client for the Dataerai transfer daemon.
Supports authenticated uploads, downloads, and metadata operations. The daemon is started automatically if it is not already running.

Requirements

• Python ≥ 3.10
• The dataerai binary must be installed and on PATH (or pass binary_path explicitly)
• The user must be logged in via dataerai auth login before calling auth_status() / upload() / download()

Installation

pip install dataerai-sdk
# or, from source:
pip install -e sdk/python/

Quick start

from dataerai import DataeraiClient

with DataeraiClient(binary_path="/usr/local/bin/dataerai") as client:
    # Check auth
    status = client.auth_status()
    print(f"Logged in as {status.user_email}, token expires {status.expires_at}")

    # Create a project to own your data (also provisions its root collection)
    project = client.create_project("My dataset project", description="Demo")
    print(f"Created project {project.project_id}")

    # Upload a file into it
    result = client.upload(
        "/path/to/data.csv",
        title="My dataset",
        owner_type="project",
        owner_id=project.project_id,
        on_progress=lambda p: print(f"  {p.percent:.0f}%  {p.rate_mbps:.1f} MB/s"),
    )
    print(f"Uploaded  asset_id={result.asset_id}  content_id={result.content_id}")

    # Download it back
    dl = client.download(result.asset_id, dest_dir="/tmp/downloads")
    for f in dl.files:
        print(f"  {f.local_path}  ({f.size:,} bytes)")

    # Read / update metadata
    meta = client.get_metadata(result.asset_id)
    updated = client.set_metadata(result.asset_id, title="My dataset v2", tags=["csv", "demo"])

API reference

DataeraiClient(*, socket_path, binary_path, auto_start, start_timeout_s, request_timeout_s)

Parameter	Default	Description
socket_path	$DATAERAI_SOCKET or /run/user/<uid>/dataerai-transfer.sock	Unix socket path
binary_path	None	Path to the dataerai binary (required for auto_start)
auto_start	True	Spawn the daemon if the socket does not exist
start_timeout_s	10.0	Seconds to wait for the daemon socket to appear
request_timeout_s	30.0	Per-request timeout in seconds

Use as a context manager (with DataeraiClient(...) as client:) for automatic cleanup, or call client.connect() / client.close() manually.

Methods

Method	Returns	Description
auth_status()	AuthStatus	Logged-in user and token expiry
create_project(name, *, description="")	Project	Create a project (+ root collection, owner membership, default allocation); requires write scope
upload(local_path, *, title, owner_type, owner_id, ...)	UploadResult	Upload a file; blocks until complete
download(asset_id, dest_dir, ...)	DownloadResult	Download latest asset content; blocks until complete
get_metadata(asset_id)	AssetMetadata	Retrieve asset metadata
set_metadata(asset_id, **fields)	AssetMetadata	Update metadata fields

upload() keyword arguments

Argument	Type	Description
title	str	Asset title (required)
owner_type	str	"project" or "collection" (required)
owner_id	str	Owner entity ID (required)
description	str | None	Free-text description
alias	str | None	Short identifier
tags	list[str] | None	Tag list
metadata	dict | None	Arbitrary key-value metadata
collection_id	str | None	Collection to add the asset to
chunk_size_mb	int | None	Override default 64 MiB chunk size
on_progress	Callable[[ProgressEvent], None] | None	Progress callback
transfer_timeout_s	float	Max seconds to wait for completion (default 3600)

Progress events

@dataclass
class ProgressEvent:
    transfer_id: str
    bytes_done: int
    bytes_total: int
    chunk_index: int
    chunk_count: int
    rate_bps: float
    file_index: int
    file_name: str

    @property
    def percent(self) -> float: ...   # 0–100

    @property
    def rate_mbps(self) -> float: ...

Error types

Exception	When raised
DaemonError(code, message)	Daemon returned a coded error (see code attribute)
DaemonTimeoutError	Request or transfer exceeded the configured timeout
ConnectionError	Daemon disconnected unexpectedly