selenium搜狗搜图简单操作（爬取任意关键字的图片）

Picture preview

Steps

这里以搜狗搜图作为一个简单的例子：

第一步就是获取我们的需求了，输入需要下载的图片名字以及需要下载的图片数量。

接下来我们需要在搜狗搜图中输入用户的需求，透过$xpath$锁定搜索框，然后输入需求，进入图片浏览页面。

然后我们点击第一张图片，进入另一个窗口，这个时候我们代码中的窗口也需要跟着变换

b.switch_to.window(b.window_handles[1])

然后就可以开始我们的爬取了，同样的通过$xpath$得到我们的下载连接，再一张一张的下载到我们设定好的地址中。

为了方便，我设置了一个默认的地址，即$D:/ + name$的形式文件。

path = "D://" + name
    if not os.path.exists(path):
        os.mkdir(path)

Coding

#-*- coding = utf-8 -*-
#@Time : 2020/6/30 17:15
#@Author : lifehappy
#@File : 搜狗搜图selenium.py
#@Software : PyCharm

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time
import requests
import os

##记得要安装driver到chorme浏览器的相应目录下。
chromedriver_path = "C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe"

b = webdriver.Chrome(chromedriver_path)

def get_page(name):
    b.get("https://pic.sogou.com/")
    search_window = b.find_element_by_xpath(r'''//*[@id="form_querytext"]''')##get到输入窗口。
    search_window.send_keys(name)##在输入窗口输入我们要查找的关键词。
    search_window.send_keys(Keys.ENTER)##ENTER键，进入图片浏览页面。
    time.sleep(2)##设置休眠时间，玄学设定。

def start_spider(num, path):
    ##点击第一张图片获得它的xpath。
    new_window = b.find_element_by_xpath(r'''/html/body/div[2]/div[2]/ul/li[1]/div/a[1]/img''')
    new_window.click()
    b.switch_to.window(b.window_handles[1])##因为新开了一个窗口，我们一定要跳转窗口才能完成后面的操作。
    for i in range(num):
        print("正在下载第%d张" % (i + 1))
        image_element = b.find_element_by_xpath(r'''//*[@id="imageBox"]/img''')##得到图片连接的xpath
        link = image_element.get_attribute("src")
        link = requests.get(link)##转换类型，为了后面的下载。
        image_path = path + "//%d.jpg" % (i + 1)##图片名称设置为第几张。
        down_load(image_path, link)##下载
        next_page = b.find_element_by_xpath(r'''//*[@id="btnPgRgt"]/span''')##点击按钮到下一张图片。
        next_page.click()
        time.sleep(1)

def down_load(path, link):
    with open(path, 'wb') as f:
        f.write(link.content)
        time.sleep(1)
        f.close()

if __name__ == "__main__":
    name = input("输入下载图片名称：")
    num = int(input("输入下载图片数量："))
    path = "D://" + name
    if not os.path.exists(path):
        os.mkdir(path)
    get_page(name)
    start_spider(num, path)
    print("下载完成")