目录

一、从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()

============================================================================

写法进行修改规整，完整获取指定数据进行百度查询：

写法①：

import xlrd

import os

from selenium import webdriver

from selenium.webdriver.common.by import By

from time import sleep

base = os.path.dirname(os.path.dirname(__file__))

base_dir = base.replace('/','\\')

file_dir = base_dir + os.sep + "test" + os.sep + "test99.xlsx"

print(file_dir)

class test():

    def __init__(self,file_dir,sheet_name):

        self.rd = xlrd.open_workbook(file_dir)

        self.sh = self.rd.sheet_by_name(sheet_name)

        self.rows = self.sh.nrows

        self.cols = self.sh.ncols

    def ExcelRd(self):

        r = []

        for i in range(1,self.rows):

            values = self.sh.row_values(i,0,self.cols)

            r.append(values)

        return r

if __name__ == '__main__':

    #指定sheet页为：rdData

    file_dir = input("路径为：")

    sheet_name = input("sheet页为：")

    data = test(file_dir,sheet_name).ExcelRd()

    driver = webdriver.Chrome()

    driver.maximize_window()

    driver.implicitly_wait(10)

    driver.get("https://www.baidu.com")

    for footballStar in data:

        driver.find_element(By.CSS_SELECTOR,"#kw").clear()

        driver.find_element(By.CSS_SELECTOR,"#kw").send_keys(footballStar[1])

        driver.find_element(By.CSS_SELECTOR,"#su").click()

        sleep(5)

    driver.quit()

写法②：添加截图方法

function.py：

import os

def screenshot(driver,img_name):

    base = os.path.dirname(os.path.dirname(__file__))

    base_dir = base.replace("/","\\")

    img_dir = base_dir + os.sep + "20180515作业" + os.sep + "image" + os.sep + img_name + ".png"

    driver.get_screenshot_as_file(img_dir)

Excel读取数据.py：

import xlrd

import os

from selenium import webdriver

from selenium.webdriver.common.by import By

from time import sleep

from function import screenshot

class test(object):

    def __init__(self):

        self.base = os.path.dirname(os.path.dirname(__file__))

        self.base_dir = self.base.replace('/', '\\')

    def ExcelRd(self):

        file_dir = self.base_dir + os.sep + "20180515作业" + os.sep + "test_xlsx.xlsx"

        rd = xlrd.open_workbook(file_dir)

        sh = rd.sheet_by_name("rdData")

        rows = sh.nrows

        cols = sh.ncols

        r = []

        for i in range(1,rows):

            values = sh.row_values(i,0,cols)

            r.append(values)

        return r

if __name__ == '__main__':

    data = test().ExcelRd()

    driver = webdriver.Chrome()

    driver.maximize_window()

    driver.implicitly_wait(10)

    driver.get("https://www.baidu.com")

    for footballStar in data:

        fbStar = footballStar[1]

        driver.find_element(By.CSS_SELECTOR,"#kw").clear()

        driver.find_element(By.CSS_SELECTOR,"#kw").send_keys(fbStar)

        driver.find_element(By.CSS_SELECTOR,"#su").click()

        sleep(3)

        fbStar_xlsx = str("xlsx_" + fbStar)

        screenshot(driver,fbStar_xlsx)

        sleep(7)

    driver.quit()

二、从CSV读取数据进行百度搜索

CSV读取数据.py：

import csv

import os

from selenium import webdriver

from selenium.webdriver.common.by import By

from time import sleep

from function import screenshot

class test(object):

    def __init__(self):

        base = os.path.dirname(os.path.dirname(__file__))

        self.base_dir = base.replace("/","\\")

    def CSVRd(self):

        base_dir = self.base_dir + os.sep + "20180515作业" + os.sep + "test_csv.csv"

        opFile = open(base_dir,'r')

        rd = csv.reader(opFile)

        r = []

        next(rd,None)

        for i in rd:

            r.append(i)

        return r

if __name__ == '__main__':

    data = test().CSVRd()

    driver = webdriver.Chrome()

    driver.maximize_window()

    driver.get("https://www.baidu.com")

    driver.implicitly_wait(10)

    for fbStar in data:

        fbStar = fbStar[1]

        driver.find_element(By.CSS_SELECTOR,"#kw").clear()

        driver.find_element(By.CSS_SELECTOR,"#kw").send_keys(fbStar)

        driver.find_element(By.CSS_SELECTOR,"#su").click()

        sleep(3)

        csv_fbStar = str("csv_" + fbStar)

        screenshot(driver,csv_fbStar)

        sleep(7)

    driver.quit()

三、从XML读取数据进行登录操作

test_xml文件：

<?xml version="1.0" encoding="utf-8"?>

<info>

    <title>博客园登录</title>

    <url_dir>https://passport.cnblogs.com/user/signin</url_dir>

    <login username="owen_name" password="owen_pwd">登录</login>

</info>

CSV读取数据.py：

import xml.dom.minidom as minidom

# import xml.etree.ElementTree as ele

import os

from selenium import webdriver

from selenium.webdriver.common.by import By

from time import sleep

from function import screenshot

base = os.path.dirname(os.path.dirname(__file__))

base_dir = base.replace("/","\\")

file_dir = base_dir + os.sep + "20180515作业" + os.sep + "test_xml.xml"

#打开xml文档

dom = minidom.parse(file_dir)

#得到文档元素

root = dom.documentElement

#由于下面getElementsByTagName点不出来方法，手写的

tag1 = root.getElementsByTagName("login")

tag2 = root.getElementsByTagName("url_dir")

tag3 = root.getElementsByTagName("title")

#获得标签属性值

username = tag1[0].getAttribute("username")

password = tag1[0].getAttribute("password")

#获得标签之间的数据

url = tag2[0].firstChild.data

title = tag3[0].firstChild.data

driver = webdriver.Chrome()

driver.maximize_window()

driver.get(url)

driver.find_element(By.CSS_SELECTOR,"#input1").clear()

driver.find_element(By.CSS_SELECTOR,"#input1").send_keys(username)

sleep(3)

driver.find_element(By.CSS_SELECTOR,"#input2").clear()

driver.find_element(By.CSS_SELECTOR,"#input2").send_keys(password)

sleep(3)

title = str("xml_" + title)

screenshot(driver,title)

driver.quit()

四、附：学习资料

《Python不归路_xml.etree.ElementTree模块》感谢作者：深海一尾鱼

《python读取xml文件》感谢作者：虫师

【Selenium + Python】之 Excel、CSV、XML文件读取数据并运用数据百度查询的更多相关文章

  1. python解析VOC的xml文件并转成自己需要的txt格式

    在进行神经网络训练的时候,自己标注的数据集往往会有数据量不够大以及代表性不强等问题,因此我们会采用开源数据集作为训练,开源数据集往往具有特定的格式,如果我们想将开源数据集为我们所用的话,就需要对其格式 ...

  2. php xml 文件读取 XMLReader

    php xml 文件读取 <?php /** $xmlString = '<xml> <persons count="10"> <person ...

  3. javascript读取xml文件读取节点数据的例子

    分享下用javascript读取xml文件读取节点数据方法. 读取的节点数据,还有一种情况是读取节点属性数据. <head> <title></title> < ...

  4. xml文件读取到数据库

    xml文件读取到数据库   第一步,导包 c3p0,dom4j,jaxen,MySQL-connector 第二步  xml文件,config文件 第三步 javabean 第四步 c3p0的工具类 ...

  5. C#程序中：如何修改xml文件中的节点（数据）

    要想在web等程序中实现动态的数据内容给新(如网页中的Flash),不会更新xml文件中的节点(数据)是远远不够的,今天在这里说一个简单的xml文件的更新,方法比较基础,很适合初学者看的,保证一看就懂 ...

  6. Excel和XML文件导入

    using System;using System.Collections;using System.Collections.Generic;using System.Configuration;us ...

  7. XML文件读取工具类

    /// <summary> /// Author: jiangxiaoqiang /// </summary> public class XmlReader { //===== ...

  8. Excel关联xml文件

    1.新建没传值的xml文件,变量名称自己定义好 2.打开excel,如果之前没有设置过,点击选项 如果当前Excel菜单栏中没有开发工具项,在自定义功能区先勾选上开发选项 3.点右下角的xml映射 弹 ...

  9. Xml 文件读取

    .NET 读取Xml文件,用到XmlDocument类. 1.要获取文档的根: DocumentElement. 2.Attributes :获取 XmlAttributeCollection 包含此 ...

随机推荐

  1. InputSplit—>RecordReder—>map(key,value,context)的过程解析

    上图首先描述了在TaskTracker端Task(MapTask.ReduceTask)的执行过程,MapTask(org.apache.hadoop.mapred)首先被TaskRunner调用,然 ...

  2. INFORMATION_SCHEMA获取数据库的信息

    简介 information_schema这张数据表保存了MySQL服务器所有数据库的信息.如数据库名,数据库的表,表栏的数据类型与访问权限等.再简单点,这台mysql服务器上,到底有哪些数据库.各个 ...

  3. JQuery插件开发格式

    原地址 一.jQuery扩展 1.$.extend(object) 类似于.Net的扩展方法,用于扩展jQuery.然后就可以用$.的方式调用. $(function(){ $.extend({ fu ...

  4. python的几个概念

    1.函数在传递实参的时候是传递的是引用而不是从内存中重新赋相同值给形参. 2.函数名带圆括号和不带圆括号.函数名带圆括号是函数的调用,而函数名代表的是函数体. 3.函数返回值,在函数没有返回值的时候默 ...

  5. 使用echarts简单制作中国地图，echarts中国地图

    网站需要一张中国地图,并且鼠标经过某个省份,该省份的省份名字显示,而且该省份的地区会变色显示. 第一种方法: 将每个省份的图片定位(先隐藏),拼合成一张中国地图,然后再定位省份名称,鼠标经过省份名字, ...

  6. iOS uitableivewCell 下划线顶格

    - (void)tableView:(UITableView *)tableView willDisplayCell:(UITableViewCell *)cell forRowAtIndexPath ...

  7. vue vue-router beforeRouteEnter

    beforeRouteEnter (to, from, next) { // 在渲染该组件的对应路由被 confirm 前调用 // 不!能!获取组件实例 `this` // 因为当守卫执行前,组件实 ...

  8. Android Studio加入插件（Genymotion）

    官方模拟器的龟速已让我们无力吐槽.幸好有genymotion这款逆天的Android虚拟机,它有着高速的开启速度,良好的交互界面. 是Android开发必备的良品.甚至有些玩家已经用genymotio ...

  9. iOS 原生二维码扫描和生成

    代码地址如下:http://www.demodashi.com/demo/12551.html 一.效果预览: 功能描述:WSLNativeScanTool是在利用原生API的条件下封装的二维码扫描工 ...

  10. JQuery DataTables学习

    1.Datatables简单介绍 DataTables是一个jQuery的表格插件.这是一个高度灵活的工具,根据的基础逐步增强,这将添加先进的互动控制.支持不论什么HTML表格. 主要特点: 自己主动 ...