【Selenium + Python】之 文件读取数据并运用数据

目录

• 从excel读取数据进行搜索

• 从xml读取数据进行搜索

• 从csv读取数据进行搜索

一、从excel读取数据进行搜索

封装读取方法：

import xlrd
from selenium import webdriver
from selenium.webdriver.common.by import By



class rdExcel():
    def __init__(self,excel_dir,sheet_name):
        self.r = []
        self.rd = xlrd.open_workbook(excel_dir)
        self.sh = self.rd.sheet_by_name(sheet_name)
        #首行设置为key
        self.key = self.sh.row_values(0)
        #获取总行数
        self.rownum = self.sh.nrows
        #获取总列数
        self.colnum = self.sh.ncols


    def function(self):
        if self.rownum<=1:
            print("没有获取到数值")
        else:
            r = []
            j=1
            #要执行的行数
            for i in range(self.rownum - 1):
                s = {}
                values = self.sh.row_values(j)
                for x in range(self.colnum):
                    s[self.key[x]] = values[x]
                r.append(s)
                j+=1
            # print(r)
            return r


if __name__ == '__main__':
    a = input("excel_dir:")
    b = input("sheet_name:")
    data = rdExcel(a,b)
    print(data.function())

基本操作：指定单元格读取数据

rd = xlrd.open_workbook("C:\\Users\\ZHANGCH\\Desktop\\test99.xlsx")
sh = rd.sheet_by_name("rdData")

value = sh.row_values(1)[0]
driver = webdriver.Chrome()
driver.maximize_window()
driver.get("http://www.baidu.com")
driver.find_element(By.CSS_SELECTOR,"#kw").send_keys(value)
driver.find_element(By.CSS_SELECTOR,"#su").click()