droidrun 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

{droidrun-0.3.2.dist-info → droidrun-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: droidrun
-Version: 0.3.2
+Version: 0.3.3
 Summary: A framework for controlling Android devices through LLM agents
 Project-URL: Homepage, https://github.com/droidrun/droidrun
 Project-URL: Bug Tracker, https://github.com/droidrun/droidrun/issues
@@ -25,9 +25,11 @@ Classifier: Topic :: Software Development :: Testing
 Classifier: Topic :: Software Development :: Testing :: Acceptance
 Classifier: Topic :: System :: Emulators
 Classifier: Topic :: Utilities
-Requires-Python: >=3.10
+Requires-Python: >=3.11
+Requires-Dist: adbutils==2.10.0
 Requires-Dist: aiofiles>=23.0.0
 Requires-Dist: anthropic>=0.7.0
+Requires-Dist: apkutils==2.0.0
 Requires-Dist: arize-phoenix
 Requires-Dist: click>=8.1.0
 Requires-Dist: llama-index
@@ -44,6 +46,7 @@ Requires-Dist: posthog==6.0.2
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: rich>=13.0.0
+Requires-Dist: typing-extensions
 Provides-Extra: dev
 Requires-Dist: bandit>=1.7.0; extra == 'dev'
 Requires-Dist: black>=23.0.0; extra == 'dev'
@@ -64,6 +67,12 @@ Description-Content-Type: text/markdown
 [![Benchmark](https://img.shields.io/badge/Benchmark-🏅-teal)](https://droidrun.ai/benchmark)
 [![Twitter Follow](https://img.shields.io/twitter/follow/droid_run?style=social)](https://x.com/droid_run)
 
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="https://api.producthunt.com/widgets/embed-image/v1/top-post-badge.svg?post_id=983810&theme=dark&period=daily&t=1753948032207">
+  <source media="(prefers-color-scheme: light)" srcset="https://api.producthunt.com/widgets/embed-image/v1/top-post-badge.svg?post_id=983810&theme=neutral&period=daily&t=1753948125523">
+  <a href="https://www.producthunt.com/products/droidrun-framework-for-mobile-agent?embed=true&utm_source=badge-top-post-badge&utm_medium=badge&utm_source=badge-droidrun" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/top-post-badge.svg?post_id=983810&theme=neutral&period=daily&t=1753948125523" alt="Droidrun - Give&#0032;AI&#0032;native&#0032;control&#0032;of&#0032;physical&#0032;&#0038;&#0032;virtual&#0032;phones&#0046; | Product Hunt" style="width: 200px; height: 54px;" width="200" height="54" /></a>
+</picture>
+
 
 
 DroidRun is a powerful framework for controlling Android and iOS devices through LLM agents. It allows you to automate device interactions using natural language commands. [Checkout our benchmark results](https://droidrun.ai/benchmark)
@@ -91,17 +100,21 @@ Read on how to get droidrun up and running within seconds in [our docs](https://
 
 ## 🎬 Demo Videos
 
-1. **Group Chat Summarization**: Let DroidRun summarize an escalated group chat for you.   
+1. **Accommodation booking**: Let Droidrun search for an apartment for you
+
+   [![Droidrun Accommodation Booking Demo](https://img.youtube.com/vi/VUpCyq1PSXw/0.jpg)](https://youtu.be/VUpCyq1PSXw)
+
+<br>
 
-   [![Group Chat Summarizer](https://img.youtube.com/vi/ofEnSUHHxX8/0.jpg)](https://www.youtube.com/watch?v=ofEnSUHHxX8)
+2. **Trend Hunter**: Let Droidrun hunt down trending posts
 
-2. **Travel Search Assistant**: Wittness DroidRun looking for the cheapest stay and share it with a colleague on telegram.    
+   [![Droidrun Trend Hunter Demo](https://img.youtube.com/vi/7V8S2f8PnkQ/0.jpg)](https://youtu.be/7V8S2f8PnkQ)
 
-   [![Travel Search Assistant](https://img.youtube.com/vi/QgtRaLS3NBM/0.jpg)](https://www.youtube.com/watch?v=QgtRaLS3NBM)
+<br>
 
-3. **Automate TikTok Shopping**: See how DroidRun looks for a stanley cup on TikTok Shop and send the product details via email.    
+3. **Streak Saver**: Let Droidrun save your streak on your favorite language learning app
 
-    [![TikTok Shopping Assistant](https://img.youtube.com/vi/ol3bivBAmn4/0.jpg)](https://www.youtube.com/watch?v=ol3bivBAmn4)
+   [![Droidrun Streak Saver Demo](https://img.youtube.com/vi/B5q2B467HKw/0.jpg)](https://youtu.be/B5q2B467HKw)
 
 
 ## 💡 Example Use Cases

{droidrun-0.3.2.dist-info → droidrun-0.3.3.dist-info}/RECORD

@@ -1,53 +1,54 @@
-droidrun/__init__.py,sha256=OqgWoczsC4gpt40pV-wuw7x5DQmmbKtNSxQ8QqPEY1E,510
+droidrun/__init__.py,sha256=Cqt4NXZ-753220dlRZXglMkFT2ygcXSErMmqupGsi9A,622
 droidrun/__main__.py,sha256=78o1Wr_Z-NrZy9yLWmEfNfRRhAiJGBr4Xi3lmbgkx3w,105
-droidrun/portal.py,sha256=4UEoTdAg6TWgxCXEhyAXMARr8Hsq5wLDx_4zim9-eaA,4061
-droidrun/adb/__init__.py,sha256=kh-iT9Sv6RZ2dFSDu1beK_GgtAq8wlMvZQ7puR8JWsI,257
-droidrun/adb/device.py,sha256=8Qc4pU2-dy8v7skBKsn74Lnke1d7x339cQCrba28vlU,11694
-droidrun/adb/manager.py,sha256=mGdQSlHmZ00t-4jmg_I_qRrBigWnWr_gQXB1TcFXqoQ,2767
-droidrun/adb/wrapper.py,sha256=Yz3_JSIidq-5bf-t0UfawTutMaLrINS_1Y15m_Uss4g,7093
+droidrun/portal.py,sha256=C0gZK9GHigsM6Vq2fQP_DKaDXqtnQSA2NUrxClVe6hw,3931
 droidrun/agent/__init__.py,sha256=4SqTJeGDvr_wT8rtN9J8hnN6P-pae663mkYr-JmzH4w,208
 droidrun/agent/codeact/__init__.py,sha256=ZLDGT_lTTzyNm7pcBzdyRIGHJ2ZgbInJdhXZRbJLhSQ,278
-droidrun/agent/codeact/codeact_agent.py,sha256=SoJPFeJUfNjs875rsbRSVrSjMIwIpWq_vYcct83PQhU,16075
+droidrun/agent/codeact/codeact_agent.py,sha256=XYyRrDAj2MoNb4WTtCzY8bqlI1WDy9TUIRku4NMUezk,16087
 droidrun/agent/codeact/events.py,sha256=skCfZ-5SR0YhhzZVxx8_VkUjfILk8rCv47k9pHNYhdc,634
 droidrun/agent/codeact/prompts.py,sha256=28HflWMNkC1ky0hGCzAxhJftjU2IIU1ZRUfya3S7M6I,1006
 droidrun/agent/common/default.py,sha256=P07el-PrHsoqQMYsYxSSln6mFl-QY75vzwp1Dll_XmY,259
-droidrun/agent/common/events.py,sha256=axauKdrXFEeB1hpd0geazwBtJySnHsvLZV0u9F9OWZI,96
+droidrun/agent/common/events.py,sha256=CUYdglSchK7W39VTbDdw24WJCwF4HfU0jKMbnjnFBdc,1085
 droidrun/agent/context/__init__.py,sha256=upSJilVQUwQYWJuGr_8QrmJaReV3SNAc_Z61wQZzbK4,697
 droidrun/agent/context/agent_persona.py,sha256=Mxd4HTyirWD-aqNlka1hQBdS-0I-lXJr2AjPMwDMUo8,390
 droidrun/agent/context/context_injection_manager.py,sha256=sA33q2KPtX_4Yap8wM11T6ewlZC_0FIbKPEc400SHrE,2188
 droidrun/agent/context/episodic_memory.py,sha256=1ImeR3jAWOpKwkQt3bMlXVOBiQbIli5fBIlBq2waREQ,394
 droidrun/agent/context/reflection.py,sha256=0hJluOz0hTlHHhReKpIJ9HU5aJbaJsvrjMfraQ84D-M,652
 droidrun/agent/context/task_manager.py,sha256=ESLs4kR6VNYiYQsc4V7WAeoSLwbaZPSWBXpveOfOv8c,4343
-droidrun/agent/context/personas/__init__.py,sha256=Kjze5UDSSjV2EZShXsIM75O1IPprSYop0y0MI4bbPUY,182
+droidrun/agent/context/personas/__init__.py,sha256=oSRa8g_xngX7JPIRPu7fLO33m3r7fdEQzIuORuqcw5M,232
 droidrun/agent/context/personas/app_starter.py,sha256=dHeknznxGEPJ7S6VPyEG_MB-HvAvQwUOnRWaShaV8Xo,1585
+droidrun/agent/context/personas/big_agent.py,sha256=Gl_y4ykz3apGc203-KG2UbSOwf7gDUiWh7GOVyiLn-Y,5091
 droidrun/agent/context/personas/default.py,sha256=Xm07YCWoKjvlHAbQRtzE3vn7BVcz6wYcSVeg4FiojJQ,5060
-droidrun/agent/context/personas/ui_expert.py,sha256=P2dAnkKiR3O4bN4ZUuWHmuu4Qo8WRiH1mU6EEOr3yJA,4710
+droidrun/agent/context/personas/ui_expert.py,sha256=j0OKfN1jQSrREHcVeomMTDPCWLsZVX4aeuWN4Y-x3z0,4739
 droidrun/agent/droid/__init__.py,sha256=3BfUVZiUQ8ATAJ_JmqQZQx53WoERRpQ4AyHW5WOgbRI,297
-droidrun/agent/droid/droid_agent.py,sha256=SEJbFoLDwFqE_nXNke-D4e0lx9DwCY-I-ryywuW6-Y8,14955
+droidrun/agent/droid/droid_agent.py,sha256=_XtGkkQM27qfxDO8V2MwehWz931Z10CBhtOIoGNsWBk,15252
 droidrun/agent/droid/events.py,sha256=Ks2D6lX5P1rpZ4nIAPXSC83z4AT5OzKt3isP5yk25F4,689
 droidrun/agent/oneflows/reflector.py,sha256=I_tE0PBjvwWbS6SA8Qd41etxJglFgn8oScuKUxc9LEE,11621
 droidrun/agent/planner/__init__.py,sha256=Fu0Ewtd-dIRLgHIL1DB_9EEKvQS_f1vjB8jgO5TbJXg,364
 droidrun/agent/planner/events.py,sha256=oyt2FNrA2uVyUeVT65-N0AC6sWBFxSnwNEqWtnRYoFM,390
-droidrun/agent/planner/planner_agent.py,sha256=vcZx8tawMss0F2nt-4QEtdF7kSaikUq3rdWPdTNicjk,10494
+droidrun/agent/planner/planner_agent.py,sha256=fxLeLeN_2kmaWHXLaDLvXOR_uvP_ult_I8SPZbgL_AU,10482
 droidrun/agent/planner/prompts.py,sha256=Ci7Oeu3J4TAhx-tKGPZ9l6Wb3a81FSqC8cWW4jW73HI,6046
 droidrun/agent/utils/__init__.py,sha256=JK6ygRjw7gzcQSG0HBEYLoVGH54QQAxJJ7HpIS5mgyc,44
 droidrun/agent/utils/async_utils.py,sha256=IQBcWPwevm89B7R_UdMXk0unWeNCBA232b5kQGqoxNI,336
 droidrun/agent/utils/chat_utils.py,sha256=5oqP2nmKs8sHWP1H_TK82yaxrxWf7FdEbFKASKpR60g,13000
-droidrun/agent/utils/executer.py,sha256=lQbk2TbPyl_F0k2FqEVimq8cQRcWM_KCO_I7fnkkxqA,4587
+droidrun/agent/utils/executer.py,sha256=UKVd7asr3RvO-YgRflGdG12FFZxDvU6hFxrdV1sWLcU,4892
 droidrun/agent/utils/llm_picker.py,sha256=16tNkNhEM9gD_uivzxLvuaa6s0Tz7Igu-3fxMP2lAtY,5968
-droidrun/agent/utils/trajectory.py,sha256=OmO8TvNO0LVtPXg2qTCv8o9ePaMeDyf-MRWN_YObXho,6845
+droidrun/agent/utils/trajectory.py,sha256=PU3nI3Zru580_bK0TvqUaf-5kiWvj6hFoedXkDgTqdc,17047
 droidrun/cli/__init__.py,sha256=DuwSRtZ8WILPd-nf-fZ7BaBsRgtofoInOF3JtJ9wag0,167
 droidrun/cli/logs.py,sha256=PsT_VbnOa_sOLXK4KkEJk4AsYCpscqrVoryMmLVwPG0,9714
-droidrun/cli/main.py,sha256=kGYub0eJt-Ly-fczFPwHs5vq7OC_iP9ZjLZRo1umKcc,16730
+droidrun/cli/main.py,sha256=UnTocHNd8za9Fj-ppnvJSD8DT3RQ1aqVPPTvAsWRX5g,17734
+droidrun/macro/__init__.py,sha256=333sMt19mA8sfQa4qPKbbCmr8Ej3nvpdxGXUZtVTEqM,344
+droidrun/macro/__main__.py,sha256=-zj42Bj7309oLPZbNsxZeNwIDaEe7Th1I4zF8yAHasw,193
+droidrun/macro/cli.py,sha256=fecato896z9OxZsAzCX7FWgF2TXwM5EnR85EcSGzc4U,9057
+droidrun/macro/replay.py,sha256=q_3ZcHVjvsdDfS2xyt_vuuwXGt9_1t38JD1cPsjzIfU,10764
 droidrun/telemetry/__init__.py,sha256=D4Mp02iGJH2Tjpv42Bzyo6_WC3NWj9Qy9hQPWFaCkhA,234
 droidrun/telemetry/events.py,sha256=S6r6_c2bGTZt6F88m_vREDD_MDhw3Pz4I53lLmsI764,518
-droidrun/telemetry/tracker.py,sha256=lFVQfOVLuiGulILqVo-nBxJ_u7iC2TDVXwOK4TtHEZA,2611
+droidrun/telemetry/tracker.py,sha256=Ljue6zivX8KnadXI9DivrayuWxAnUwbJKvCvNtY1Y4Y,2717
 droidrun/tools/__init__.py,sha256=9ReauavtSKDQG9ya9_Fr9O0TQnDFixgOPaP5n82_iEk,271
-droidrun/tools/adb.py,sha256=8ElV7pNfm-c5FPDoZ7qjzkqtHSjnHzQ_iyEg2e9lhZU,23819
-droidrun/tools/ios.py,sha256=DrmL_4xtQ5IgJn0vdPKTCJBFxxkz3eNNoo74Ha8ZZVs,23053
-droidrun/tools/tools.py,sha256=fgl4B62UA5KmxPnAXI1aZY5sncmKl0N97zKrjTdL510,3215
-droidrun-0.3.2.dist-info/METADATA,sha256=frA--iht9pvq_bTlAnEjWJ_gSbSd4yi42xW6gpF03gs,5838
-droidrun-0.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-droidrun-0.3.2.dist-info/entry_points.txt,sha256=o259U66js8TIybQ7zs814Oe_LQ_GpZsp6a9Cr-xm5zE,51
-droidrun-0.3.2.dist-info/licenses/LICENSE,sha256=s-uxn9qChu-kFdRXUp6v_0HhsaJ_5OANmfNOFVm2zdk,1069
-droidrun-0.3.2.dist-info/RECORD,,
+droidrun/tools/adb.py,sha256=XRa9TnuS_0-vqMJvn6RDjCRzu-MvSbdtBVAuhVHdSzo,42083
+droidrun/tools/ios.py,sha256=imzojiS6gqz4IKexUEz1ga7-flSOaC5QRpHIJTwcgSQ,21807
+droidrun/tools/tools.py,sha256=eXCaFjb_FNJFPrIfKi9W1WghxfrfnsbfutW3f0XlTzw,3720
+droidrun-0.3.3.dist-info/METADATA,sha256=9aqH7cPVJ24Snsta0Fkyv8tG9b7dMLZ1H3ZhjcvEScM,6718
+droidrun-0.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+droidrun-0.3.3.dist-info/entry_points.txt,sha256=o259U66js8TIybQ7zs814Oe_LQ_GpZsp6a9Cr-xm5zE,51
+droidrun-0.3.3.dist-info/licenses/LICENSE,sha256=s-uxn9qChu-kFdRXUp6v_0HhsaJ_5OANmfNOFVm2zdk,1069
+droidrun-0.3.3.dist-info/RECORD,,

droidrun/adb/__init__.py

@@ -1,13 +0,0 @@
-"""
-ADB Package - Android Debug Bridge functionality.
-"""
-
-from droidrun.adb.device import Device
-from droidrun.adb.manager import DeviceManager
-from droidrun.adb.wrapper import ADBWrapper
-
-__all__ = [
-    'Device',
-    'DeviceManager',
-    'ADBWrapper',
-] 

droidrun/adb/device.py

@@ -1,345 +0,0 @@
-"""
-Device - High-level representation of an Android device.
-"""
-
-import os
-import tempfile
-import time
-import random
-import string
-from typing import Dict, Optional, Tuple, List
-from droidrun.adb.wrapper import ADBWrapper
-
-
-class Device:
-    """High-level representation of an Android device."""
-
-    def __init__(self, serial: str, adb: ADBWrapper):
-        """Initialize device.
-
-        Args:
-            serial: Device serial number
-            adb: ADB wrapper instance
-        """
-        self._serial = serial
-        self._adb = adb
-        self._properties_cache: Dict[str, str] = {}
-
-    @property
-    def serial(self) -> str:
-        """Get device serial number."""
-        return self._serial
-
-    async def get_properties(self) -> Dict[str, str]:
-        """Get all device properties."""
-        if not self._properties_cache:
-            self._properties_cache = await self._adb.get_properties(self._serial)
-        return self._properties_cache
-
-    async def get_property(self, name: str) -> str:
-        """Get a specific device property."""
-        props = await self.get_properties()
-        return props.get(name, "")
-
-    @property
-    async def model(self) -> str:
-        """Get device model."""
-        return await self.get_property("ro.product.model")
-
-    @property
-    async def brand(self) -> str:
-        """Get device brand."""
-        return await self.get_property("ro.product.brand")
-
-    @property
-    async def android_version(self) -> str:
-        """Get Android version."""
-        return await self.get_property("ro.build.version.release")
-
-    @property
-    async def sdk_level(self) -> str:
-        """Get SDK level."""
-        return await self.get_property("ro.build.version.sdk")
-
-    async def shell(self, command: str, timeout: float | None = None) -> str:
-        """Execute a shell command on the device."""
-        return await self._adb.shell(self._serial, command, timeout)
-
-    async def tap(self, x: int, y: int) -> None:
-        """Tap at coordinates.
-
-        Args:
-            x: X coordinate
-            y: Y coordinate
-        """
-        await self._adb.shell(self._serial, f"input tap {x} {y}")
-
-    async def swipe(
-        self, start_x: int, start_y: int, end_x: int, end_y: int, duration_ms: int = 300
-    ) -> None:
-        """Perform swipe gesture.
-
-        Args:
-            start_x: Starting X coordinate
-            start_y: Starting Y coordinate
-            end_x: Ending X coordinate
-            end_y: Ending Y coordinate
-            duration_ms: Swipe duration in milliseconds
-        """
-        await self._adb.shell(
-            self._serial,
-            f"input swipe {start_x} {start_y} {end_x} {end_y} {duration_ms}",
-        )
-
-    async def input_text(self, text: str) -> None:
-        """Input text.
-
-        Args:
-            text: Text to input
-        """
-        await self._adb.shell(self._serial, f"input text {text}")
-
-    async def press_key(self, keycode: int) -> None:
-        """Press a key.
-
-        Args:
-            keycode: Android keycode to press
-        """
-        await self._adb.shell(self._serial, f"input keyevent {keycode}")
-
-    async def start_activity(
-        self,
-        package: str,
-        activity: str = ".MainActivity",
-        extras: Optional[Dict[str, str]] = None,
-    ) -> None:
-        """Start an app activity.
-
-        Args:
-            package: Package name
-            activity: Activity name
-            extras: Intent extras
-        """
-        cmd = f"am start -n {package}/{activity}"
-        if extras:
-            for key, value in extras.items():
-                cmd += f" -e {key} {value}"
-        await self._adb.shell(self._serial, cmd)
-
-    async def start_app(self, package: str, activity: str = "") -> str:
-        """Start an app on the device.
-
-        Args:
-            package: Package name
-            activity: Optional activity name (if empty, launches default activity)
-
-        Returns:
-            Result message
-        """
-        if activity:
-            if not activity.startswith(".") and "." not in activity:
-                activity = f".{activity}"
-
-            if (
-                not activity.startswith(".")
-                and "." in activity
-                and not activity.startswith(package)
-            ):
-                # Fully qualified activity name
-                component = activity.split("/", 1)
-                return await self.start_activity(
-                    component[0], component[1] if len(component) > 1 else activity
-                )
-
-            # Relative activity name
-            return await self.start_activity(package, activity)
-
-        # Start main activity using monkey
-        cmd = f"monkey -p {package} -c android.intent.category.LAUNCHER 1"
-        result = await self._adb.shell(self._serial, cmd)
-        return f"Started {package}"
-
-    async def install_app(
-        self, apk_path: str, reinstall: bool = False, grant_permissions: bool = True
-    ) -> str:
-        """Install an APK on the device.
-
-        Args:
-            apk_path: Path to the APK file
-            reinstall: Whether to reinstall if app exists
-            grant_permissions: Whether to grant all requested permissions
-
-        Returns:
-            Installation result
-        """
-        if not os.path.exists(apk_path):
-            return f"Error: APK file not found: {apk_path}"
-
-        # Build install command args
-        install_args = ["install"]
-        if reinstall:
-            install_args.append("-r")
-        if grant_permissions:
-            install_args.append("-g")
-        install_args.append(apk_path)
-
-        try:
-            stdout, stderr = await self._adb._run_device_command(
-                self._serial,
-                install_args,
-                timeout=120,  # Longer timeout for installation
-            )
-
-            if "success" in stdout.lower():
-                return f"Successfully installed {os.path.basename(apk_path)}"
-            return f"Installation failed: {stdout or stderr}"
-
-        except Exception as e:
-            return f"Installation failed: {str(e)}"
-
-    async def uninstall_app(self, package: str, keep_data: bool = False) -> str:
-        """Uninstall an app from the device.
-
-        Args:
-            package: Package name to uninstall
-            keep_data: Whether to keep app data and cache directories
-
-        Returns:
-            Uninstallation result
-        """
-        cmd = ["pm", "uninstall"]
-        if keep_data:
-            cmd.append("-k")
-        cmd.append(package)
-
-        result = await self._adb.shell(self._serial, " ".join(cmd))
-        return result.strip()
-
-    async def take_screenshot(self, quality: int = 75) -> Tuple[str, bytes]:
-        """Take a screenshot of the device and compress it.
-
-        Args:
-            quality: JPEG quality (1-100, lower means smaller file size)
-
-        Returns:
-            Tuple of (local file path, screenshot data as bytes)
-        """
-        # Create a temporary file for the screenshot
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp:
-            screenshot_path = temp.name
-
-        try:
-            # Generate a random filename for the device
-            timestamp = int(time.time())
-            random_suffix = "".join(
-                random.choices(string.ascii_lowercase + string.digits, k=8)
-            )
-            device_path = f"/sdcard/screenshot_{timestamp}_{random_suffix}.png"
-
-            # Take screenshot using screencap command
-            await self._adb.shell(self._serial, f"screencap -p {device_path}")
-
-            # Pull screenshot to local machine
-            await self._adb._run_device_command(
-                self._serial, ["pull", device_path, screenshot_path]
-            )
-
-            # Clean up on device
-            await self._adb.shell(self._serial, f"rm {device_path}")
-
-            # Read the screenshot file
-            with open(screenshot_path, "rb") as f:
-                screenshot_data = f.read()
-
-            # Convert and compress the image
-            try:
-                from PIL import Image
-                import io
-
-                # Create buffer for the compressed image
-                buffer = io.BytesIO()
-
-                # Load the PNG data into a PIL Image
-                with Image.open(io.BytesIO(screenshot_data)) as img:
-                    # Convert to RGB (removing alpha channel if present) and save as JPEG
-                    converted_img = img.convert("RGB") if img.mode == "RGBA" else img
-                    converted_img.save(
-                        buffer, format="JPEG", quality=quality, optimize=True
-                    )
-                    compressed_data = buffer.getvalue()
-
-                # Get size reduction info for logging
-                png_size = len(screenshot_data) / 1024
-                jpg_size = len(compressed_data) / 1024
-                reduction = 100 - (jpg_size / png_size * 100) if png_size > 0 else 0
-
-                import logging
-
-                logger = logging.getLogger("droidrun")
-                logger.debug(
-                    f"Screenshot compressed successfully: {png_size:.1f}KB → {jpg_size:.1f}KB ({reduction:.1f}% reduction)"
-                )
-
-                return screenshot_path, compressed_data
-            except ImportError:
-                # If PIL is not available, return the original PNG data
-                logger.warning("PIL not available, returning uncompressed screenshot")
-                return screenshot_path, screenshot_data
-            except Exception as e:
-                # If compression fails, return the original PNG data
-                logger.warning(
-                    f"Screenshot compression failed: {e}, returning uncompressed"
-                )
-                return screenshot_path, screenshot_data
-
-        except Exception as e:
-            # Clean up in case of error
-            try:
-                os.unlink(screenshot_path)
-            except OSError:
-                pass
-            raise RuntimeError(f"Screenshot capture failed: {str(e)}")
-        
-    def _parse_package_list(self, output: str) -> List[Dict[str, str]]:
-        """Parse the output of 'pm list packages -f' command.
-
-        Args:
-            output: Raw command output from 'pm list packages -f'
-
-        Returns:
-            List of dictionaries containing package info with 'package' and 'path' keys
-        """
-        apps = []
-        for line in output.splitlines():
-            if line.startswith("package:"):
-                # Format is: "package:/path/to/base.apk=com.package.name"
-                path_and_pkg = line[8:]  # Strip "package:"
-                if "=" in path_and_pkg:
-                    path, package = path_and_pkg.rsplit("=", 1)
-                    apps.append({"package": package.strip(), "path": path.strip()})
-        return apps
-
-    async def list_packages(self, include_system_apps: bool = False) -> List[str]:
-        """
-        List installed packages on the device.
-
-        Args:
-            include_system_apps: Whether to include system apps (default: False)
-
-        Returns:
-            List of package names
-        """
-        # Use the direct ADB command to get packages with paths
-        cmd = ["pm", "list", "packages", "-f"]
-        if not include_system_apps:
-            cmd.append("-3")
-
-        output = await self.shell(" ".join(cmd))
-
-        # Parse the package list using the function
-        packages = self._parse_package_list(output)
-        # Format package list for better readability
-        package_list = [pack["package"] for pack in packages]
-        #for package in package_list:
-        #    print(package)
-        return package_list

droidrun/adb/manager.py

@@ -1,93 +0,0 @@
-"""
-Device Manager - Manages Android device connections.
-"""
-
-from typing import Dict, List, Optional
-from droidrun.adb.wrapper import ADBWrapper
-from droidrun.adb.device import Device
-
-class DeviceManager:
-    """Manages Android device connections."""
-
-    def __init__(self, adb_path: Optional[str] = None):
-        """Initialize device manager.
-        
-        Args:
-            adb_path: Path to ADB binary
-        """
-        self._adb = ADBWrapper(adb_path)
-        self._devices: Dict[str, Device] = {}
-
-    async def list_devices(self) -> List[Device]:
-        """List connected devices.
-        
-        Returns:
-            List of connected devices
-        """
-        devices_info = await self._adb.get_devices()
-        
-        # Update device cache
-        current_serials = set()
-        for device_info in devices_info:
-            serial = device_info["serial"]
-            current_serials.add(serial)
-            
-            if serial not in self._devices:
-                self._devices[serial] = Device(serial, self._adb)
-                
-        # Remove disconnected devices
-        for serial in list(self._devices.keys()):
-            if serial not in current_serials:
-                del self._devices[serial]
-                
-        return list(self._devices.values())
-
-    async def get_device(self, serial: str | None = None) -> Optional[Device]:
-        """Get a specific device.
-        
-        Args:
-            serial: Device serial number
-            
-        Returns:
-            Device instance if found, None otherwise
-        """
-        if serial and serial in self._devices:
-            return self._devices[serial]
-            
-        # Try to find the device
-        devices = await self.list_devices()
-        for device in devices:
-            if device.serial == serial or not serial:
-                return device
-                
-        return None
-
-    async def connect(self, host: str, port: int = 5555) -> Optional[Device]:
-        """Connect to a device over TCP/IP.
-        
-        Args:
-            host: Device IP address
-            port: Device port
-            
-        Returns:
-            Connected device instance
-        """
-        try:
-            serial = await self._adb.connect(host, port)
-            return await self.get_device(serial)
-        except Exception:
-            return None
-
-    async def disconnect(self, serial: str) -> bool:
-        """Disconnect from a device.
-        
-        Args:
-            serial: Device serial number
-            
-        Returns:
-            True if disconnected successfully
-        """
-        success = await self._adb.disconnect(serial)
-        if success and serial in self._devices:
-            del self._devices[serial]
-        return success