python自动化浏览器操作之WorkBuddy智能体交互测试

#0：不做页面登录。第一次运行时这里改成1，然后运行时登录一下，系统会记录登录后的cookies

v_first_time=0

def workbuddychat(v_input='北京今日天气'):

    loginurl="https://www.workbuddy.cn/login/?platform=agents&state=0"

    chaturl="https://www.workbuddy.cn/app"

    options = webdriver.ChromeOptions()

    options.add_experimental_option("useAutomationExtension", False)

    options.add_experimental_option("excludeSwitches", ["enable-automation"])

    driver = webdriver.Chrome(options=options)

    global v_first_time

    #cookies_workbuddy.json文件不存在，需要做登录

    try:

        with open("c:\cookies_workbuddy.json", "r"):

            pass

    except FileNotFoundError:

        v_first_time=1        

    if v_first_time==1:

        driver.get(loginurl)

        print("首次需要登录WorkBuddy，请处理页面登录信息(20秒)....")

        time.sleep(20)

        cookies =driver.get_cookies()

        with open("c:\cookies_workbuddy.json", "w") as f:

            json.dump(cookies, f)

        v_first_time=0

    else:

        #driver.delete_all_cookies()

        driver.get(chaturl)

        with open("c:\cookies_workbuddy.json", "r") as f:

            cookies = json.load(f)

        for cookie in cookies:

            driver.add_cookie(cookie)

    driver.get(chaturl)

    driver.refresh()

    print("等待WorkBuddy页面加载...")

    #等待页面加载...

    time.sleep(5)

    v_div = driver.find_element(By.CSS_SELECTOR, "div._editable_1xsqb_1")

    v_div.send_keys(v_input)

    #v_spana = driver.find_element(By.CSS_SELECTOR, "div._editable_1xsqb_1 > p > span > span")

    #driver.execute_script("arguments[0].dispatchEvent(new KeyboardEvent('keydown',{key: '1'}))", v_spana)

    #driver.execute_script("arguments[0].dispatchEvent(new KeyboardEvent('keypress',{key: '1'}))", v_spana)

    v_sub=driver.find_element(By.CSS_SELECTOR,"[class='_icon_981i8_1 _large_981i8_32 _customContent_981i8_42']")

    ActionChains(driver).move_to_element(v_sub).click().perform()

    print("词条：'"+v_input+"',已提交WorkBuddy，深度思考中....")

    time.sleep(5)

    v_waiteflay=1

    v_result=0

    while v_waiteflay < 16:

        chat_stats=driver.find_elements(By.CSS_SELECTOR, "div.avatar-fold-line > span.avatar-fold-status")

        if chat_stats:

            if chat_stats[-1].text.startswith('已完成'):

                print("WorkBuddy交互",chat_stats[-1].text)

                v_result=1

                break;

        time.sleep(2)

        v_waiteflay+=1

        if v_waiteflay == 16:

            print('30+秒未收到反馈，本次交互失败被认定失败！请检查网络或者系统配置...')

    if v_waiteflay<16 and v_result ==1:

        chat_results=driver.find_elements(By.CLASS_NAME, "_assistantTextContent_7jxnj_195")

        print(chat_results[-1].text)

    print('WorkBuddy退出中...')

    time.sleep(3)    

    driver.quit()

def main(args):

    print('--------python模拟web页面类操作---------')

    print('--------WorkBuddy智能体交互测试---------')

    while True:

        print('----------------------------------------')

        v_input=input("请输入WorkBuddy对话内容(0退出)：")

        if v_input=='0':

            print('即将退出...')

            break

        elif v_input:

            workbuddychat(v_input)

        else:

            print('输入内容为空，系统将默认反馈："北京今日天气"词条的交互')

            workbuddychat('北京今日天气')

    return 0

if __name__ == '__main__':

    sys.exit(main(sys.argv[1:]))

ok，大概就是这些了，只是简单集成哈，跑的通就行:) ,如有需要再进行后续开发了

代码里有很多等待time.sleep时间哈，包括我设置了WorkBuddy交互的时长(如果30+秒就做退出)，大家可以根据自己网络和硬件配置进行调整

我们看一下跑的结果

我本地之前登录过，上边视频没有做WorkBuddy登录，别的也没什么了

这个集成，大家觉得有用吗？

有需要随时微信1364125229反馈哈