Python自動化之實現桌面桌布下載器

pip install requests -i https://pypi.tuna.tsinghua.edu.cn/simple/

pip install PyQt5 -i https://pypi.tuna.tsinghua.edu.cn/simple/

# It's importing the traceback module.
import traceback

# Importing all the classes from the QtWidgets module.
from PyQt5.QtWidgets import *

# Importing all the classes from the QtCore module.
from PyQt5.QtCore import *

# Importing all the classes from the QtGui module.
from PyQt5.QtGui import *

# It's importing the requests module.
import requests

# It's importing the os module.
import os

# It's importing the sys module.
import sys

class WallPaperUI(QWidget):
    def __init__(self):
        """
        A constructor. It is called when an object is created from a class and it allows the class to initialize the
        attributes of a class.
        """
        super(WallPaperUI, self).__init__()
        self.init_ui()

    def init_ui(self):
        """
        This function initializes the UI.
        """
        self.setWindowTitle('批次桌面桌布下載器  公眾號：Python 集中營')
        self.setWindowIcon(QIcon('download.ico'))
        self.resize(500, 200)

        self.save_dir_path = QLineEdit()
        self.save_dir_path.setPlaceholderText('桌面桌布儲存路徑')

        self.save_dir_btn = QPushButton()
        self.save_dir_btn.setText('儲存路徑')
        self.save_dir_btn.clicked.connect(self.save_dir_btn_click)

        self.set_dwonload_num_label = QLabel()
        self.set_dwonload_num_label.setText('設定下載數量：')

        self.set_dwonload_num_in = QLineEdit()
        self.set_dwonload_num_in.setPlaceholderText('例如：10')

        self.start_btn = QPushButton()
        self.start_btn.setText('立即下載')
        self.start_btn.clicked.connect(self.start_btn_click)

        self.brower = QTextBrowser()
        self.brower.setReadOnly(True)
        self.brower.setFont(QFont('宋體', 8))
        self.brower.setPlaceholderText('處理程序展示區域...')
        self.brower.ensureCursorVisible()

        hbox = QHBoxLayout()

        left_box = QVBoxLayout()
        left_box.addWidget(self.brower)

        right_box = QVBoxLayout()

        right_form_box = QFormLayout()
        right_form_box.addRow(self.save_dir_path, self.save_dir_btn)
        right_form_box.addRow(self.set_dwonload_num_label, self.set_dwonload_num_in)

        right_h_box = QHBoxLayout()
        right_h_box.addWidget(self.start_btn)

        right_box.addLayout(right_form_box)
        right_box.addLayout(right_h_box)

        hbox.addLayout(left_box)
        hbox.addLayout(right_box)

        self.download_thread = DownloadWork(self)
        self.download_thread.finished.connect(self.finished)
        self.download_thread.message.connect(self.show_message)

        self.setLayout(hbox)

    def show_message(self, text):
        cursor = self.brower.textCursor()
        cursor.movePosition(QTextCursor.End)
        self.brower.append(text)
        self.brower.setTextCursor(cursor)
        self.brower.ensureCursorVisible()

    def save_dir_btn_click(self):
        directory = QFileDialog.getExistingDirectory(self, '選擇資料夾', os.getcwd())
        self.save_dir_path.setText(directory)

    def start_btn_click(self):
        self.start_btn.setEnabled(False)
        self.download_thread.start()

    def finished(self, finished):
        if finished is True:
            self.start_btn.setEnabled(True)

class DownloadWork(QThread):
    finished = pyqtSignal(bool)
    message = pyqtSignal(str)

    def __init__(self, parent=None):
        super(DownloadWork, self).__init__(parent)
        self.working = True
        self.parent = parent

    def __del__(self):
        self.working = False

    def run(self) -> None:
        try:
            save_dir_path = self.parent.save_dir_path.text().strip()
            set_dwonload_num_in = self.parent.set_dwonload_num_in.text().strip()

            self.message.emit('儲存路徑：{}'.format(save_dir_path))
            self.message.emit('下載數量：{}'.format(set_dwonload_num_in))

            if save_dir_path == '':
                self.message.emit('出現錯誤：儲存路徑不能為空！')
                return

            if set_dwonload_num_in == '':
                self.message.emit('出現錯誤：下載數量不能為空！')
                return

            for n in range(int(set_dwonload_num_in)):
                pic_url = f"http://bingw.jasonzeng.dev?resolution=UHD&index={n}"
                self.message.emit('正在下載第{0}張桌布！'.format(str(n)))
                with requests.get(pic_url) as r:
                    with open(os.path.join(save_dir_path, str(n) + '.jpg'), "wb") as w:
                        w.write(r.content)
            self.message.emit('全部桌布下載完成！')
            self.finished.emit(True)
        except:
            traceback.print_exc()
            self.message.emit('執行錯誤，請檢查引數項是否設定正確！')
            self.finished.emit(True)

# A special variable in Python that evaluates to True if the module is being run as the main program.
if __name__ == '__main__':
    app = QApplication(sys.argv)
    main = WallPaperUI()
    main.show()
    sys.exit(app.exec_())