Process Management

from beam import Sandbox

sb = Sandbox(memory="1Gi").create()
try:
    result = sb.process.run_code("print(6 * 7)")
    print(result.stdout)
    print(result.stderr)
    print(result.exit_code)
finally:
    sb.terminate()

process = sb.process.run_code("import time; time.sleep(10)", blocking=False)
print(process.pid)
exit_code = process.wait(timeout=30)

process = sb.process.exec("ls", "-la", "/workspace")
process.wait()
print(process.stdout.read())

process = sb.process.exec(
    "sh", "-c", 'printf "%s\n" "$GREETING" | tr a-z A-Z',
    env={"GREETING": "hello"},
    cwd="/workspace",
)
process.wait()
print(process.stdout.read())

process = sb.process.exec("python3", "-c", "import sys; print(sys.stdin.read().upper())", stdin="hello")
process.wait()
print(process.stdout.read())

process = sb.process.exec("python3", "-u", "-c", "print('ready')")
for line in process.logs:
    print(line, end="")
print(process.wait())

for pid, process in sb.process.list_processes().items():
    print(pid, process.status())

process = sb.process.get_process(pid)

import asyncio
from beam import Sandbox

async def main():
    sb = await asyncio.to_thread(Sandbox(memory="1Gi").create)
    try:
        result = await sb.aio.process.run_code("print('async hello')")
        print(result.stdout)
        process = await sb.aio.process.exec("python3", "-u", "-c", "print('streaming')")
        async for line in process.logs:
            print(line, end="")
        await process.wait(timeout=30)
    finally:
        await sb.aio.terminate()

asyncio.run(main())