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}")
|