Kea2-python 0.2.0__tar.gz → 0.2.2__tar.gz

{kea2_python-0.2.0 → kea2_python-0.2.2}/Kea2_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Kea2-python
-Version: 0.2.0
+Version: 0.2.2
 Summary: A python library for supporting and customizing automated UI testing for mobile apps
 Author-email: Xixian Liang <xixian@stu.ecnu.edu.cn>
 Requires-Python: >=3.8
@@ -20,7 +20,7 @@ Dynamic: license-file
 
 
 <div>
-    <img src="https://github.com/user-attachments/assets/aa5839fc-4542-46f6-918b-c9f891356c84" style="border-radius: 14px; width: 20%; height: 20%;"/> 
+    <img src="https://github.com/user-attachments/assets/36ec9f2f-a3d8-482a-9a61-4785f3278991" style="border-radius: 14px; width: 20%; height: 20%;"/> 
 </div>
 
 ### Github repo link

{kea2_python-0.2.0 → kea2_python-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Kea2-python
-Version: 0.2.0
+Version: 0.2.2
 Summary: A python library for supporting and customizing automated UI testing for mobile apps
 Author-email: Xixian Liang <xixian@stu.ecnu.edu.cn>
 Requires-Python: >=3.8
@@ -20,7 +20,7 @@ Dynamic: license-file
 
 
 <div>
-    <img src="https://github.com/user-attachments/assets/aa5839fc-4542-46f6-918b-c9f891356c84" style="border-radius: 14px; width: 20%; height: 20%;"/> 
+    <img src="https://github.com/user-attachments/assets/36ec9f2f-a3d8-482a-9a61-4785f3278991" style="border-radius: 14px; width: 20%; height: 20%;"/> 
 </div>
 
 ### Github repo link

{kea2_python-0.2.0 → kea2_python-0.2.2}/README.md

@@ -6,7 +6,7 @@
 
 
 <div>
-    <img src="https://github.com/user-attachments/assets/aa5839fc-4542-46f6-918b-c9f891356c84" style="border-radius: 14px; width: 20%; height: 20%;"/> 
+    <img src="https://github.com/user-attachments/assets/36ec9f2f-a3d8-482a-9a61-4785f3278991" style="border-radius: 14px; width: 20%; height: 20%;"/> 
 </div>
 
 ### Github repo link

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/adbUtils.py

@@ -6,7 +6,6 @@ from typing import List, Optional, Set, Tuple
 from kea2.utils import getLogger
 from adbutils import AdbDevice, adb
 from typing import IO, TYPE_CHECKING, Generator, Optional, List, Union
-from adbutils import AdbConnection
 
 logger = getLogger(__name__)
 
@@ -44,20 +43,29 @@ class ADBDevice(AdbDevice):
         super().__init__(client=adb, serial=ADBDevice.serial, transport_id=ADBDevice.transport_id)
 
     @property
-    def stream_shell(self) -> "ADBStreamShell_V2":
+    def stream_shell(self) -> "StreamShell":
         if "shell_v2" in self.get_features():
             return ADBStreamShell_V2(session=self)
         logger.warning("Using ADBStreamShell_V1. All output will be printed to stdout.")
         return ADBStreamShell_V1(session=self)
 
+    def kill_proc(self, proc_name):
+        r = self.shell(f"ps -ef")
+        pids = [l for l in r.splitlines() if proc_name in l]
+        if pids:
+            logger.info(f"{proc_name} running, trying to kill it.")
+            pid = pids[0].split()[1]
+            self.shell(f"kill {pid}")
+
 
 class StreamShell:
     def __init__(self, session: "ADBDevice"):
         self.dev: ADBDevice = session
-        self._thread = None
+        self._thread: threading.Thread = None
         self._exit_code = 255
         self.stdout = sys.stdout
         self.stderr = sys.stderr
+        self._finished = False
 
     def __call__(self, cmdargs: Union[List[str], str], stdout: IO = None, 
                  stderr: IO = None, timeout: Union[float, None] = None) -> "StreamShell":
@@ -107,7 +115,7 @@ class StreamShell:
 class ADBStreamShell_V1(StreamShell):
 
     def __call__(
-        self, cmdargs: Union[List[str]], stdout: IO = None, 
+        self, cmdargs: Union[List[str], str], stdout: IO = None, 
         stderr: IO = None, timeout: Union[float, None] = None
     ) -> "StreamShell":
         return self.shell_v1(cmdargs, stdout, stderr, timeout)
@@ -168,7 +176,7 @@ class ADBStreamShell_V2(StreamShell):
         self._exit_code = 255
 
     def __call__(
-        self, cmdargs: Union[List[str]], stdout: IO = None, 
+        self, cmdargs: Union[List[str], str], stdout: IO = None, 
         stderr: IO = None, timeout: Union[float, None] = None
     ) -> "StreamShell":
         return self.shell_v2(cmdargs, stdout, stderr, timeout)

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/bug_report_generator.py

@@ -94,7 +94,7 @@ class BugReportGenerator:
             with open(report_path, "w", encoding="utf-8") as f:
                 f.write(html_content)
 
-            logger.info(f"Bug report saved to: {report_path}")
+            logger.debug(f"Bug report saved to: {report_path}")
 
         except Exception as e:
             logger.error(f"Error generating bug report: {e}")

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/fastbotManager.py

@@ -1,5 +1,5 @@
-import subprocess
-import threading
+from retry import retry
+from retry.api import retry_call
 from dataclasses import asdict
 import requests
 from time import sleep
@@ -70,6 +70,21 @@ class FastbotManager:
             "/data/local/tmp/x86_64/libfastbot_native.so",
         )
 
+        whitelist = self.options.act_whitelist_file
+        blacklist = self.options.act_blacklist_file
+        if bool(whitelist) ^ bool(blacklist):
+            if whitelist:
+                file_to_push = cur_dir.parent / 'configs' / 'awl.strings'
+                remote_path = whitelist
+            else:
+                file_to_push = cur_dir.parent / 'configs' / 'abl.strings'
+                remote_path = blacklist
+
+            self.dev.sync.push(
+                file_to_push,
+                remote_path
+            )
+
         t = self._startFastbotService()
         logger.info("Running Fastbot...")
 
@@ -80,25 +95,18 @@ class FastbotManager:
         """
         check if the script driver and proxy server are alive.
         """
+        def _check_alive_request():
+            _http_request(dev=self.dev, device_port=8090, method="GET", path="/ping")
 
-        for _ in range(10):
-            sleep(2)
-            try:
-                _http_request(
-                    dev=self.dev,
-                    device_port=8090,
-                    method="GET",
-                    path="/ping",
-                )
-                return
-            except requests.ConnectionError:
-                logger.info("waiting for connection.")
-                pass
-        raise RuntimeError("Failed to connect fastbot")
+        try:
+            retry_call(_check_alive_request, tries=10, delay=2)
+        except requests.ConnectionError:
+            raise RuntimeError("Failed to connect fastbot")
 
     def request(self, method: str, path: str, data: Dict=None, timeout: int=10) -> HTTPResponse:
         return _http_request(self.dev, 8090, method, path, data, timeout)
 
+    @retry(Exception, tries=2, delay=2)
     def init(self, options: "Options", stamp):
         post_data = {
             "takeScreenshots": options.take_screenshots,
@@ -117,6 +125,7 @@ class FastbotManager:
         self._device_output_dir = re.match(r"outputDir:(.+)", r.text).group(1)
         print(f"[INFO] Fastbot initiated. outputDir: {r.text}", flush=True)
     
+    @retry(Exception, tries=2, delay=2)
     def stepMonkey(self, monkeyStepInfo):
         r = self.request(
             method="POST",
@@ -125,6 +134,7 @@ class FastbotManager:
         )
         return r.json()["result"]
 
+    @retry(Exception, tries=2, delay=2)
     def stopMonkey(self):
         """
         send a stop monkey request to the server.
@@ -136,6 +146,7 @@ class FastbotManager:
 
         print(f"[Server INFO] {r.text}", flush=True)
     
+    @retry(Exception, tries=2, delay=2)
     def logScript(self, execution_info: Dict):
         r = self.request(
             method="POST",
@@ -171,6 +182,14 @@ class FastbotManager:
         if self.options.profile_period:
             shell_command += ["--profile-period", f"{self.options.profile_period}"]
 
+        whitelist = self.options.act_whitelist_file
+        blacklist = self.options.act_blacklist_file
+        if bool(whitelist) ^ bool(blacklist):
+            if whitelist:
+                shell_command += ["--act-whitelist-file", f"{whitelist}"]
+            else:
+                shell_command += ["--act-blacklist-file", f"{blacklist}"]
+
         shell_command += ["-v", "-v", "-v"]
 
         full_cmd = ["adb"] + (["-s", self.options.serial] if self.options.serial else []) + ["shell"] + shell_command
@@ -182,12 +201,7 @@ class FastbotManager:
         logger.info("Launching fastbot with shell command:\n{}".format(" ".join(full_cmd)))
         logger.info("Fastbot log will be saved to {}".format(outfile.name))
 
-        # stream = self.dev.shell(shell_command, encoding="utf-8", stream=True, timeout=float("inf"))
-        # process handler
         t = self.dev.stream_shell(shell_command, stdout=outfile, stderr=outfile)
-        # proc = subprocess.Popen(full_cmd, stdout=outfile, stderr=outfile)
-        # t = threading.Thread(target=self.close_on_exit, args=(proc, outfile), daemon=True)
-        # t.start()
         return t
 
     def close_on_exit(self, proc: ADBStreamShell_V2, f: IO):
@@ -203,6 +217,8 @@ class FastbotManager:
         return self.thread.poll()
 
     def start(self):
+        # kill the fastbot process if runing.
+        self.dev.kill_proc("com.android.commands.monkey")
         self.thread = self._activateFastbot()
 
     def join(self):

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/keaUtils.py

@@ -9,14 +9,15 @@ import random
 import warnings
 from dataclasses import dataclass, asdict
 import requests
-from .absDriver import AbstractDriver
+from kea2.absDriver import AbstractDriver
 from functools import wraps
-from .bug_report_generator import BugReportGenerator
-from .resultSyncer import ResultSyncer
-from .logWatcher import LogWatcher
-from .utils import TimeStamp, getProjectRoot, getLogger
-from .u2Driver import StaticU2UiObject
-from .fastbotManager import FastbotManager
+from kea2.bug_report_generator import BugReportGenerator
+from kea2.resultSyncer import ResultSyncer
+from kea2.logWatcher import LogWatcher
+from kea2.utils import TimeStamp, getProjectRoot, getLogger
+from kea2.u2Driver import StaticU2UiObject
+from kea2.fastbotManager import FastbotManager
+from kea2.adbUtils import ADBDevice
 import uiautomator2 as u2
 import types
 
@@ -133,6 +134,10 @@ class Options:
     device_output_root: str = "/sdcard"
     # the debug mode
     debug: bool = False
+    # Activity WhiteList File
+    act_whitelist_file: str = None
+    # Activity BlackList File
+    act_blacklist_file: str = None
 
     def __setattr__(self, name, value):
         if value is None:
@@ -149,6 +154,7 @@ class Options:
             if self.transport_id:
                 target_device["transport_id"] = self.transport_id
             self.Driver.setDevice(target_device)
+            ADBDevice.setDevice(self.serial, self.transport_id)
         global LOGFILE, RESFILE, STAMP
         if self.log_stamp:
             illegal_chars = ['/', '\\', ':', '*', '?', '"', '<', '>', '|', '\n', '\r', '\t', '\0']
@@ -170,12 +176,11 @@ class Options:
         if self.throttle < 0:
             raise ValueError("--throttle should be greater than or equal to 0")
 
-        _check_package_installation(self.serial, self.packageNames)
+        _check_package_installation(self.packageNames)
 
 
-def _check_package_installation(serial, packageNames):
-    from .adbUtils import get_packages
-    installed_packages = get_packages(device=serial)
+def _check_package_installation(packageNames):
+    installed_packages = set(ADBDevice().list_packages())
 
     for package in packageNames:
         if package not in installed_packages:

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/kea_launcher.py

@@ -122,6 +122,22 @@ def _set_runner_parser(subparsers: "argparse._SubParsersAction[argparse.Argument
         help="Take screenshots for every step.",
     )
 
+    parser.add_argument(
+        "--act-whitelist-file",
+        dest="act_whitelist_file",
+        required=False,
+        type=str,
+        help="Add Activity Whitelist File.",
+    )
+
+    parser.add_argument(
+        "--act-blacklist-file",
+        dest="act_blacklist_file",
+        required=False,
+        type=str,
+        help="Add Activity Blacklist File.",
+    )
+
     parser.add_argument(
         "extra",
         nargs=argparse.REMAINDER,
@@ -195,7 +211,9 @@ def run(args=None):
         log_stamp=args.log_stamp,
         profile_period=args.profile_period,
         take_screenshots=args.take_screenshots,
-        device_output_root=args.device_output_root
+        device_output_root=args.device_output_root,
+        act_whitelist_file = args.act_whitelist_file,
+        act_blacklist_file=args.act_blacklist_file
     )
 
     KeaTestRunner.setOptions(options)

kea2_python-0.2.2/kea2/logWatcher.py

@@ -0,0 +1,79 @@
+import re
+import os
+import threading
+import time
+from typing import IO
+from kea2.utils import getLogger
+
+
+logger = getLogger(__name__)
+
+
+PATTERN_EXCEPTION = re.compile(r"\[Fastbot\].+Internal\serror\n([\s\S]*)")
+PATTERN_STATISTIC = re.compile(r".+Monkey\sis\sover!\n([\s\S]+)")
+
+
+def thread_excepthook(args):
+    print(args.exc_value, flush=True)
+    os._exit(1)
+
+
+
+class LogWatcher:
+
+    def watcher(self, poll_interval=3):
+        self.last_pos = 0
+
+        with open(self.log_file, "r", encoding="utf-8") as fp:
+            while not self.end_flag:
+                self.read_log(fp)
+                time.sleep(poll_interval)
+            
+            time.sleep(0.2)
+            self.read_log(fp)
+        
+    def read_log(self, f: IO):
+        f.seek(self.last_pos)
+        buffer = f.read()
+        self.last_pos = f.tell()
+
+        self.parse_log(buffer)
+
+    def parse_log(self, content):
+        exception_match = PATTERN_EXCEPTION.search(content)
+        if exception_match:
+            exception_body = exception_match.group(1).strip()
+            if exception_body:
+                raise RuntimeError(
+                    "[Error] Fatal Execption while running fastbot:\n" + 
+                    exception_body + 
+                    "\nSee fastbot.log for details."
+                )
+        
+        statistic_match = PATTERN_STATISTIC.search(content)
+        if statistic_match:
+            statistic_body = statistic_match.group(1).strip()
+            if statistic_body:
+                print(
+                    "[INFO] Fastbot exit:\n" + 
+                    statistic_body
+                , flush=True)
+
+    def __init__(self, log_file):
+        logger.info(f"Watching log: {log_file}")
+        self.log_file = log_file
+        self.end_flag = False
+
+        threading.excepthook = thread_excepthook
+        self.t = threading.Thread(target=self.watcher, daemon=True)
+        self.t.start()
+    
+    def close(self):
+        logger.info("Close: LogWatcher")
+        self.end_flag = True
+        if self.t:
+            self.t.join()
+
+
+if __name__ == "__main__":
+    LogWatcher("/Users/atria/Desktop/coding/Kea2/output/res_2025062510_0420056539/fastbot_2025062510_0420056539.log")

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/templates/bug_report_template.html

@@ -696,6 +696,7 @@
                         x: {
                             type: 'linear',
                             beginAtZero: true,
+                            max: steps.length > 0 ? Math.max(...steps) : 10,
                             grid: {
                                 display: false
                             },
@@ -707,9 +708,12 @@
                                 }
                             },
                             ticks: {
-                                stepSize: Math.max(10, Math.ceil(Math.max(...steps) / 10)),
+                                stepSize: steps.length > 0 ? Math.max(1, Math.ceil(Math.max(...steps) / 10)) : 1,
                                 callback: function(value) {
-                                    return value;
+                                    if (Number.isInteger(value) && value <= (steps.length > 0 ? Math.max(...steps) : 10)) {
+                                        return value;
+                                    }
+                                    return null;
                                 }
                             }
                         },

{kea2_python-0.2.0 → kea2_python-0.2.2}/kea2/u2Driver.py

@@ -285,13 +285,14 @@ class _HindenWidgetFilter:
 
 
 class U2StaticDevice(u2.Device):
-    def __init__(self, script_driver):
+    def __init__(self, script_driver=None):
         self.xml: etree._Element = None
         self._script_driver = script_driver
 
     def __call__(self, **kwargs):
         ui = StaticU2UiObject(session=self, selector=u2.Selector(**kwargs))
-        ui.jsonrpc = self._script_driver.jsonrpc
+        if self._script_driver:
+            ui.jsonrpc = self._script_driver.jsonrpc
         return ui
 
     @property

{kea2_python-0.2.0 → kea2_python-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "Kea2-python"
-version = "0.2.0"
+version = "0.2.2"
 description = "A python library for supporting and customizing automated UI testing for mobile apps"
 readme = "README.md"
 requires-python = ">=3.8"

{kea2_python-0.2.0 → kea2_python-0.2.2}/tests/test_u2Selector.py

@@ -1,16 +1,22 @@
 import unittest
-from kea2.u2Driver import _HindenWidgetFilter, U2Driver
+from kea2.u2Driver import U2StaticChecker, U2StaticDevice
 from lxml import etree
 from pathlib import Path
 
 
+
 XML_PATH = Path(__file__).parent / "hidden_widget_test.xml"
 
 
+class U2StaticCheckerForTest(U2StaticChecker):
+    def __init__(self):
+        self.d = U2StaticDevice(script_driver=None)
+
+
 def get_static_checker():
     xml: etree._ElementTree = etree.parse(XML_PATH)
-    d = U2Driver.getStaticChecker(xml)
-    return d
+    d = U2StaticCheckerForTest()
+    return d.getInstance(xml)
 
 
 class TestHiddenWidget(unittest.TestCase):

kea2_python-0.2.0/kea2/logWatcher.py

@@ -1,82 +0,0 @@
-import re
-import os
-import threading
-import time
-from .utils import getLogger
-
-
-logger = getLogger(__name__)
-
-
-PATTERN_EXCEPTION = re.compile(r"\[Fastbot\].+Internal\serror\n([\s\S]*)")
-PATTERN_STATISTIC = re.compile(r".+Monkey\sis\sover!\n([\s\S]+)")
-
-
-def thread_excepthook(args):
-    print(args.exc_value, flush=True)
-    os._exit(1)
-
-
-
-class LogWatcher:
-
-    def watcher(self, poll_interval=0.5):
-        self.buffer = ""
-        self.last_pos = 0
-
-        while not self.end_flag:
-            self.read_log()
-            time.sleep(poll_interval)
-        
-        time.sleep(0.2)
-        self.read_log()
-        
-    def read_log(self):
-        with open(self.log_file, 'r', encoding='utf-8') as f:
-            f.seek(self.last_pos)
-            new_data = f.read()
-            self.last_pos = f.tell()
-
-            if new_data:
-                self.buffer += new_data
-            self.parse_log()
-
-    def parse_log(self):
-        buffer = self.buffer
-        exception_match = PATTERN_EXCEPTION.search(buffer)
-        if exception_match:
-            exception_body = exception_match.group(1).strip()
-            if exception_body:
-                raise RuntimeError(
-                    "[Error] Execption while running fastbot:\n" + 
-                    exception_body + 
-                    "\nSee fastbot.log for details."
-                )
-        if self.end_flag:
-            statistic_match = PATTERN_STATISTIC.search(buffer)
-            if statistic_match:
-                statistic_body = statistic_match.group(1).strip()
-                if statistic_body:
-                    print(
-                        "[INFO] Fastbot exit:\n" + 
-                        statistic_body
-                    , flush=True)
-
-    def __init__(self, log_file):
-        logger.info(f"Watching log: {log_file}")
-        self.log_file = log_file
-        self.end_flag = False
-
-        threading.excepthook = thread_excepthook
-        self.t = threading.Thread(target=self.watcher, daemon=True)
-        self.t.start()
-    
-    def close(self):
-        logger.info("Close: LogWatcher")
-        self.end_flag = True
-        if self.t:
-            self.t.join()
-
-
-if __name__ == "__main__":
-    LogWatcher("fastbot.log")