# Time: 2020/05/12
#Author: Xiaohong
# 運行環境: OS: Windows 7
# Python: 3.7
# 功能: 根據關鍵詞 key 到baidu 下載 pic_number 張的圖片
# https://www.jianshu.com/p/33bee611fa7d
# https://www.cnblogs.com/SeekHit/p/6284974.html
界面及思路基本同 本人前幾天寫的 Python 練習之Baidu 圖片下載,差異爲:本次是利用QT5 作爲界面開發工具,之前是使用Tkinter
界面如下:
主程序部分爲:兩個函數 def baidu_dl_pic01(self):、def sel_directory(self) 是QT5寫法
import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import *
from PyQt5.QtWidgets import QFileDialog, QMessageBox, QDockWidget, QListWidget
from PyQt5.QtGui import *
# from Ui_Qt5_baidu_pic1 import Ui_Frm_baidu_pic1 # 導入創建的GUI類
from Ui_Qt5_baidu_pic1 import Ui_Frm_baidu_pic1
from Tl_baidu_DownPic02 import Tl_baidu_DownPic02
# 自己建一個mywindows類,Tl_baidu_DownPic02 是自己的類名。QtWidgets.QMainWindow:繼承該類方法
class mywindow(QtWidgets.QMainWindow, Ui_Frm_baidu_pic1):
# __init__:析構函數,也就是類被創建後就會預先加載的項目。
# 馬上運行,這個方法可以用來對你的對象做一些你希望的初始化。
def __init__(self):
# 這裏需要重載一下mywindow,同時也包含了QtWidgets.QMainWindow的預加載項。
super(mywindow, self).__init__()
self.setupUi(self)
self.btn_select.clicked.connect(self.sel_directory)
self.btn_download.clicked.connect(self.baidu_dl_pic01)
def baidu_dl_pic01(self):
pass
t_keyword = self.t_baidu_Keyword.text()
t_directory = self.t_directory.text()
t_pic_count = self.t_pic_count.text()
print(t_keyword)
print(t_directory)
print(t_pic_count)
if t_keyword == '' or t_directory == '' or t_pic_count == '':
reply = QMessageBox.information(
self, '提醒', '參數爲空,請檢查並補錄完整', QMessageBox.Yes, QMessageBox.Yes)
else:
reply = QMessageBox.question(
self, '確認', '是否要從Baidu下載圖片,這將花費較長時間', QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
if reply == QMessageBox.Yes:
print('下載')
Tl_baidu_DownPic02(t_keyword,t_directory,int(t_pic_count))
else:
print('不下載')
def sel_directory(self):
pass
# 打開文件有以下3種:
# 1、單個文件打開 QFileDialog.getOpenFileName()
# 2、多個文件打開 QFileDialog.getOpenFileNames()
# 3、打開文件夾 QFileDialog.getExistingDirectory()
dir_path = QFileDialog.getExistingDirectory(self, "請選擇文件夾路徑", "e:\\")
self.t_directory.setText(dir_path)
print(dir_path)
if __name__ == '__main__': # 如果整個程序是主程序
# QApplication相當於main函數,也就是整個程序(很多文件)的主入口函數。
# 對於GUI程序必須至少有一個這樣的實例來讓程序運行。
app = QtWidgets.QApplication(sys.argv)
# 生成 mywindow 類的實例。
window = mywindow()
# 有了實例,就得讓它顯示,show()是QWidget的方法,用於顯示窗口。
window.show()
# 調用sys庫的exit退出方法,條件是app.exec_(),也就是整個窗口關閉。
# 有時候退出程序後,sys.exit(app.exec_())會報錯,改用app.exec_()就沒事
# https://stackoverflow.com/questions/25719524/difference-between-sys-exitapp-exec-and-app-exec
sys.exit(app.exec_())
Ui_Qt5_baidu_pic1.py 是通過QT5 Designer 可視化工具開發後,轉化而來
# -*- coding: utf-8 -*-
# Form implementation generated from reading ui file 'e:\vscode_2020\vstestfxh\fxh_qt501\fxh_qt501\Qt5_baidu_pic1.ui'
#
# Created by: PyQt5 UI code generator 5.13.0
#
# WARNING! All changes made in this file will be lost!
from PyQt5 import QtCore, QtGui, QtWidgets
class Ui_Frm_baidu_pic1(object):
def setupUi(self, Frm_baidu_pic1):
Frm_baidu_pic1.setObjectName("Frm_baidu_pic1")
Frm_baidu_pic1.resize(655, 388)
self.frame = QtWidgets.QFrame(Frm_baidu_pic1)
self.frame.setGeometry(QtCore.QRect(20, 10, 551, 131))
self.frame.setFrameShape(QtWidgets.QFrame.StyledPanel)
self.frame.setFrameShadow(QtWidgets.QFrame.Raised)
self.frame.setObjectName("frame")
self.label = QtWidgets.QLabel(self.frame)
self.label.setGeometry(QtCore.QRect(30, 20, 71, 16))
self.label.setObjectName("label")
self.label_2 = QtWidgets.QLabel(self.frame)
self.label_2.setGeometry(QtCore.QRect(30, 60, 71, 16))
self.label_2.setObjectName("label_2")
self.t_directory = QtWidgets.QLineEdit(self.frame)
self.t_directory.setGeometry(QtCore.QRect(110, 60, 371, 20))
self.t_directory.setObjectName("t_directory")
self.t_baidu_Keyword = QtWidgets.QLineEdit(self.frame)
self.t_baidu_Keyword.setGeometry(QtCore.QRect(110, 20, 371, 20))
self.t_baidu_Keyword.setObjectName("t_baidu_Keyword")
self.btn_select = QtWidgets.QPushButton(self.frame)
self.btn_select.setGeometry(QtCore.QRect(480, 60, 31, 23))
self.btn_select.setObjectName("btn_select")
self.label_3 = QtWidgets.QLabel(self.frame)
self.label_3.setGeometry(QtCore.QRect(30, 100, 54, 12))
self.label_3.setObjectName("label_3")
self.t_pic_count = QtWidgets.QLineEdit(self.frame)
self.t_pic_count.setGeometry(QtCore.QRect(110, 100, 113, 20))
self.t_pic_count.setObjectName("t_pic_count")
self.btn_download = QtWidgets.QPushButton(self.frame)
self.btn_download.setGeometry(QtCore.QRect(240, 100, 75, 23))
self.btn_download.setObjectName("btn_download")
self.btn_Close = QtWidgets.QPushButton(self.frame)
self.btn_Close.setGeometry(QtCore.QRect(320, 100, 75, 23))
self.btn_Close.setObjectName("btn_Close")
self.retranslateUi(Frm_baidu_pic1)
self.btn_Close.clicked.connect(Frm_baidu_pic1.close)
# self.btn_select.clicked.connect(Frm_baidu_pic1.sel_directory)
QtCore.QMetaObject.connectSlotsByName(Frm_baidu_pic1)
def retranslateUi(self, Frm_baidu_pic1):
_translate = QtCore.QCoreApplication.translate
Frm_baidu_pic1.setWindowTitle(_translate("Frm_baidu_pic1", "Baidu 圖片下載"))
self.label.setText(_translate("Frm_baidu_pic1", "圖片 關鍵詞"))
self.label_2.setText(_translate("Frm_baidu_pic1", "文件夾路徑"))
self.btn_select.setText(_translate("Frm_baidu_pic1", "..."))
self.label_3.setText(_translate("Frm_baidu_pic1", "圖片數量"))
self.btn_download.setText(_translate("Frm_baidu_pic1", "下載"))
self.btn_Close.setText(_translate("Frm_baidu_pic1", "退出"))
Tl_baidu_DownPic02 寫法如下:
# -*- coding: utf-8 -*-
# Time: 2020/05/08
#Author: Xiaohong
# 運行環境: OS: Windows 7
# Python: 3.7
# 功能: 根據關鍵詞 key 到baidu 下載 pic_number 張的圖片
# https://www.jianshu.com/p/33bee611fa7d
# https://www.cnblogs.com/SeekHit/p/6284974.html
import json
import itertools
import urllib
import requests
import os
import re
import sys
import time
# 設立函數,來取得當前時間,作爲文件名的一部分,以免文件名重複
def get_sysdate():
now = time.strftime("%m%d-%H%M%S", time.localtime(time.time()))
print(now)
return now
str_table = {
'_z2C$q': ':',
'_z&e3B': '.',
'AzdH3F': '/'
}
char_table = {
'w': 'a',
'k': 'b',
'v': 'c',
'1': 'd',
'j': 'e',
'u': 'f',
'2': 'g',
'i': 'h',
't': 'i',
'3': 'j',
'h': 'k',
's': 'l',
'4': 'm',
'g': 'n',
'5': 'o',
'r': 'p',
'q': 'q',
'6': 'r',
'f': 's',
'p': 't',
'7': 'u',
'e': 'v',
'o': 'w',
'8': '1',
'd': '2',
'n': '3',
'9': '4',
'c': '5',
'm': '6',
'0': '7',
'b': '8',
'l': '9',
'a': '0'
}
# str 的translate方法需要用單個字符的十進制unicode編碼作爲key
# value 中的數字會被當成十進制unicode編碼轉換成字符
# 也可以直接用字符串作爲value
char_table = {ord(key): ord(value) for key, value in char_table.items()}
# 解碼圖片URL
def decode(url):
# 先替換字符串
for key, value in str_table.items():
url = url.replace(key, value)
# 再替換剩下的字符
return url.translate(char_table)
# 生成網址列表
def buildUrls(word):
word = urllib.parse.quote(word) # 屏蔽特殊的字符、比如如果url裏面的空格!url裏面是不允許出現空格的
url = r"http://image.baidu.com/search/acjson?tn=resultjson_com&ipn=rj&ct=201326592&fp=result&queryWord={word}&cl=2&lm=-1&ie=utf-8&oe=utf-8&st=-1&ic=0&word={word}&face=0&istype=2nc=1&pn={pn}&rn=60"
urls = (url.format(word=word, pn=x)
for x in itertools.count(start=0, step=60)) # itertools是python內置的模塊,count是生成無界限序列,從0開始,步長爲60,無限循環
return urls
# 解析JSON獲取圖片URL,
# 一個圖片對應許多鏈接,經測試圖片鏈接在 objURL 這個標籤內。用正則找出
# 括號表示分組,將括號的內容捕獲到分組當中
# 這個括號也就可以匹配網頁中圖片的url了
re_url = re.compile(r'"objURL":"(.*?)"') # 根據包含正則表達式的字符串創建模式對象
def resolveImgUrl(html):
imgUrls = [decode(x) for x in re_url.findall(html)]
# print(imgUrls)
return imgUrls
# for x in re_url.findall(html):
# print('aaa:')
# print(x)
# imgUrls=decode(x)
# print('bbb:')
# print(imgUrls)
# return imgUrls
def downImg(imgUrl, dirpath, imgName):
filename = os.path.join(dirpath, imgName)
try:
res = requests.get(imgUrl, timeout=15)
if str(res.status_code)[0] == "4":
# print(str(res.status_code), ":", imgUrl)
return False
except Exception as e:
print("拋出異常:", imgUrl)
print(e)
return False
with open(filename, "wb") as f:
f.write(res.content)
return True
def mkDir(dirName):
dirpath = os.path.join(sys.path[0], dirName)
if not os.path.exists(dirpath):
os.mkdir(dirpath)
return dirpath
def Tl_baidu_DownPic02(keyword, download_dir, pic_number):
word = keyword
dir2 = os.path.join(download_dir, word, get_sysdate())
print(dir2)
if not os.path.exists(dir2):
os.makedirs(dir2)
# dirpath = mkDir("results")
dirpath = dir2
urls = buildUrls(word)
index = 0
for url in urls:
# print("正在請求:", url)
html = requests.get(url, timeout=10).content.decode(
'utf-8') # 變量html 保留了一份Json文件,其中 objURL 是 圖片對應在網址
# print("正在請求:", html)
imgUrls = resolveImgUrl(html)
if len(imgUrls) == 0 or index >= pic_number: # 沒有圖片則結束
break
for url in imgUrls:
if downImg(url, dirpath, str(index) + ".jpg"):
index += 1
print("已下載 %s 張" % index)
if index >= pic_number:
break
if __name__ == '__main__':
# print("歡迎使用百度圖片下載腳本!\n目前僅支持單個關鍵詞。")
# print("下載結果保存在腳本目錄下的results文件夾中。")
# print("=" * 50)
# word = input("請輸入你要下載的圖片關鍵詞:\n")
Tl_baidu_DownPic02('Raspberry', 'e://', 50)
def call_DownPic02_main(keyword, download_dir, pic_number):
# print("歡迎使用百度圖片下載腳本!\n目前僅支持單個關鍵詞。")
# print("下載結果保存在腳本目錄下的results文件夾中。")
# print("=" * 50)
# word = input("請輸入你要下載的圖片關鍵詞:\n")
# Tl_baidu_DownPic02('Raspberry',50)
Tl_baidu_DownPic02(keyword, download_dir, pic_number)