Sessions

from sandbox_auth import AUTH
import cwsandbox
from cwsandbox import SandboxDefaults

# 共有設定を定義する
defaults = SandboxDefaults(
    container_image="python:3.11",
    tags=("project-alpha",),
    resources={"cpu": "1", "memory": "1Gi"},
)

# コンテキストマネージャーでセッションを作成する
with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    # セッションを通じてサンドボックスを作成する
    sb1 = session.sandbox()
    sb2 = session.sandbox()

    # サンドボックスを使用する
    result1 = sb1.exec(["echo", "sandbox 1"]).result()
    result2 = sb2.exec(["echo", "sandbox 2"]).result()

# コンテキストを終了するとすべてのサンドボックスが自動的に停止される

from sandbox_auth import AUTH
with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    # sandbox() は未開始のサンドボックスを返します
    sandbox = session.sandbox()

    # オペレーションは初回使用時にサンドボックスを自動開始します
    result = sandbox.exec(["echo", "hello"]).result()

from sandbox_auth import AUTH
with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    sandbox = session.sandbox()
    sandbox.start().result()  # 明示的な開始
    sandbox.wait()            # 起動または完了を待機
    if sandbox.status != cwsandbox.SandboxStatus.RUNNING:
        raise RuntimeError(f"Sandbox is not running: {sandbox.status}")
    result = sandbox.exec(["echo", "hello"]).result()

from sandbox_auth import AUTH
import cwsandbox
from cwsandbox import ResourceOptions, SandboxDefaults

# 基本設定を使用する Session
defaults = SandboxDefaults(
    container_image="python:3.11",
    tags=("my-project",),
    resources={"cpu": "1", "memory": "1Gi"},
)

with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    # このサンドボックスのイメージとリソースを上書きします
    gpu_sandbox = session.sandbox(
        command="sleep",
        args=["infinity"],
        container_image="pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime",
        resources=ResourceOptions(
            requests={"cpu": "4", "memory": "16Gi"},
            limits={"cpu": "4", "memory": "16Gi"},
            gpu={"count": 1},
        ),
        tags=["gpu-workload"],  # セッションのタグと統合されます
    )

from sandbox_auth import AUTH
with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    # サンドボックスのプールを作成します
    sandboxes = [
        session.sandbox()
        for _ in range(5)
    ]

    # プール内で処理を分散します
    processes = [
        sb.exec(["python", "-c", f"print({i})"])
        for i, sb in enumerate(sandboxes)
    ]

    # 結果を収集します
    results = [p.result() for p in processes]

# 5 つのサンドボックスはすべて自動的にクリーンアップされます

from sandbox_auth import AUTH
session = cwsandbox.Session(defaults=defaults, auth=AUTH)

sandbox = session.sandbox()
result = sandbox.exec(["echo", "hello"]).result()

# セッションを閉じます（すべてのサンドボックスを停止します）
session.close().result()

from sandbox_auth import AUTH
with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    sandbox = session.sandbox()
    raise RuntimeError("Something went wrong!")
# サンドボックスはクリーンアップされます

from sandbox_auth import AUTH
import cwsandbox
from cwsandbox import Sandbox

with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    # 既存のサンドボックスを検索する
    existing = Sandbox.list(tags=["orphaned-work"], auth=AUTH).result()

    # クリーンアップのために取り込みます
    for sandbox in existing:
        session.adopt(sandbox)

    # これで session.close() はこれらもクリーンアップします

from sandbox_auth import AUTH
session = cwsandbox.Session(defaults=defaults, auth=AUTH)

# 追跡しているサンドボックスの数
print(session.sandbox_count)  # 0

sandbox = session.sandbox()
print(session.sandbox_count)  # 1

from sandbox_auth import AUTH
from cwsandbox import Sandbox

# セッションを使わずに直接利用します
with Sandbox.run(resources={"cpu": "1", "memory": "1Gi"}, auth=AUTH) as sandbox:
    result = sandbox.exec(["echo", "hello"]).result()
# コンテキストマネージャーがクリーンアップを処理します