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

# 기본 설정을 사용하는 세션
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()
# 컨텍스트 관리자가 정리 작업 처리