Python QT5 练习1 之 Baidu 图片下载

# 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)