【实测有点东西】使用IMA知识库投喂影刀RPA开发文档后让DeepSeek帮忙写代码

from xbot import web, app
from xbot import sleep
import xbot
def main(args):
    # 1.读取Excel中的关键词（假设数据在A列）
    keywords = app.databook.get_range(1, 'A', 0, 'A')  # end_row=0表示读取到最后
    
    # 创建结果存储列表
    all_results = []
    
    # 2.遍历每个关键词
    for keyword_row in keywords:
        keyword = keyword_row[0]  # 取A列单元格内容
        
        try:
            # 3.打开浏览器访问百度
            browser = web.create("https://www.baidu.com", "chrome")
            
            # 4.输入关键词并搜索
            search_input = browser.find('//input[@id="kw"]', by='xpath')  # 百度搜索框[2](@ref)
            search_input.input(keyword)
            search_btn = browser.find('//input[@id="su"]', by='xpath')     # 百度一下按钮
            search_btn.click()
            sleep(2)  # 等待搜索结果加载[1](@ref)
            
            current_page = 1
            while current_page <= 20:  # 控制翻页20次
                # 5.提取本页搜索结果（示例：获取所有结果标题）
                results = browser.find_all('//h3[@class="t"]', by='xpath')  # 标题元素
                page_data = []
                
                for result in results:
                    try:
                        title = result.get_text().strip()
                        page_data.append([keyword, current_page, title])
                    except:
                        continue
                
                # 存储到总结果
                all_results.extend(page_data)
                
                # 6.翻页操作
                try:
                    next_btn = browser.find('//a[@class="n"]', by='xpath')  # 下一页按钮[1](@ref)
                    next_btn.click()
                    sleep(3)  # 等待新页面加载[1](@ref)
                    current_page += 1
                except:
                    break  # 翻页失败则终止当前关键词
                
            # 关闭浏览器
            browser.close()
            
        except Exception as e:
            print(f"处理关键词【{keyword}】时出错：{str(e)}")
            continue
    
    # 7.将结果写入Excel（写入新的Sheet）
    if len(all_results) > 0:
        app.databook.set_range(
            row_num=1, 
            col_name='A', 
            values=all_results,
            sheet_name='搜索结果'
        )
if __name__ == '__main__':
    main(None)