Behave+PyAutoGUI自动化脚本报错求助：红气泡点击步骤异常

先把你的测试场景和报错代码梳理清楚，方便定位问题：

你的Gherkin测试场景 #

Scenario: 从概览页导航至系统详情页
Given 登录页已显示
When 选择登录后用户点击红色气泡
Then 气泡详情页显示

报错的@When步骤代码 #

@When('the user selects a red bubble')
def click_checkpoint(self):
    def find_checkpoint(self):
        p = pyautogui.locateOnScreen('venv/screens/2018-04-26_0806.png')
        c = pyautogui.center(p)
        d = list(c)
        find_checkpoint(pyautogui.click(d[0], d[1], duration=1))

代码中的核心问题分析 #

1. 步骤文本不匹配
   你的Gherkin里的When步骤是选择登录后用户点击红色气泡，但代码里的@When装饰器文本是'the user selects a red bubble'，Behave无法匹配到对应的步骤定义，这是最基础的匹配问题。

2. 递归调用完全错误
   嵌套的find_checkpoint函数存在两个致命问题：

   • 递归调用时传入的是pyautogui.click(...)的返回值（该方法返回None），但函数定义要求接收self参数，直接触发参数不匹配报错；
   • 没有任何递归终止条件，哪怕成功点击气泡后，函数还会无限递归，最终导致栈溢出。

3. 未处理图片查找失败的情况
   pyautogui.locateOnScreen如果找不到目标图片会返回None，此时调用pyautogui.center(p)会直接抛出AttributeError: 'NoneType' object has no attribute 'center'。

修复后的代码示例 #

from behave import when
import pyautogui
import time

@When('选择登录后用户点击红色气泡')  # 严格匹配Gherkin中的步骤文本
def step_impl(context):
    max_retries = 5
    retry_count = 0
    bubble_location = None
    
    # 增加重试逻辑，避免单次查找失败
    while retry_count < max_retries:
        try:
            # 可选：添加confidence参数（需安装opencv-python）提升容错率
            bubble_location = pyautogui.locateOnScreen(
                'venv/screens/2018-04-26_0806.png', 
                confidence=0.8
            )
            if bubble_location:
                break
        except Exception as e:
            print(f"第{retry_count+1}次查找红色气泡失败: {str(e)}")
            time.sleep(1)
            retry_count += 1
    
    if not bubble_location:
        raise AssertionError("多次重试后仍无法找到红色气泡图片")
    
    # 获取气泡中心并点击
    center_point = pyautogui.center(bubble_location)
    pyautogui.click(center_point.x, center_point.y, duration=1)

额外优化建议 #

• 尽量使用绝对路径指定图片位置，避免因工作目录变化导致的路径错误；
• PyAutoGUI的图片识别受屏幕分辨率、窗口缩放影响极大，如果仍不稳定，可以尝试截取更小的特征区域作为识别图；
• Behave步骤函数建议保留context参数（标准写法），方便在不同步骤间传递测试数据。

内容的提问来源于stack exchange，提问作者Christian Busse