scalebox-sdk 0.1.0__py3-none-any.whl

scalebox/test/testcodeinterpreter_sync.py

@@ -0,0 +1,239 @@
+#!/usr/bin/env python3
+"""
+Simple sync CodeInterpreter test example.
+Similar to testsandbox_sync.py but for code interpreter functionality.
+"""
+
+from scalebox.code_interpreter import Context, Sandbox
+
+
+def output_handler(output):
+    """处理输出的回调函数"""
+    print(f"输出: {output.content}")
+
+
+def result_handler(result):
+    """处理结果的回调函数"""
+    print(f"结果: {result}")
+
+
+def error_handler(error):
+    """处理错误的回调函数"""
+    print(f"错误: {error.name} - {error.value}")
+
+
+# 创建代码解释器沙箱
+sandbox = Sandbox(template="code-interpreter-v1")
+
+print("=== 基础Python代码执行 ===")
+# 基础Python代码执行
+result = sandbox.run_code(
+    """
+print("Hello from CodeInterpreter!")
+x = 1 + 2
+y = x * 3
+print(f"计算结果: x={x}, y={y}")
+{"x": x, "y": y}
+""",
+    language="python",
+)
+
+print(f"执行结果: {result}")
+print(f"标准输出: {result.logs.stdout}")
+if result.error:
+    print(f"错误: {result.error}")
+
+print("\n=== 数学计算示例 ===")
+# 数学计算
+math_result = sandbox.run_code(
+    """
+import math
+import numpy as np
+
+# 计算圆的面积和周长
+radius = 5
+area = math.pi * radius ** 2
+circumference = 2 * math.pi * radius
+
+print(f"半径: {radius}")
+print(f"面积: {area:.2f}")
+print(f"周长: {circumference:.2f}")
+
+# 使用numpy
+arr = np.array([1, 2, 3, 4, 5])
+mean_val = np.mean(arr)
+print(f"数组平均值: {mean_val}")
+
+{"radius": radius, "area": area, "mean": mean_val}
+"""
+)
+
+print(f"数学计算结果: {math_result}")
+
+print("\n=== 使用回调函数 ===")
+# 使用回调函数
+callback_result = sandbox.run_code(
+    """
+print("开始执行带回调的代码")
+for i in range(3):
+    print(f"步骤 {i+1}")
+
+result_data = {"steps": 3, "status": "completed"}
+print(f"执行完成: {result_data}")
+result_data
+""",
+    on_stdout=output_handler,
+    on_result=result_handler,
+    on_error=error_handler,
+)
+
+print("\n=== 创建和使用上下文 ===")
+# 创建上下文
+context = sandbox.create_code_context(language="python", cwd="/tmp")
+print(f"创建上下文: {context.id}")
+
+# 在上下文中执行代码
+context_result1 = sandbox.run_code(
+    """
+# 在上下文中定义变量
+context_var = "Hello from context"
+numbers = [1, 2, 3, 4, 5]
+print(f"定义了变量: {context_var}")
+print(f"数组: {numbers}")
+""",
+    context=context,
+)
+
+# 在同一上下文中使用变量
+context_result2 = sandbox.run_code(
+    """
+print(f"从上下文读取: {context_var}")
+numbers.append(6)
+print(f"修改后的数组: {numbers}")
+sum(numbers)
+""",
+    context=context,
+)
+
+print(f"上下文测试结果: {context_result2}")
+
+print("\n=== 错误处理示例 ===")
+# 故意产生错误
+error_result = sandbox.run_code(
+    """
+print("这行会执行")
+x = 10 / 0  # 这里会产生除零错误
+print("这行不会执行")
+""",
+    on_error=error_handler,
+)
+
+print(f"错误处理结果: {error_result.error}")
+
+print("\n=== 数据处理示例 ===")
+# 数据处理
+data_result = sandbox.run_code(
+    """
+import pandas as pd
+
+# 创建示例数据
+data = {
+    'name': ['Alice', 'Bob', 'Charlie'],
+    'age': [25, 30, 35],
+    'city': ['New York', 'London', 'Tokyo']
+}
+
+df = pd.DataFrame(data)
+print("数据框:")
+print(df)
+
+# 简单分析
+avg_age = df['age'].mean()
+print(f"平均年龄: {avg_age}")
+
+{"total_people": len(df), "avg_age": avg_age}
+"""
+)
+
+print(f"数据处理结果: {data_result}")
+
+print("\n=== 结果格式示例 ===")
+# 展示不同格式的结果生成
+format_result = sandbox.run_code(
+    """
+import matplotlib.pyplot as plt
+import numpy as np
+import json
+import base64
+import io
+
+print("演示多种结果格式:")
+
+# 1. 文本格式
+text_result = "这是文本格式的结果示例"
+print(f"文本: {text_result}")
+
+# 2. JSON数据格式
+json_data = {
+    "test_name": "CodeInterpreter格式测试",
+    "formats": ["text", "json", "html", "image"],
+    "timestamp": "2024-09-17T10:30:00Z",
+    "results": {
+        "success": True,
+        "format_count": 4
+    }
+}
+print(f"JSON: {json.dumps(json_data, ensure_ascii=False, indent=2)}")
+
+# 3. HTML格式示例
+html_content = \"\"\"
+<div class="result-demo">
+    <h3>CodeInterpreter 结果格式演示</h3>
+    <p><strong>状态:</strong> <span style="color: green;">成功</span></p>
+    <ul>
+        <li>文本格式支持</li>
+        <li>JSON数据结构</li>
+        <li>HTML内容渲染</li>
+        <li>图像base64编码</li>
+    </ul>
+</div>
+\"\"\"
+print(f"HTML: {html_content[:100]}...")
+
+# 4. 简单图像生成
+fig, ax = plt.subplots(1, 1, figsize=(6, 4))
+x = np.linspace(0, 2*np.pi, 50)
+y = np.sin(x)
+ax.plot(x, y, 'b-', linewidth=2)
+ax.set_title('简单正弦波示例')
+ax.grid(True)
+
+# 转换为base64
+img_buffer = io.BytesIO()
+plt.savefig(img_buffer, format='png', dpi=80, bbox_inches='tight')
+img_buffer.seek(0)
+img_base64 = base64.b64encode(img_buffer.getvalue()).decode()
+img_buffer.close()
+plt.close()
+
+print(f"图像: PNG base64编码 ({len(img_base64)} 字符)")
+
+# 返回综合结果
+{
+    "text": text_result,
+    "json_data": json_data,
+    "html": html_content,
+    "png_base64": img_base64[:50] + "...",  # 只显示前50个字符
+    "summary": {
+        "formats_demonstrated": 4,
+        "total_size": len(text_result) + len(str(json_data)) + len(html_content) + len(img_base64),
+        "description": "多格式结果演示完成"
+    }
+}
+"""
+)
+
+print(f"结果格式测试: {format_result}")
+
+print("\n=== 测试完成 ===")
+print("CodeInterpreter基础功能测试完成!")

scalebox/test/testcomputeuse.py

@@ -0,0 +1,243 @@
+import time
+
+from scalebox.csx_desktop.main import Sandbox
+
+
+def desktop_automation_demo():
+    """桌面自动化功能演示"""
+
+    # 1. 创建桌面沙箱实例
+    print("正在启动桌面沙箱...")
+    desktop = Sandbox(timeout=3600, template="browser-use")
+    # print(f"沙箱已启动，ID: {desktop.sandbox_id}")
+
+    # 2. 启动VNC流以便远程查看桌面
+    print("启动VNC远程桌面...")
+    time.sleep(5)
+    desktop.stream.start()
+    vnc_url = desktop.stream.get_url(auto_connect=True)
+    print(f"VNC访问URL: {vnc_url}")
+    # print(f"VNC认证密钥: {desktop.stream.get_auth_key()}")
+
+    # 等待系统完全启动
+    time.sleep(3)
+
+    # 3. 基本鼠标操作示例
+    print("执行鼠标操作...")
+
+    # 获取屏幕尺寸
+    screen_width, screen_height = desktop.get_screen_size()
+    print(f"屏幕尺寸: {screen_width}x{screen_height}")
+
+    # 移动鼠标到屏幕中心
+    center_x, center_y = screen_width // 2, screen_height // 2
+    desktop.move_mouse(center_x, center_y)
+
+    # 获取当前光标位置
+    cursor_x, cursor_y = desktop.get_cursor_position()
+    print(f"光标位置: ({cursor_x}, {cursor_y})")
+
+    # 左键点击
+    desktop.left_click(center_x, center_y)
+
+    # 右键点击
+    desktop.right_click(center_x + 100, center_y)
+
+    # 双击
+    desktop.double_click(center_x, center_y + 100)
+
+    # 鼠标拖动示例
+    desktop.drag((center_x, center_y), (center_x + 200, center_y))
+
+    # 鼠标滚动
+    desktop.scroll("down", 2)  # 向下滚动2次
+    desktop.scroll("up", 1)  # 向上滚动1次
+
+    # 4. 键盘输入示例
+    print("执行键盘操作...")
+
+    # 打开终端
+    desktop.launch("xfce4-terminal")  # 或 xfce4-terminal
+    time.sleep(2)
+
+    # 获取当前窗口ID
+    current_window = desktop.get_current_window_id()
+    print(f"当前窗口ID: {current_window}")
+
+    # 输入命令
+    desktop.write("echo 'Hello from SBX Sandbox!'")
+    desktop.press("enter")
+
+    # 特殊按键
+    desktop.press("ctrl+shift+t")  # 新建标签页
+    time.sleep(1)
+
+    desktop.write("ls -la")
+    desktop.press("enter")
+
+    # 5. 窗口管理示例
+    print("执行窗口管理操作...")
+
+    # 获取所有可见窗口
+    terminal_windows = desktop.get_application_windows("xfce4-terminal")
+    print(f"终端窗口: {terminal_windows}")
+
+    if terminal_windows:
+        # 获取窗口标题
+        window_title = desktop.get_window_title(terminal_windows[0])
+        print(f"窗口标题: {window_title}")
+
+    # 6. 启动应用程序示例
+    print("启动其他应用程序...")
+
+    # 使用gtk-launch启动应用程序
+    try:
+        desktop.launch("firefox-esr")  # 启动Firefox浏览器
+        time.sleep(3)
+
+        # 在Firefox中输入URL
+        desktop.press("ctrl+l")  # 聚焦地址栏
+        desktop.write("https://www.google.com")
+        desktop.press("enter")
+
+    except Exception as e:
+        print(f"启动Firefox失败: {e}")
+
+    # 7. 文件操作示例 - 创建并打开文件
+    print("执行文件操作...")
+
+    # 创建测试文件
+    test_content = """Hello World!
+This is a test file created from SBX Sandbox SDK.
+Current time: {time}
+""".format(
+        time=time.ctime()
+    )
+
+    desktop.files.write("/tmp/test_file.txt", test_content)
+    print("已创建测试文件")
+
+    # 使用默认程序打开文件
+    # desktop.open("/tmp/test_file.txt")
+    # time.sleep(2)
+
+    # 8. 截图功能示例
+    print("执行截图操作...")
+
+    # 截取屏幕并保存为bytes
+    screenshot_bytes = desktop.screenshot(format="bytes")
+    print(f"截图大小: {len(screenshot_bytes)} 字节")
+
+    # 也可以获取截图流
+    # screenshot_stream = desktop.screenshot(format="stream")
+    # for chunk in screenshot_stream:
+    #     # 处理截图数据
+    #     pass
+
+    # 9. 复杂自动化脚本示例
+    print("执行复杂自动化任务...")
+
+    # 返回终端窗口
+    terminal_windows = desktop.get_application_windows("xfce4-terminal")
+    if terminal_windows:
+        # 激活终端窗口
+        desktop.commands.run(
+            f"DISPLAY={desktop._display} xdotool windowactivate {terminal_windows[0]}"
+        )
+
+        # 创建Python脚本并执行
+        python_script = """#!/usr/bin/env python3
+import time
+print("starting...")
+for i in range(5):
+    print(f"count: {i}")
+time.sleep(0.5)
+print("success!")
+"""
+
+        script_path = "/tmp/automation_script.py"
+        desktop.files.write(script_path, python_script)
+
+        desktop.write(f"python3 /uploads{script_path}")
+        desktop.press("enter")
+
+    # 10. 等待和定时操作
+    print("等待操作完成...")
+    desktop.wait(2000)  # 等待2秒
+
+    # 展示所有功能已完成
+    desktop.write("echo 'All successfully!'")
+    desktop.press("enter")
+
+    print("演示完成! 沙箱将在10分钟后自动关闭或按Ctrl+C提前结束")
+
+    # 保持沙箱运行以便观察
+    try:
+        time.sleep(600)  # 保持10分钟
+    except KeyboardInterrupt:
+        print("提前结束沙箱会话")
+
+    # 11. 清理工作会在with语句退出时自动执行
+
+
+# def specific_use_cases():
+#     """特定用例示例"""
+#
+#     # 用例1: 网页自动化测试
+#     def web_automation():
+#         with Sandbox() as desktop:
+#             desktop.open("firefox")
+#             time.sleep(3)
+#             desktop.press("ctrl+l")
+#             desktop.write("https://www.google.com")
+#             desktop.press("enter")
+#             time.sleep(2)
+#             desktop.write("E2B Sandbox Automation")
+#             desktop.press("enter")
+#             # 更多网页自动化操作...
+#
+#     # 用例2: GUI应用测试
+#     def gui_app_testing():
+#         with Sandbox() as desktop:
+#             # 启动待测试的GUI应用
+#             desktop.launch("some-gui-application")
+#             time.sleep(2)
+#
+#             # 执行一系列测试操作
+#             desktop.move_mouse(100, 100)
+#             desktop.left_click()
+#             desktop.write("测试输入")
+#             desktop.press("enter")
+#
+#             # 验证结果
+#             screenshot = desktop.screenshot(format="bytes")
+#             # 这里可以添加截图分析逻辑
+#
+#     # 用例3: 教育演示
+#     def educational_demo():
+#         with Sandbox() as desktop:
+#             # 打开编程环境
+#             desktop.open("vscode")  # 或其他IDE
+#
+#             # 逐步演示代码编写和执行过程
+#             time.sleep(2)
+#             desktop.write("# Python编程演示")
+#             desktop.press("enter")
+#             desktop.write("print('Hello, Students!')")
+#             desktop.press("enter")
+#
+#             # 保存并运行
+#             desktop.press("ctrl+s")
+#             desktop.press("ctrl+alt+t")  # 打开终端
+#             desktop.write("python3 demo.py")
+#             desktop.press("enter")
+
+
+if __name__ == "__main__":
+    try:
+        desktop_automation_demo()
+    except Exception as e:
+        print(f"演示过程中发生错误: {e}")
+        import traceback
+
+        traceback.print_exc()

scalebox/test/testnovnc.py

@@ -0,0 +1,12 @@
+from email.mime import application
+
+from csx_desktop.main import Sandbox
+
+sandbox = Sandbox()
+sandbox.stream.start()
+
+print(sandbox.get_screen_size())
+sandbox.launch("firefox-esr")
+windpws_id = sandbox.get_current_window_id()
+print(windpws_id)
+print(sandbox.get_window_title(window_id=windpws_id))

scalebox/test/testsandbox_async.py

@@ -0,0 +1,118 @@
+import asyncio
+
+from sandbox_async.main import AsyncSandbox
+
+from scalebox.sandbox.commands.command_handle import PtySize
+
+# from scalebox.sandbox_async.main import AsyncSandbox
+
+
+async def pty_output_handler(output):
+    """处理 PTY 输出的回调函数"""
+    print(f"PTY 输出: {output}")
+
+
+async def main():
+    sandbox = AsyncSandbox()
+    proc = await sandbox.commands.run("echo hello from async")
+    print("exit_code =", proc.exit_code)
+    print("stdout   =", proc.stdout)
+    # 4. 创建 PTY
+    pty_size = PtySize(rows=24, cols=80)
+    pty_handle = await sandbox.pty.create(
+        size=pty_size,
+        on_data=pty_output_handler,
+        cwd="/root",
+        envs={"CUSTOM_ENV": "value"},
+        timeout=120,
+        request_timeout=30,
+    )
+    print(f"PTY 已创建，PID: {pty_handle.pid}")
+
+    # 5. 向 PTY 发送输入
+    await sandbox.pty.send_stdin(
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY'\n", request_timeout=10
+    )
+    await asyncio.sleep(2)
+
+    # 6. 调整 PTY 大小
+    new_size = PtySize(rows=30, cols=100)
+    await sandbox.pty.resize(pid=pty_handle.pid, size=new_size, request_timeout=10)
+    print("PTY 大小已调整")
+    killed = await sandbox.pty.kill(pty_handle.pid)
+    print(f"PTY 是否被杀死: {killed}")
+
+
+# 静态方法使用示例
+async def static_methods_example():
+    # 1. 创建沙箱
+    sandbox = await AsyncSandbox.create()
+    sandbox_id = sandbox.sandbox_id
+
+    try:
+        # 2. 使用静态方法获取沙箱信息
+        info = await AsyncSandbox.get_info(sandbox_id)
+        print(f"静态方法获取的沙箱信息: {info}")
+
+        # 3. 使用静态方法设置超时
+        await AsyncSandbox.set_timeout(sandbox_id, 1800)  # 30分钟
+
+        # 4. 使用静态方法获取指标
+        try:
+            metrics = await AsyncSandbox.get_metrics(sandbox_id)
+            print(f"静态方法获取的指标: {metrics}")
+        except Exception as e:
+            print(f"静态方法获取指标失败: {e}")
+
+    finally:
+        # 5. 使用静态方法关闭沙箱
+        killed = await AsyncSandbox.kill(sandbox_id)
+        print(f"静态方法关闭沙箱结果: {killed}")
+
+
+# 连接现有沙箱示例
+async def connect_example():
+    # 1. 创建沙箱并获取ID
+    sandbox = await AsyncSandbox.create()
+    sandbox_id = sandbox.sandbox_id
+    await sandbox.kill()  # 先关闭
+
+    # 2. 连接到现有沙箱
+    try:
+        connected_sandbox = await AsyncSandbox.connect(sandbox_id)
+        print(f"成功连接到沙箱: {connected_sandbox.sandbox_id}")
+
+        # 使用连接的沙箱
+        is_running = await connected_sandbox.is_running()
+        print(f"连接后沙箱状态: {is_running}")
+
+        await connected_sandbox.kill()
+    except Exception as e:
+        print(f"连接沙箱失败: {e}")
+
+
+# 上下文管理器使用示例
+async def context_manager_example():
+    sandbox = AsyncSandbox()
+    # async with AsyncSandbox() as sandbox:
+    #     # 在上下文中使用沙箱
+    #     is_running = await sandbox.is_running()
+    #     print(f"上下文管理器中的沙箱状态: {is_running}")
+
+    # 创建和使用 PTY
+    pty_handle = await sandbox.pty.create(
+        size=PtySize(rows=24, cols=80), on_data=pty_output_handler
+    )
+
+    # 发送命令
+    await sandbox.pty.send_stdin(pty_handle.pid, b"ls -la\n")
+
+    # 等待一段时间
+    await asyncio.sleep(10)
+
+
+if __name__ == "__main__":
+    # asyncio.run(main())0
+    # asyncio.run(static_methods_example())
+    # asyncio.run(connect_example())
+    asyncio.run(context_manager_example())

scalebox/test/testsandbox_sync.py

@@ -0,0 +1,38 @@
+from scalebox.sandbox.commands.command_handle import PtySize
+from scalebox.sandbox_sync.main import Sandbox
+
+
+def output_handler(output):
+    """处理 输出的回调函数"""
+    print(f"PTY 输出: {output}")
+
+
+sandbox = Sandbox(api_key=f"sk-Wk4IgtUYOqnttxGaxZmELEV4p2FXh15Evt0FIcSa")
+# print(sandbox.files.list("/root",2))
+# proc = sandbox.commands._start(
+#         cmd="python3 -c \"import math; print('π =', math.pi); exit(42)\"",
+#     ).wait(on_pty=None,
+#            on_stdout=lambda data: print("[STDOUT]", data, end=""),
+#            on_stderr=lambda data: print("[STDERR]", data, end=""),)
+
+# proc = sandbox.commands._start(
+#         cmd="ls /",
+#     ).wait(on_pty=None,
+#            on_stdout=lambda data: print("[STDOUT]", data, end=""),
+#            on_stderr=lambda data: print("[STDERR]", data, end=""),)
+#
+# print("exit_code =", proc.exit_code)   # 42
+# print("full_output =", proc.stdout)    # π = 3.141592653589793
+# pty=sandbox.pty.create(size=PtySize(1024, 768),user="root",cwd="/root/",)
+# sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"hello\"")
+# sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\"")
+# result=pty.wait(
+#     on_pty=lambda data: print("[STDOUT]", data, end=""),
+# )
+# print("exit_code =", result.exit_code)
+# print("full_output =", result.stdout)
+result = sandbox.commands.run(
+    cmd="ls /", on_stdout=output_handler, on_stderr=output_handler
+)
+print(result.exit_code)
+print(result.error)