FTPprogram/remote_transfer.py at master · ShehanAT/FTPprogram · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from PyQt5.QtWidgets import (QMessageBox, QApplication)
from pysftp import paramiko
import pysftp, os, logging
from PyQt5.QtCore import Qt

logger = logging.getLogger("FTP-program")
def remoteToLocalTransfer(self, remoteFile):
    self.notificationLabel.setText("Loading...")
    QApplication.setOverrideCursor(Qt.WaitCursor)
    if remoteFile == []:
        errorMessage = QMessageBox(QMessageBox.Critical, "Error", "No file selected to file transfer. Make sure you double-click the file to transfer!")
        errorMessage.exec_()
        return
    try:
        if self.currentLocalPath[-1] == "/":
            size = len(self.currentLocalPath)
            self.currentLocalPath = self.currentLocalPath[:size - 1]
        remoteFileName = remoteFile.text(0)
        if self.currentLocalPath[-1] != "\\":
            self.currentLocalPath += "\\"
        newLocalFileName = remoteFileName
        with self.connection.cd(self.currentRemotePath):
            localPathDest = self.currentLocalPath + newLocalFileName
            self.connection.get(remoteFileName, localPathDest)
            updateLocalFiles(self)
            QApplication.restoreOverrideCursor()
            showFileTransferSuccessMsg(self)
    except IsADirectoryError:
        QApplication.restoreOverrideCursor()
        errorMessage = QMessageBox(QMessageBox.Critical, "Error", "The selected file is a directory, please select a file instead.")
        errorMessage.exec_()
    except PermissionError:
        QApplication.restoreOverrideCursor()
        errorMessage = QMessageBox(QMessageBox.Critical, "Error", "Permission Denied for file transfer")
        errorMessage.exec_()
    except FileNotFoundError:
        QApplication.restoreOverrideCursor()
        errorMessage = QMessageBox(QMessageBox.Critical, "Error", "No such file")
        errorMessage.exec_()
    except OSError:
        QApplication.restoreOverrideCursor()
        tracebackString = traceback.print_exc()
        errorMessage = QMessageBox(QMessageBox.Critical, "Error", "The selected file is a directory, please select a file instead.")
        errorMessage.exec_()
    except Exception as e:
        QApplication.restoreOverrideCursor()
        logger.error(e)
        errorMessage = QMessageBox(QMessageBox.Critical, "Error", str(e))
        errorMessage.exec_()
    self.notificationLabel.setText("")

def updateLocalFiles(self):
    self.LocalFilesList.clear()
    localFiles = os.scandir(self.currentLocalPath)
    self.createLocalFilesList(localFiles)

def showFileTransferSuccessMsg(self):
    transfer_success_msg = QMessageBox()
    transfer_success_msg.setWindowTitle("File Transferred")
    transfer_success_msg.setText("File transferred successfully!")
    transfer_success_msg.setIcon(QMessageBox.Information)
    transfer_success_msg.exec_()