最終實現一個快速下載照片的小程序。並且能實現根據你輸入的關鍵詞自動搜索下載。還可以指定保存目錄。
下面直接上代碼:
import requests
import re
from urllib import parse
import os
class BaiduImageSpider(object):
def __init__(self):
self.url = 'https://image.baidu.com/search/flip?tn=baiduimage&word={}'
self.headers = {'User-Agent': 'Mozilla/4.0'}
# 獲取圖片
def get_image(self, url, word):
# 使用 requests模塊得到響應對象
res = requests.get(url, headers=self.headers)
# 更改編碼格式
res.encoding = "utf-8"
# 得到html網頁
html = res.text
print(html)
# 正則解析
pattern = re.compile('"hoverURL":"(.*?)"', re.S)
img_link_list = pattern.findall(html)
# 存儲圖片的url鏈接
print(img_link_list)
# 創建目錄,用於保存圖片
directory = 'd:/image/{}/'.format(word)
# 如果目錄不存在則創建,此方法常用
if not os.path.exists(directory):
os.makedirs(directory)
# 添加計數
i = 1
for img_link in img_link_list:
filename = '{}{}_{}.jpg'.format(directory, word, i)
self.save_image(img_link, filename)
i += 1
# 下載圖片
def save_image(self, img_link, filename):
html = requests.get(url=img_link, headers=self.headers).content
with open(filename, 'wb') as f:
f.write(html)
print(filename, '下載成功')
# 入口函數
def run(self):
word = input("請問你想要下載什麼照片?")
word_parse = parse.quote(word)
url = self.url.format(word_parse)
self.get_image(url, word)
if __name__ == '__main__':
spider = BaiduImageSpider()
spider.run()
用百度是很方便的,但是百度是一個搜索引擎,所以你要什麼類型的圖片,就需要你輸入關鍵詞,因此程序第一步是提示你輸入關鍵詞,然後纔是按照你輸入的關鍵詞去查找你要的圖片。最後就是保存結果。
________________END______________