[python] PySide6로 올린 gui로 원격 서버의 파일 다운로드하기

개요

이번에 진행한 프로젝트해서 2가지를 개발해야했다.

 

1. 원격 서버에 있는 파일을 로컬에 다운로드해야한다. (파일이 커서 오래 걸리니, progress bar로 진행 사항을 보여주자)

2. PySide6로 gui를 올린다.

 

1️⃣ 파일 다운로드

import paramiko
import time

def download_file(hostname, port, username, password, remote_file_path, local_file_path, progress_callback):
    try:
        start = time.perf_counter()
        
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(hostname, port, username, password)

        sftp = ssh.open_sftp()
        
        remote_file_size = sftp.stat(remote_file_path).st_size
        
        with sftp.open(remote_file_path, "rb") as remote_file, open(local_file_path, "wb") as local_file:
            downloaded_size = 0
            chunk_size = 8192
            
            while True:
                data = remote_file.read(chunk_size)
                if not data:
                    break
                local_file.write(data)
                downloaded_size += len(data)
                
                # 진행률 계산 및 콜백 호출
                progress = int((downloaded_size / remote_file_size) * 100)
                progress_callback(progress)

        print(f"File downloaded successfully from {remote_file_path} to {local_file_path}")

        sftp.close()
        ssh.close()
        
        end = time.perf_counter()
        print(f"model 다운로드 시간: {end - start:.6f} 초")

    except Exception as e:
        print(f"An error occurred: {e}")

 

- paramiko 라이브러리를 이용해서 ssh 접속 정보를 입력해 원격 서버에 있는 파일에 접근하는 방식으로 진행했다.

- 다운로드 해야하는 파일의 크기와 로컬에 다운로드 받아지는 파일의 크기를 계산해서 진행률을 계산했다.

 

2️⃣ PySide6으로 GUI 올리기

from PySide6.QtWidgets import (
    QApplication, QVBoxLayout, QWidget, QProgressBar, QPushButton, QLabel, QDialog, QLineEdit, QFormLayout, QMessageBox
)
from PySide6.QtCore import QThread, Signal
from download_file_pyside6 import download_file


class DownloadThread(QThread):
    progress = Signal(int)
    completed = Signal()

    def __init__(self, hostname, port, username, password, remote_file_path, local_file_path):
        super().__init__()
        self.hostname = hostname
        self.port = port
        self.username = username
        self.password = password
        self.remote_file_path = remote_file_path
        self.local_file_path = local_file_path

    def run(self):
        download_file(
            hostname=self.hostname,
            port=self.port,
            username=self.username,
            password=self.password,
            remote_file_path=self.remote_file_path,
            local_file_path=self.local_file_path,
            progress_callback=self.progress.emit
        )
        self.completed.emit()


class DownloadDialog(QDialog):
    def __init__(self, parent, hostname, port, username, password, remote_file_path, local_file_path):
        super().__init__(parent)
        self.setWindowTitle("Downloading File")
        self.resize(400, 150)
        
        self.layout = QVBoxLayout(self)

        # 상태 레이블
        self.status_label = QLabel("Preparing to download...", self)
        self.layout.addWidget(self.status_label)

        # 프로그래스바
        self.progress_bar = QProgressBar(self)
        self.progress_bar.setValue(0)
        self.progress_bar.setMaximum(100)
        self.layout.addWidget(self.progress_bar)

        # 닫기 버튼
        self.close_button = QPushButton("Close", self)
        self.close_button.setEnabled(False)
        self.close_button.clicked.connect(self.close)
        self.layout.addWidget(self.close_button)

        # 다운로드 스레드 설정
        self.download_thread = DownloadThread(
            hostname, port, username, password, remote_file_path, local_file_path
        )
        self.download_thread.progress.connect(self.update_progress)
        self.download_thread.completed.connect(self.download_completed)

        # 다운로드 시작
        self.download_thread.start()

    def update_progress(self, value):
        self.progress_bar.setValue(value)
        self.status_label.setText(f"Downloading... {value}%")

    def download_completed(self):
        self.status_label.setText("Download complete!")
        self.close_button.setEnabled(True)
        
    def closeEvent(self, event):
        if self.download_completed:
            event.accept()
        else:
            confirm = QMessageBox.question(
                self,
                "Confirm Close",
                "Are you sure you want to stop downloading the model?",
                QMessageBox.Yes | QMessageBox.No,
            )
            if confirm == QMessageBox.Yes:
                event.accept()
                os.remove(local_file_path)
            else:
                event.ignore()

class InputDialog(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("Enter Download Details")
        self.resize(400, 300)
        
        self.layout = QFormLayout(self)

        # 입력 필드 생성
        self.hostname_input = QLineEdit(self)
        self.port_input = QLineEdit(self)
        self.port_input.setText("5022")  # 기본값
        self.username_input = QLineEdit(self)
        self.password_input = QLineEdit(self)
        self.password_input.setEchoMode(QLineEdit.Password)
        self.remote_file_input = QLineEdit(self)
        self.local_file_input = QLineEdit(self)
        
        # 필드 레이아웃 추가
        self.layout.addRow("Hostname:", self.hostname_input)
        self.layout.addRow("Port:", self.port_input)
        self.layout.addRow("Username:", self.username_input)
        self.layout.addRow("Password:", self.password_input)
        self.layout.addRow("Remote File Path:", self.remote_file_input)
        self.layout.addRow("Local File Path:", self.local_file_input)

        # 확인 및 취소 버튼
        self.ok_button = QPushButton("OK", self)
        self.ok_button.clicked.connect(self.accept)
        self.layout.addRow(self.ok_button)

        self.setLayout(self.layout)

    def get_inputs(self):
        return {
            "hostname": self.hostname_input.text(),
            "port": int(self.port_input.text()),
            "username": self.username_input.text(),
            "password": self.password_input.text(),
            "remote_file_path": self.remote_file_input.text(),
            "local_file_path": self.local_file_input.text(),
        }


class MainWindow(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("Main Window")
        self.resize(400, 200)
        
        self.layout = QVBoxLayout(self)

        # 다운로드 버튼
        self.download_button = QPushButton("Download File", self)
        self.download_button.clicked.connect(self.start_download_dialog)
        self.layout.addWidget(self.download_button)

    def start_download_dialog(self):
        # 입력 대화 상자 열기
        input_dialog = InputDialog(self)
        if input_dialog.exec():  # 사용자가 확인 버튼을 눌렀는지 확인
            inputs = input_dialog.get_inputs()
            dialog = DownloadDialog(
                self,
                hostname=inputs["hostname"],
                port=inputs["port"],
                username=inputs["username"],
                password=inputs["password"],
                remote_file_path=inputs["remote_file_path"],
                local_file_path=inputs["local_file_path"],
            )
            dialog.exec()


if __name__ == "__main__":
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec()

 

1. MainWindow를 첫 페이지로 show한다.

2. Download File 버튼을 클릭하면 InputDialog 가 실행되며 ssh 접속 정보를 입력할 수 있는 Dialog가 띄어진다.

 

default 값 (placeholder)로 값을 넣을 수 있고, 입력하는 값을 password로 암호화해서 보여줄 수도 있다.

 

3. 정보를 입력하고 OK를 클릭하면 DownloadDialog가 실행된다

 

다운로드 중에는 Close 버튼이 비활성화되고, 다운로드가 완료되면 Close 버튼이 활성화된다.

다운로드 진행 중에 Downloading File dialog를 닫으려고 한다면, "Are you sure you want to stop downloading the model?" 을 묻는다. Yes → 다운로드는 중지하고, 다운로드 중이었던 파일을 삭제한다. No → 다운로드는 계속 이어간다.