python爬虫---自定义爬取百度图片项目

import requests
import re
import os

if __name__ == '__main__':
    # 先导包os，创建一个文件夹，保存所有的图片
    str_name = input('你想要什么图片')
    if not os.path.exists('./' + str_name):
        os.mkdir('./' + str_name)
    url = 'https://image.baidu.com/search/index?tn=baiduimage&ps=1&ct=201326592&lm=-1&cl=2&nc=1&ie=utf-8&word=' + str_name
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.67 Safari/537.36 Edg/87.0.623.47'
    }
    # 使用通用爬虫对url对应的一整张页面进行爬取
    page_text = requests.get(url=url, headers=headers).text
    # 使用聚焦将页面中所有的图片进行解析/提取
    ex = '{"thumbURL":"(.*?)"'
    img_src_list = re.findall(ex, page_text, re.S)
    for src in img_src_list:
        # content是请求到图片的二进制数据
        img_data = requests.get(url=src, headers=headers).content
        # 生成图片名称
        post_name = src.split('/')[-1]
        # 图片存储的路径
        imgpath = './' + str_name + '/'+ str_name  + post_name
        with open(imgpath, 'wb') as fp:
            fp.write(img_data)
            print(post_name, '文件下载成功')