使用 PyQt5 設計下載遠端伺服器紀錄檔檔案程式的思路

2022-11-13 14:00:58

前期準備

軟體

QT5
Python 模組PyQt5==5.15.7paramiko==2.9.2
PyCharm 新增擴充套件工具 PyUIC

PyUIC 擴充套件用於將使用 Qt Designer 生成的 ui 檔案轉成 py 檔案，可以在 PyCharm 中通過 Preferences-Tools-External Tools 進行設定，截圖如下：

Program：/Users/macbookpro/workspace/projects/DownloadServerLog/venv/bin/python3.9 Arguments：-m PyQt5.uic.pyuic $FileName$ -o $FileNameWithoutExtension$.py Working directory：/Users/macbookpro/workspace/projects/DownloadServerLog/ui

實操步驟

1. 建立專案

建立 DownloadServerLog 專案，設計程式結構如下：

DownloadServerLog
├── app
│   ├── downloadlog.py
│   └── downloadlog_qtui.py
├── main.py
└── ui
│   └── downloadlog_qtui.ui
├── .env

main.py 作為程式入口檔案，.env 存放環境變數，ui 存放使用 Qt Designer 設計介面匯出的原始碼檔案，app 存放下載程式檔案。

2.使用 QtDesigner 設計介面

Qt Designer 使用起來非常簡單，可以通過“拖拉拽”的形式生成 UI 介面（檔案：https://doc.qt.io/qtcreator/creator-using-qt-designer.html），設計介面如下：

這個程式功能一目瞭然，左側幾個輸入框用於輸入必要的資訊，右側一個展示框用於展示程式實時紀錄檔。介面設計好後可以將其儲存至專案 DownloadServerLog 下的 ui 目錄下 downloadlog_qtui.ui，供後續使用。

3. 使用 ui 生成對應的 py 檔案

使用 PyCharm 開啟專案，在 downloadlog_qtui.ui 檔案上右鍵，選擇 External Tools 使用 PyUIC 根據 ui 檔案生成對應的 py 檔案 downloadlog_qtui.py，將檔案存放至 app 目錄。

4. 新建 main.py 作為程式入口

在專案根目錄下建立 main.py 檔案：

import sys

from PyQt5 import QtCore
from PyQt5.QtCore import QObject, pyqtSignal
from PyQt5.QtWidgets import QApplication, QMainWindow
from threading import Thread

from app.downloadlog_qtui import Ui_Dialog
from app.downloadlog import DownloadLog


class CommunicateSignal(QObject):
    text_print = pyqtSignal(str)


# MyWindow 是主視窗程式，繼承自 PyQt5.QtWidgets.QMainWindow
# 和通過 ui 檔案生成的 downloadlog_qtui.py 中的 Ui_Dialog 類
class MyWindow(QMainWindow, Ui_Dialog):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setupUi(self)
        self.btn_download.clicked.connect(self.click_download)

        # 自定義訊號處理常式
        self.comm_signal = CommunicateSignal()
        self.comm_signal.text_print.connect(self.show_text)

        self.set_window_init_data()

    def set_window_init_data(self):
        """設定程式表單初始值"""
        # 從 .env 讀取環境變數
        result_dict = dict()
        with open('.env', 'r', encoding='utf-8') as f:
            for line in f.readlines():
                key = line.split('=')[0].strip()
                value = line.split('=')[-1].strip()
                result_dict[key] = value

        # 設定輸入框值
        _translate = QtCore.QCoreApplication.translate
        self.host.setText(_translate("Dialog", result_dict.get("HOST", '')))
        self.port.setText(_translate("Dialog", result_dict.get("PORT", '22')))
        self.username.setText(_translate("Dialog", result_dict.get("USERNAME", 'root')))
        self.password.setText(_translate("Dialog", result_dict.get("PASSWORD", '')))
        self.directory.setPlainText(_translate("Dialog", result_dict.get("DIRECTORY", '')))
        self.startTime.setDate(QtCore.QDate.currentDate())
        self.endTime.setDate(QtCore.QDate.currentDate())


    def get_window_input_value(self):
        """獲取程式各「輸入框」元件值"""
        return {
            "host": self.host.text(),
            "port": self.port.text(),
            "username": self.username.text(),
            "password": self.password.text(),
            "directory": self.directory.toPlainText(),
            "start_time": self.startTime.date().toString("yyyy-MM-dd"),
            "end_time": self.endTime.date().toString("yyyy-MM-dd"),
            "suffix": ".log",
        }

    def show_text(self, text):
        """將文字內容追加到程式「展示框」"""
        self.textBrowser.append(text)

    def click_download(self):
        """處理點選「下載」按鈕事件"""
        params = self.get_window_input_value()

        def run():
            res = DownloadLog(conn_type='ssh', comm_signal=self.comm_signal, **params)
            res.main()

        t = Thread(target=run)
        t.start()
if __name__ == '__main__':
    app = QApplication(sys.argv)
    myWin = MyWindow()
    myWin.show()
    sys.exit(app.exec_())

MyWindow 作為主視窗程式，程式初始化時會將 self.click_download 方法註冊到下載按鈕的點選事件，並自動呼叫 self.set_window_init_data 方法來設定輸入框初始值。

5. 下載

下載紀錄檔程式 DownloadLog 定義在 app/downloadlog.py 中，遠端下載檔案主要步驟有兩步：

通過 SSH 登入遠端伺服器
通過 FTP 進行檔案下載

這裡採用 paramiko 來實現遠端下載功能，paramiko 是一個純 Python 庫，它實現了 SSHv2 協定，提供了 SSH 和 FTP 的能力。

核心程式碼如下，讀者可以根據自己的需求實現 DownloadLog：

class DownloadLog(object):
    def __init__(self, **kwargs):
        """初始化一些引數"""
        ...

    def main(self):
        # 獲取 Transport 範例
        tran = paramiko.Transport((self.host, int(self.port)))
        # 連線 SSH 伺服器端
        tran.connect(username=self.username, password=self.password)
        # 建立 SFTP 範例
        self.sftp = paramiko.SFTPClient.from_transport(tran)
        # 下載檔案
        # :param str remotepath: the remote file to copy
        # :param str localpath: the destination path on the local host
        self.sftp.get(remotepath=self.remote_path, localpath=self.local_path)

6. 展示下載過程

為了將下載程式執行步驟實時展示到輸出框，這裡需要引入 PyQt5 的訊號處理機制。

由於 PyQt 建議只在主執行緒中操作介面，可以發現我們在 main.py 中呼叫 DownloadLog.main 方法時建立了一個新的執行緒。

所有的 GUI 程式都是事件驅動的，事件可能由使用者觸發，比如點選下載按鈕事件，也可能由程式觸發，比如我們現在要實現的展示下載過程的功能，就需要使用程式主動觸發事件。

在 PyQt5 中通過 Signal 訊號來處理事件，其基本使用步驟如下：

自定義一個 CommunicateSignal 類，繼承自 PyQt5 的 QObject 類，裡面封裝自定義的 Signal 訊號（Signal 範例物件的初始化引數指定的型別，就是發出訊號物件時，傳遞的引數資料型別。因為 PyQt5 底層是 C++ 開發的，必須指定型別）。

class CommunicateSignal(QObject):
    text_print = pyqtSignal(str)

定義主執行緒執行的函數處理 Signal 訊號（通過 connect 方法系結）。

# 自定義訊號處理常式
self.comm_signal = CommunicateSignal()
self.comm_signal.text_print.connect(self.show_text)

在 DownloadLog 執行緒需要操作介面的時候，就通過自定義物件（CommunicateSignal）發出訊號（使用 emit 方法發出訊號)，所以在範例化 DownloadLog 時會將 comm_signal 傳遞進去。

# 通過該訊號物件的 emit 方法發出訊號，emit 方法的引數傳遞必要的資料。
# 引數型別遵循定義 Signal 時指定的型別。
self.comm_signal.text_print.emit(text)

主執行緒訊號處理常式，被觸發執行，獲取 Signal 裡面的引數，執行必要的更新介面操作，這裡將每次通過事件傳過來的文字內容展示到輸出框內。

def show_text(self, text):
    """將文字內容追加到程式「展示框」"""
    self.textBrowser.append(text)

7. 效果展示

通過以上步驟我們完成的程式設計，現在可以驗證下這個下載紀錄檔檔案的小程式了：

檢視下載結果：

總結

我們通過 PyQt5 實現了一個下載遠端伺服器紀錄檔檔案的小程式，其實它不止可以用來下載紀錄檔，同樣可以用來下載其他檔案。
藉助 PyQt5 強大的能力，我們可以通過“拖拉拽”的形式很容易地實現桌面端程式，只需要將原來的 Python 指令碼繫結到 UI 程式的事件中，就實現了命令列程式到桌面程式的演進。
接下來你可以根據自己的需求來客製化自己的桌面小程式啦~