ai-kf/fix_confirm_btn.py

import sys

def replace_locators():
    with open('/Users/getmind/ds/kf/erp_automation.py', 'r') as f:
        content = f.read()

    # The string that might appear
    old_locators = [
        "\"input#confirm_confirm, span#confirmBtn, .layui-layer-btn0, a:text-is('确定'), button:text-is('确定')\"",
        "\"input#confirm_confirm, span#confirmBtn, span.btn_1:has-text('确认'), span:text-is('确认'), .layui-layer-btn0, a:text-is('确定'), button:text-is('确定')\"",
        "'input#confirm_confirm, span#confirmBtn, .layui-layer-btn0, a:text-is(\\'确定\\'), button:text-is(\\'确定\\')'",
    ]
    
    new_locator = "\"input#confirm_confirm, span#confirmBtn, span.btn_1:has-text('确认'), span:text-is('确认'), .layui-layer-btn0, a:text-is('确定'), button:text-is('确定'), a:text-is('是'), button:text-is('是'), span:text-is('是'), div:text-is('是')\""

    for old_loc in old_locators:
        content = content.replace(old_loc, new_locator)
        
    with open('/Users/getmind/ds/kf/erp_automation.py', 'w') as f:
        f.write(content)

if __name__ == '__main__':
    replace_locators()