[Python知识库]使用selenium下载百度图片

简单说明

通过输入想要查找的关键词，使用selenium自动化工具下载其图片
网址为:‘https://image.baidu.com/search/index?tn=baiduimage&word=’+关键词

使用工具及模块

1. Pycharm-----python3.7
2. selenium:pip install selenium -i https://pypi.douban.com/simple
3. requests:pip install requests -i https://pypi.douban.com/simple
4. time
5. lxml: pip install lxml -i https://pypi.douban.com/simple

实现步骤

1.实例化webdriver
2.输入关键词,使用selenium打开百度图片页面
3.由于百度图片页面是通过不断下拉，进而动态加载图片出来，因此，想要让它维持一个持续下来的动作

• 首先获取初始页面的高度，以及初始时间，并将滚动条滑倒底部
• 其次，获取当前时间，并判断初始时间和当前时间的差是否大于30秒，小于则下拉，获得当前高度
• 若当前高度大于初始高度，重置页面高度为当前高度以及重置初始时间，重新计时
• 若当前高度小于初始高度，尝试重试3次
• 最后当重试次数大于3次时，跳出循环，认为页面加载完毕

4.解析页面内容，并获取图片链接
5.使用requests对图片链接发送请求，将图片保存到本地文件夹，完成下载

相关代码

from selenium import webdriver
from lxml import etree
import requests
import time

#### 爬取部分
def main(word):
   # 无界面模式
   options = webdriver.ChromeOptions()
   # #     无界面启动，也可以直接设置options.headless=True
   options.add_argument('headless')
   driver = webdriver.Chrome(options = options)
   driver.implicitly_wait(10)  # 设置隐式等待
   url = 'https://image.baidu.com/search/index?tn=baiduimage&word=' + word
   driver.get(url)
   driver.enconding = 'UTF-8'
   # 获取页面初始高度
   js = "return action=document.body.scrollHeight"
   height = driver.execute_script(js)
   # 将滚动条调整至页面底部
   driver.execute_script('window.scrollTo(0, document.body.scrollHeight)')
   time.sleep(3)
   # 定义初始时间戳（秒）
   t1 = int(time.time())
   # 重试次数
   # 定义循环标识，用于终止while循环
   status = True
   # 重试次数
   num = 0
   while status:
       # 获取当前时间戳（秒）
       t2 = int(time.time())
       # 判断时间初始时间戳和当前时间戳相差是否大于30秒，小于30秒则下拉滚动条
       if t2 - t1 < 30:
           new_height = driver.execute_script(js)
           if new_height > height:
               time.sleep(2)
               driver.execute_script('window.scrollTo(0, document.body.scrollHeight)')
               # 重置初始页面高度
               height = new_height
               # 重置初始时间戳，重新计时
               t1 = int(time.time())
       elif num < 3:  # 当超过30秒页面高度仍然没有更新时，进入重试逻辑，重试3次，每次等待30秒
           time.sleep(3)
           num = num + 1
       else:  # 超时并超过重试次数，程序结束跳出循环，并认为页面已经加载完毕！
           print("滚动条已经处于页面最下方！")
           status = False
           # 滚动条调整至页面顶部
           driver.execute_script('window.scrollTo(0, 0)')
           break
       content = driver.page_source.encode('utf-8')
       # 解析网页数据
       source = etree.HTML(content)
       # 获取图片链接
       pic_img = source.xpath('//div[@class="imgbox-border"]/a/img/@data-imgurl')
       header = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.82 Safari/537.36'}
       for i in range(len(pic_img)):
           pic = requests.get(pic_img[i], headers=header)
           # 获取图片保存的路径
           path ='./img'
           base_url = path + "\image{}.jpg".format((i + 1))
           with open(base_url, 'wb') as f:
               f.write(pic.content)


if __name__=='__main__':
   start = time.time()
   main('泰勒斯威夫特')
   end = time.time()
   print('下载图片需要%s秒' % (end - start))

结果截图

1. 运行结果截图

2. 下载图片完成

创作打卡挑战赛

赢取流量/现金/CSDN周边激励大奖