78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
|
import asyncio
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
import shutil
|
|||
|
|
|
|||
|
|
import uiautomator2 as u2
|
|||
|
|
|
|||
|
|
|
|||
|
|
project_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|||
|
|
if project_root not in sys.path:
|
|||
|
|
sys.path.append(project_root)
|
|||
|
|
|
|||
|
|
|
|||
|
|
from Apps.XinDianTu.Kit import setup_logger, take_screenshot
|
|||
|
|
from Apps.XinDianTu.Config.Setting import SCROLL_DISTANCE_RATIO, WAIT_AFTER_SCROLL
|
|||
|
|
from Apps.XinDianTu.FirstPageKit import run_ocr_rect
|
|||
|
|
|
|||
|
|
|
|||
|
|
LOGS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Logs")
|
|||
|
|
os.makedirs(LOGS_DIR, exist_ok=True)
|
|||
|
|
LOG_FILE_PATH = os.path.join(LOGS_DIR, "T_XinDianTu_FirstPageScrollTest.log")
|
|||
|
|
|
|||
|
|
logger = setup_logger(
|
|||
|
|
"XinDianTu.FirstPageScrollTest", log_file=LOG_FILE_PATH, clear_old_log=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def capture_list_page(d, base_dir: str, index: int):
|
|||
|
|
ts = int(time.time())
|
|||
|
|
image_uuid = f"xdt_firstpage_{index}_{ts}"
|
|||
|
|
list_path = take_screenshot(d, image_uuid, save_dir=base_dir)
|
|||
|
|
logger.info(f"[{index}] 列表截图路径: {list_path}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
await run_ocr_rect(list_path, log_path=LOG_FILE_PATH)
|
|||
|
|
logger.info(f"[{index}] 已根据 OCR+LLM 算法生成首屏绿框调试图。")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception(f"[{index}] 运行 OCR 绿框测试失败: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def run():
|
|||
|
|
logger.info("请先手动打开微信并进入新电途小程序的场站列表第一页。")
|
|||
|
|
|
|||
|
|
d = u2.connect()
|
|||
|
|
w, h = d.window_size()
|
|||
|
|
device_info = d.info or {}
|
|||
|
|
device_info["width"] = w
|
|||
|
|
device_info["height"] = h
|
|||
|
|
|
|||
|
|
logger.info(f"当前设备: {device_info.get('productName')} | 分辨率: {w}x{h}")
|
|||
|
|
|
|||
|
|
base_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Output")
|
|||
|
|
if os.path.exists(base_dir):
|
|||
|
|
for name in os.listdir(base_dir):
|
|||
|
|
path = os.path.join(base_dir, name)
|
|||
|
|
if os.path.isfile(path) or os.path.islink(path):
|
|||
|
|
os.remove(path)
|
|||
|
|
elif os.path.isdir(path):
|
|||
|
|
shutil.rmtree(path)
|
|||
|
|
os.makedirs(base_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
for i in range(1, 11):
|
|||
|
|
await capture_list_page(d, base_dir, i)
|
|||
|
|
if i < 10:
|
|||
|
|
logger.info(f"[{i}] 向上滑动，scale={SCROLL_DISTANCE_RATIO}，等待 {WAIT_AFTER_SCROLL} 秒后继续截图。")
|
|||
|
|
d.swipe_ext("up", scale=SCROLL_DISTANCE_RATIO)
|
|||
|
|
await asyncio.sleep(WAIT_AFTER_SCROLL)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
try:
|
|||
|
|
asyncio.run(run())
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
logger.info("用户手动中断测试脚本。")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception(f"测试脚本运行异常: {e}")
|