Python Selenium邮件主题关键词匹配逻辑优化咨询

问题原因排查 #

第一个版本any()无法正常匹配的核心问题有三个：

• 代码缩进错误：if any(...) 对应的else缩进层级不匹配，分支逻辑未按预期执行
• 循环中反复调用Selenium元素的get_property方法，容易触发StaleElementReferenceException导致主题内容获取失败
• 关键词大小写重复定义，匹配逻辑没有做大小写兼容，容易漏匹配

优化实现方案 #

from time import sleep
from selenium.common.exceptions import StaleElementReferenceException

def email_delete(browser):
    user_input = input("Would you like to delete some emails?: > Y/N")
    # 统一用小写关键词，后续匹配时统一转小写判断，无需重复定义大小写版本
    key_words = ["order", "re:", "purchase"]
    email_search(browser)
    sleep(3)
    
    # 提前一次性提取所有邮件主题到本地列表，避免循环操作DOM元素引发失效问题
    email_subject_list = []
    try:
        email_subject_elements = browser.find_elements_by_xpath("//span[@data-testid='message-column:subject']")
        for ele in email_subject_elements:
            subject_title = str(ele.get_property('title')).strip()
            email_subject_list.append(subject_title)
    except StaleElementReferenceException as err:
        print(f"邮件主题拉取失败：{err}")
        return

    if user_input.strip().lower() == "y":
        for subject in email_subject_list:
            # 单个主题只判断一次，命中任意关键词即保留
            if any(key in subject.lower() for key in key_words):
                print(f"Keep > {subject}")
            else:
                print(f"Loss > {subject}")
                # 后续可直接补充对应邮件的删除逻辑

方案优势 #

• 修复原代码缩进、参数拼写错误，逻辑分支可正常执行
• 提前拉取全量主题存到本地，完全规避Selenium元素失效问题
• 大小写统一匹配逻辑更严谨，无需重复定义多版本关键词
• 仅需一次遍历所有邮件主题即可完成判断，时间复杂度为O(n)（n为邮件数量），比双层循环效率高，无重复输出问题

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