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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: use-computer
-Version: 0.0.38
+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.38 → use_computer-0.0.39}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "use-computer"
-version = "0.0.38"
+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.38 → 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):
@@ -139,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)