pantoqa-bridge 0.4.4__py3-none-any.whl

pantoqa_bridge/utils/process.py

@@ -0,0 +1,183 @@
+import asyncio
+import os
+import signal
+import socket
+import subprocess
+import time
+from collections.abc import AsyncIterator, Awaitable, Callable
+
+from pantoqa_bridge.config import IS_WINDOWS
+from pantoqa_bridge.logger import logger
+
+
+def create_stream_pipe() -> tuple[
+  AsyncIterator[str],
+  Callable[[str], Awaitable[None]],
+  asyncio.Event,
+]:
+  queue: asyncio.Queue[str] = asyncio.Queue()
+  done: asyncio.Event = asyncio.Event()
+
+  async def push_data_to_stream(message: str) -> None:
+    queue.put_nowait(message)
+
+  async def create_stream() -> AsyncIterator[str]:
+    try:
+      while not done.is_set():
+        yield await queue.get()
+    except asyncio.CancelledError:
+      pass
+
+  return create_stream(), push_data_to_stream, done
+
+
+async def stream_process(
+  process: asyncio.subprocess.Process,
+  on_data: Callable[[str, str], Awaitable[None]],
+) -> int:
+  queue: asyncio.Queue[tuple[str, str] | None] = asyncio.Queue()
+
+  async def stream_output(stream: asyncio.StreamReader | None, label: str) -> None:
+    if not stream:
+      queue.put_nowait(None)
+      return
+    while True:
+      line = await stream.readline()
+      if not line:
+        queue.put_nowait(None)
+        break
+      queue.put_nowait((label, line.decode().strip()))
+
+  tasks: list[asyncio.Task[None]] = []
+  tasks.append(asyncio.create_task(stream_output(process.stdout, "stdout")))
+  tasks.append(asyncio.create_task(stream_output(process.stderr, "stderr")))
+
+  # Stream output as it comes
+  completed_streams = 0
+  while completed_streams < len(tasks):
+    try:
+      item = await asyncio.wait_for(queue.get(), timeout=0.1)
+      if item is None:
+        completed_streams += 1
+      else:
+        stream_type, data = item
+        await on_data(stream_type, data)
+    except TimeoutError:
+      continue
+
+  # Wait for tasks to complete
+  await asyncio.gather(*tasks)
+  return await process.wait()
+
+
+async def wait_for_port_to_alive(port: int, host: str = "127.0.0.1", timeout=15):
+  start = time.time()
+  while time.time() - start < timeout:
+    try:
+      with socket.create_connection((host, port), timeout=1):
+        return True
+    except OSError:
+      await asyncio.sleep(0.5)
+  raise TimeoutError("Appium did not start in time")
+
+
+def watch_process_bg(
+  pid: int,
+  on_exit: Callable[[int], Awaitable[None] | None],
+  *,
+  poll_interval: float = 2.0,
+) -> None:
+
+  async def _watcher():
+    try:
+      while True:
+        if not is_process_alive_sync(pid):
+          result = on_exit(pid)
+          if asyncio.iscoroutine(result):
+            await result
+          return
+
+        await asyncio.sleep(poll_interval)
+    except asyncio.CancelledError:
+      return
+
+  asyncio.create_task(_watcher())
+
+
+def is_process_alive_sync(pid: int) -> bool:
+  # TODO: Can we use psutil here?
+  try:
+    if IS_WINDOWS:
+      # Windows: use tasklist to check if PID exists
+      out = subprocess.check_output(
+        ["tasklist", "/FI", f"PID eq {pid}", "/NH"],
+        stderr=subprocess.DEVNULL,
+      ).decode().strip()
+      # tasklist returns "INFO: No tasks are running..." if PID doesn't exist
+      return str(pid) in out and "INFO:" not in out
+    else:
+      # Unix/Linux: use ps
+      out = subprocess.check_output(["ps", "-o", "state=", "-p", str(pid)]).decode().strip()
+      return "z" not in out.lower()
+  except subprocess.CalledProcessError:
+    return False
+
+
+def kill_by_pid(pid: int):
+  # TODO: Can we use psutil here?
+  if IS_WINDOWS:
+    subprocess.run(
+      ["taskkill", "/PID", str(pid)],
+      stderr=subprocess.DEVNULL,
+      stdout=subprocess.DEVNULL,
+    )
+  else:
+    os.kill(pid, signal.SIGTERM)
+
+
+def force_kill_by_pid(pid: int):
+  # TODO: Can we use psutil here?
+  if IS_WINDOWS:
+    subprocess.run(
+      ["taskkill", "/F", "/PID", str(pid)],
+      stderr=subprocess.DEVNULL,
+      stdout=subprocess.DEVNULL,
+    )
+  else:
+    os.kill(pid, signal.SIGKILL)
+
+
+def kill_self_process():
+  # DON'T WAIT HERE
+  self_pid = os.getpid()
+  logger.info(f"Killing self process {self_pid}...")
+  kill_by_pid(self_pid)
+
+
+def kill_process_sync(pid: int, timeout: int = 10) -> None:
+  if pid <= 0:
+    return
+
+  if not is_process_alive_sync(pid):
+    return
+
+  # Step 1: graceful terminate
+  logger.info(f"Terminating process {pid}...")
+  if kill_by_pid(pid):
+    return
+
+  if not is_process_alive_sync(pid):
+    return
+
+  # Step 2: wait
+  logger.info(f"Waiting for process {pid} to terminate...")
+  start = time.time()
+  while time.time() - start < timeout:
+    if not is_process_alive_sync(pid):
+      logger.info(f"Process {pid} terminated.")
+      return
+    time.sleep(0.2)
+
+  # Step 3: force kill
+  logger.info(f"Killing process {pid}...")
+  force_kill_by_pid(pid)

pantoqa_bridge-0.4.4.dist-info/METADATA

@@ -0,0 +1,48 @@
+Metadata-Version: 2.4
+Name: pantoqa_bridge
+Version: 0.4.4
+Summary: Panto QA Bridge
+Author-email: Ritwick Dey <ritwick@getpanto.ai>
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.110.0
+Requires-Dist: uvicorn[standard]>=0.24.0
+Requires-Dist: Appium-Python-Client>=4.0.0
+Requires-Dist: click>=8.2.1
+Requires-Dist: dotenv>=0.9.9
+Requires-Dist: aiohttp>=3.13.2
+Requires-Dist: rich>=14.2.0
+Requires-Dist: uiautomator2>=3.5.0
+Requires-Dist: adbutils>=2.12.0
+Requires-Dist: appium-utility>=0.3.0
+Requires-Dist: packaging>=25.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: build>=1.3.0; extra == "dev"
+Requires-Dist: pre_commit>=4.5.1; extra == "dev"
+Requires-Dist: mypy>=1.19.1; extra == "dev"
+
+# PantoAI QA Bridge
+
+PantoAI QA Bridge connects the PantoAI dashboard to your local environment so you can execute mobile tests on real devices connected to your machine. Keep the bridge running to unlock local testing features from the dashboard.
+
+## Installation
+
+```bash
+pipx install pantoqa_bridge
+pantoqa_bridge
+```
+
+## Prerequisites
+
+- Python
+- Node.js and npm
+- Appium with the uiautomator2 driver
+- Android Debug Bridge (adb)
+
+## How It Fits Together
+
+```
+PantoAI Dashboard  ->  PantoAI QA Bridge  ->  Your Devices
+	(Browser)              (this tool)         (USB/Wi‑Fi)
+```

pantoqa_bridge-0.4.4.dist-info/RECORD

@@ -0,0 +1,25 @@
+pantoqa_bridge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pantoqa_bridge/__main__.py,sha256=5m0xHXIz0lDmzAOmkxmaRSUUVKA6_CDisaFxfd9Xe08,57
+pantoqa_bridge/cli.py,sha256=1EK3P3XcmFl4Il3-KNBNWddbL11ENR7Fi8zstXj1QIM,2016
+pantoqa_bridge/config.py,sha256=yJ_860ih9BG5e6n70tN_3EkpxCLzjgrpvDx-QFo86Fc,852
+pantoqa_bridge/logger.py,sha256=emPHG2YMPvHifHVbmzh3G2ZHDiX7tiS-iB4mUtSMfMY,899
+pantoqa_bridge/scrcpy-server-v3.3.1,sha256=oPcLIKpJmPv2WMlBGM1sjatqu7Bkejvas0TXC8Hry7g,90788
+pantoqa_bridge/server.py,sha256=tFvnQ9xbJQ14vuDEuJLuisxwdaXAIDhiBou8Yy59PGA,6509
+pantoqa_bridge/lib/adb.py,sha256=_4Vd3VUfklRqHFLhoMQdbzH2ZnS-lhQcCwu-p3a-SLY,2532
+pantoqa_bridge/lib/scrcpy.py,sha256=UNesa2XrV54r56MIm15X67HLka94I6mheNqKh4Bw54A,4023
+pantoqa_bridge/models/code_execute.py,sha256=SsvzgBmP1FweXhRXIirmQ_R2EtFmf6Kox567mrGht9U,87
+pantoqa_bridge/routes/action.py,sha256=pplP4eQI1qK7LIKxXb4I5j6tkcT6MowoIAtz-RMPwQI,3339
+pantoqa_bridge/routes/adb.py,sha256=sDy3UjeB9bNZPn7w07avoLKJ1K3KCk0V5gSX3pxoDdE,369
+pantoqa_bridge/routes/executor.py,sha256=gHRlNcncohPsD-VtaFTTBcu6JHrD4F-LvvETdoisuZI,3503
+pantoqa_bridge/routes/misc.py,sha256=TWz9aErGGdg96K-wnxvCfh2vjmdZmJEy9Wmih0YsbWg,2122
+pantoqa_bridge/routes/screen_mirror.py,sha256=KXpqKC0GzTW5JAoLZ0cKUY1RqZE3bNpk3zVVc5MReRE,2380
+pantoqa_bridge/service/uiautomator_action.py,sha256=UE_5-oyE_z5SjGa6Mtioap5jEM5MR9leXTdRuDl-vDI,2766
+pantoqa_bridge/tasks/executor.py,sha256=PWAS5hP79nYMLz4sNHLEYhF33YmslbcrjnqVr78rs2k,2683
+pantoqa_bridge/utils/misc.py,sha256=8tFn8p2OhLNIdT31w0LczQ8hjkhkpx3fkaeeMgs71sM,3668
+pantoqa_bridge/utils/pkg.py,sha256=difSA1MW6auFUNzOH7Emmukh5mqdsKb-Ow3jdiKKatg,1297
+pantoqa_bridge/utils/process.py,sha256=O71jC8twabT2MfcowAlMTOFeCQRb17MGTc3BFG4E9Bs,4674
+pantoqa_bridge-0.4.4.dist-info/METADATA,sha256=XuyB6iCj-9sH0RyU4h7Xzr2pZylaAkNJJi02z-PLXBE,1350
+pantoqa_bridge-0.4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pantoqa_bridge-0.4.4.dist-info/entry_points.txt,sha256=GBytwhIuiZGX_cj_1JQpUteBsP51TSxTk4NxTp4bN_I,58
+pantoqa_bridge-0.4.4.dist-info/top_level.txt,sha256=r03tgM1pQrHwfxF9gkvU94HgR_s8tqDvZVSAzq8qwrA,15
+pantoqa_bridge-0.4.4.dist-info/RECORD,,

pantoqa_bridge-0.4.4.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

pantoqa_bridge-0.4.4.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+pantoqa_bridge = pantoqa_bridge.cli:cli

pantoqa_bridge-0.4.4.dist-info/top_level.txt

@@ -0,0 +1 @@
+pantoqa_bridge