import requests
import re
import os

headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.88 Safari/537.36'
}

# 1. 构建搜索URL
keyword = input("请输入图片关键词: ")
url = f'http://image.baidu.com/search/flip?tn=baiduimage&ps=1&ct=201326592&lm=-1&cl=2&nc=1&ie=utf-8&word={keyword}'

# 2. 获取页面源码
r = requests.get(url, headers=headers)
ret = r.text

# 3. 正则提取图片URL
result = re.findall('"objURL":"(.*?)"', ret)

# 4. 创建保存目录
if not os.path.exists('img'):
    os.makedirs('img')

# 5. 下载图片
for i, img_url in enumerate(result):
    # 获取图片文件名
    name = img_url[-20:].split('/')[-1] or f'image_{i}.jpg'
    
    try:
        r = requests.get(img_url, timeout=10)
        with open(f'img/{name}', 'wb') as f:
            f.write(r.content)
        print(f"下载成功: {name}")
    except Exception as e:
        print(f"下载失败: {e}")

print(f"共下载 {len(result)} 张图片")