use-computer 0.0.37__tar.gz → 0.0.39__tar.gz

{use_computer-0.0.37 → use_computer-0.0.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: use-computer
-Version: 0.0.37
+Version: 0.0.39
 Summary: Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu
 Project-URL: Homepage, https://use.computer
 Project-URL: Documentation, https://api.use.computer/docs

{use_computer-0.0.37 → use_computer-0.0.39}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "use-computer"
-version = "0.0.37"
+version = "0.0.39"
 description = "Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu"
 readme = "README.md"
 requires-python = ">=3.10"

{use_computer-0.0.37 → use_computer-0.0.39}/use_computer/harbor/environment/platforms/__init__.py

@@ -127,7 +127,12 @@ class WindowsAdapter(PlatformAdapter):
     is_mounted = True
 
     async def create(self, env: Any) -> AsyncSandbox:
-        return await env._client.create(type="windows", host=env._host, version=env._version)
+        return await env._client.create(
+            type="windows",
+            host=env._host,
+            version="" if env._snapshot else env._version,
+            snapshot=env._snapshot,
+        )
 
     async def setup(self, env: Any) -> None:
         await asyncio.sleep(5.0)
@@ -157,7 +162,12 @@ class UbuntuAdapter(PlatformAdapter):
     is_mounted = True
 
     async def create(self, env: Any) -> AsyncSandbox:
-        return await env._client.create(type="ubuntu", host=env._host, version=env._version)
+        return await env._client.create(
+            type="ubuntu",
+            host=env._host,
+            version="" if env._snapshot else env._version,
+            snapshot=env._snapshot,
+        )
 
     async def setup(self, env: Any) -> None:
         await asyncio.sleep(5.0)

{use_computer-0.0.37 → use_computer-0.0.39}/use_computer/harbor/environment/platforms/ubuntu.py

@@ -109,6 +109,8 @@ async def exec_local(
 
 
 async def run_setup(env: Any) -> None:
+    await wake_desktop(env)
+
     setup_dir = env._task_dir / "tests" / "setup"
 
     for setup_file in read_setup_files(setup_dir):
@@ -120,6 +122,8 @@ async def run_setup(env: Any) -> None:
         if not local.exists():
             env.logger.warning(f"setup file missing on disk: {local}")
             continue
+        parent = sh_quote(str(Path(vm_path).parent))
+        await env.ubuntu_sandbox.shell.run(f"mkdir -p {parent}", shell="bash", timeout=60)
         env.logger.info(f"uploading {local.name} -> {vm_path} ({local.stat().st_size}B)")
         await env.ubuntu_sandbox.upload(str(local), vm_path)
 
@@ -137,6 +141,28 @@ async def run_setup(env: Any) -> None:
     env.logger.info("task setup complete")
 
 
+async def wake_desktop(env: Any) -> None:
+    script = """
+echo 'user:lume' | sudo chpasswd || true
+export DISPLAY=:0
+export XAUTHORITY=/home/user/.Xauthority
+gsettings set org.gnome.desktop.screensaver lock-enabled false || true
+gsettings set org.gnome.desktop.session idle-delay 0 || true
+xset s off -dpms || true
+xset dpms force on || true
+xdotool key Escape || true
+sleep 1
+xdotool key ctrl+a || true
+xdotool type lume || true
+xdotool key Return || true
+sleep 3
+xset dpms force on || true
+"""
+    result = await env.ubuntu_sandbox.shell.run(script, shell="bash", timeout=60)
+    if result.return_code != 0:
+        env.logger.warning(f"ubuntu desktop wake rc={result.return_code}: {result.stderr or ''}")
+
+
 async def upload_file(env: Any, source: Path | str, target: str) -> None:
     await env.ubuntu_sandbox.upload(str(source), target)
 

{use_computer-0.0.37 → use_computer-0.0.39}/use_computer/harbor/environment/use_computer.py

@@ -54,6 +54,7 @@ class UseComputerEnvironment(BaseEnvironment):
         host: str = "",
         platform: str = "macos",
         version: str = "",
+        snapshot: str = "",
         device_type: str = "",
         runtime: str = "",
         **kwargs,
@@ -64,6 +65,7 @@ class UseComputerEnvironment(BaseEnvironment):
         self._platform = platform
         self._platform_adapter = adapter_for(platform)
         self._version = version or os.environ.get("USE_COMPUTER_VERSION", "")
+        self._snapshot = snapshot or os.environ.get("USE_COMPUTER_SNAPSHOT", "")
         # YAML-level iOS pin (job-adhoc.yaml `environment.kwargs.device_type` /
         # `.runtime`). Used as a fallback when the task.toml doesn't pin one.
         # Lets the same dataset target different sims by editing one YAML line