fcloud-sdk 0.1.0__py3-none-any.whl

fcloud/__init__.py

@@ -0,0 +1,85 @@
+"""fcloud Python SDK — launch code on GPU and CPU hosts in seconds.
+
+Setup (one time, from terminal):
+    $ pip install fcloud-sdk
+    $ fcloud set_token fcloud_sk_...
+
+Usage:
+    from fcloud import Client, Image
+
+    client = Client()  # uses saved token + default URL
+
+    image = Image.debian_slim().pip_install(["torch"])
+    project = client.project("my-project", image=image)
+    with project.session(sku="gpu_1x_a100") as s:
+        s.run(["python", "train.py"])
+        proc = s.spawn(["python", "eval.py"])
+        proc.wait()
+        print(s.run(["cat", "results.json"]).stdout)
+"""
+
+import fcloud._legacy_env  # noqa: F401 — must run before any env read below
+from fcloud.client import (
+    Client,
+    FcloudError,
+    PaymentOverdueError,
+    SessionNotFoundError,
+)
+from fcloud.config import get_token, set_token
+from fcloud.errors import (
+    ClientTooOldError,
+    FcloudRuntimeError,
+    FoomError,
+    RateLimitedError,
+    VolumeHydrationError,
+)
+from fcloud.image import Image
+from fcloud.job import Job  # noqa: F401
+from fcloud.session import Project, Session
+from fcloud.sweeps import MapSpecError, SweepHandle  # noqa: F401
+from fcloud.types import (
+    ExecResult,
+    Process,
+    ProcessLog,
+    ProcessStatus,
+    SessionInfo,
+)
+from fcloud.version import VERSION as __version__
+from fcloud.volumes import (
+    VolumeConflictError,
+    VolumeIsDirectoryError,
+    VolumeNotFoundError,
+)
+
+__all__ = [
+    "__version__",
+    # Entry points
+    "Client",
+    "FoomError",
+    "Image",
+    "set_token",
+    "get_token",
+    # Handles
+    "Project",
+    "Session",
+    "Job",
+    "SweepHandle",
+    # Result / record types
+    "ExecResult",
+    "Process",
+    "ProcessLog",
+    "ProcessStatus",
+    "SessionInfo",
+    # Errors
+    "ClientTooOldError",
+    "FcloudError",
+    "FcloudRuntimeError",
+    "MapSpecError",
+    "PaymentOverdueError",
+    "RateLimitedError",
+    "SessionNotFoundError",
+    "VolumeConflictError",
+    "VolumeHydrationError",
+    "VolumeIsDirectoryError",
+    "VolumeNotFoundError",
+]

fcloud/__main__.py

@@ -0,0 +1,150 @@
+"""fcloud CLI — GPU compute from the command line.
+
+Compatibility shim. The CLI lives in the `fcloud.cli` package (one module per
+`fcloud help` group); importing it registers every command. This module keeps
+the historical `fcloud.__main__` import surface — the console script, the
+command registry, every `cmd_*` handler and the helpers tests reach for —
+and runs `main()` under `python -m fcloud`.
+
+New code should import from the owning `fcloud.cli.*` module directly. Note
+that `_JSON_OUTPUT` is NOT re-exported: it is a mutable flag, and a copy here
+would silently diverge from the one the commands read. Patch and read
+`fcloud.cli.output._JSON_OUTPUT` instead (see `__getattr__` below).
+"""
+
+from fcloud.cli.attach import (  # noqa: F401
+    _ATTACH_RETRY_DELAYS_S,
+    _ENDED_SESSION_STATES,
+    _RESUME_BUDGET_S,
+    _TERMINAL_SESSION_STATES,
+    _attach_with_retry,
+    _authoritative_session_status,
+    _exit_session_unreachable,
+    _get_session_handle,
+)
+from fcloud.cli.common import (  # noqa: F401
+    _consume_positional_session,
+    _extract_flag,
+    _extract_secrets,
+    _extract_upload_options,
+    _parse_duration_seconds,
+    _parse_keep_warm,
+    _parse_min_disk,
+    _parse_volume_mount,
+    _print_upload_summary,
+    _resolve_session_and_rest,
+    _shape_exec_command,
+    _split_trailing_args,
+)
+from fcloud.cli.console import (  # noqa: F401
+    _exec_result,
+    _live_console_sink,
+    _maybe_keepalive,
+    _pid_emitter,
+)
+from fcloud.cli.context import (  # noqa: F401
+    _api_base,
+    _api_key,
+    _resolve_min_disk_gb,
+    _resolve_project,
+    _resolve_volume_mounts,
+)
+from fcloud.cli.exec_cmd import _parse_exec_args, cmd_exec  # noqa: F401
+from fcloud.cli.files import (  # noqa: F401
+    _download_file,
+    _volume_path_hint,
+    cmd_download,
+    cmd_ls,
+    cmd_upload,
+)
+from fcloud.cli.hardware import cmd_health, cmd_skus  # noqa: F401
+from fcloud.cli.help import (  # noqa: F401
+    _HELP_GROUPS,
+    print_command_help,
+    print_help,
+)
+from fcloud.cli.interactive import _parse_shell_args, cmd_shell  # noqa: F401
+from fcloud.cli.job import cmd_job  # noqa: F401
+from fcloud.cli.main import _report_cli_error, _strip_precommand_json, main  # noqa: F401
+from fcloud.cli.migration import (  # noqa: F401
+    _drain_resumed_process,
+    _follow_reconnect,
+    _maybe_respawn_after_rebuild,
+    _migration_ui,
+    _ride_migration,
+    _tail_follow_delta,
+)
+from fcloud.cli.mount import cmd_mount  # noqa: F401
+from fcloud.cli.output import _out  # noqa: F401
+from fcloud.cli.processes import (  # noqa: F401
+    _emit_spawned,
+    _logs_follow,
+    _parse_logs_args,
+    _print_log_result,
+    cmd_kill,
+    cmd_logs,
+    cmd_spawn,
+)
+from fcloud.cli.registry import (  # noqa: F401
+    COMMANDS,
+    HIDDEN_COMMANDS,
+    LONG_HELP,
+    cmd,
+)
+from fcloud.cli.run import (  # noqa: F401
+    _infer_runner,
+    _parse_run_args,
+    _spawn_detached_run,
+    cmd_run,
+)
+from fcloud.cli.sessions import (  # noqa: F401
+    _parse_create_args,
+    _session_state,
+    cmd_create,
+    cmd_history,
+    cmd_sessions,
+    cmd_stop,
+)
+from fcloud.cli.setup import cmd_set_token, cmd_setup  # noqa: F401
+from fcloud.cli.ssh import (  # noqa: F401
+    _fcloud_command_prefix,
+    _validated_session_id,
+    cmd_ssh,
+    cmd_tunnel,
+)
+from fcloud.cli.sweep import (  # noqa: F401
+    _print_sweep_status,
+    cmd_map,
+    cmd_sweep,
+    cmd_sweeps,
+)
+from fcloud.cli.volume import (  # noqa: F401
+    _is_within,
+    _volume_cat,
+    _volume_create,
+    _volume_delete,
+    _volume_download,
+    _volume_files,
+    _volume_import,
+    _volume_list,
+    cmd_volume,
+)
+from fcloud.cli.wait import (  # noqa: F401
+    _emit_wait_result,
+    _parse_wait_args,
+    _wait_for_process,
+    cmd_wait,
+)
+from fcloud.cli_args import SPECS  # noqa: F401
+
+
+def __getattr__(name: str):
+    if name == "_JSON_OUTPUT":
+        raise AttributeError(
+            "fcloud.__main__._JSON_OUTPUT moved: read/patch "
+            "fcloud.cli.output._JSON_OUTPUT (the binding every command uses)")
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+if __name__ == "__main__":
+    main()