AI Studio 数据集创建自动化:Selenium 代码示例
以下是针对 aistudio.baidu.com 网站,生成'数据集'相关的 Selenium 前端自动化代码:
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
# 初始化浏览器
browser = webdriver.Chrome()
# 打开网站
browser.get('https://aistudio.baidu.com/aistudio/datasetoverview')
# 等待页面加载完成
wait = WebDriverWait(browser, 10)
wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'dataset-card')))
# 点击'创建数据集'按钮
create_button = browser.find_element_by_xpath('//div[@class='create-dataset']/button')
create_button.click()
# 等待弹窗出现
wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'create-dialog')))
# 填写数据集名称
name_input = browser.find_element_by_xpath('//div[@class='create-dialog']//input[@name='name']')
name_input.send_keys('My Dataset')
# 选择数据集类型
type_select = browser.find_element_by_xpath('//div[@class='create-dialog']//select[@name='type']')
type_select.send_keys(Keys.DOWN)
type_select.send_keys(Keys.RETURN)
# 上传数据集文件
file_input = browser.find_element_by_xpath('//div[@class='create-dialog']//input[@type='file']')
file_input.send_keys('/path/to/my/dataset/file')
# 点击'创建'按钮
submit_button = browser.find_element_by_xpath('//div[@class='create-dialog']/button[@type='submit']')
submit_button.click()
# 等待数据集创建完成
wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'dataset-card')))
# 关闭浏览器
browser.quit()
这段代码实现了以下步骤:
- 初始化浏览器,打开 aistudio.baidu.com 网站;
- 等待页面加载完成,并点击'创建数据集'按钮;
- 等待弹窗出现,并填写数据集名称、选择数据集类型、上传数据集文件;
- 点击'创建'按钮,等待数据集创建完成;
- 关闭浏览器。
你可以根据自己的需求,对这段代码进行修改和扩展。
原文地址: https://www.cveoy.top/t/topic/n471 著作权归作者所有。请勿转载和采集!