【PyQt4 实例22】实现ftp的上传和下载

# -*- coding: utf-8 -*-
#另外友情说明一下，如果要测试，可以用webpy框架搭建一个服务器。
from PyQt4.QtGui import *
from PyQt4.QtCore import *
import sys
from PyQt4.QtNetwork import *

QTextCodec.setCodecForTr(QTextCodec.codecForName("utf8"))

class FtpClient(QDialog):
def __init__(self,parent=None):
super(FtpClient,self).__init__(parent)
self.setWindowTitle(self.tr("实现Ftp上传和下载"))

vbMain = QVBoxLayout(self)

hb1 = QHBoxLayout()
LabelServer = QLabel(self)
LabelServer.setText(self.tr("服务器地址:"))
self.LineEditServer = QLineEdit(self)
hb1.addWidget(LabelServer)
hb1.addWidget(self.LineEditServer)

hb2 = QHBoxLayout()
UserName = QLabel(self)
UserName.setText(self.tr("用户名:"))
self.LineEditUser = QLineEdit(self)
hb2.addWidget(UserName)
hb2.addWidget(self.LineEditUser)

hb3 = QHBoxLayout()
PasswordName = QLabel(self)
PasswordName.setText(self.tr("密码:"))
self.LineEditPassword = QLineEdit(self)
hb3.addWidget(PasswordName)
hb3.addWidget(self.LineEditPassword)

hb4 = QHBoxLayout()

self.PushButtonLogin = QPushButton(self)
self.PushButtonLogin.setText(self.tr("登录"))
hb4.addWidget(self.PushButtonLogin)

hb5 = QHBoxLayout()

self.PushButtonPut = QPushButton(self)
self.PushButtonPut.setText(self.tr("上传"))

self.PushButtonGet = QPushButton(self)
self.PushButtonGet.setText(self.tr("下载"))

hb5.addWidget(self.PushButtonPut)
hb5.addWidget(self.PushButtonGet)

vbMain.addLayout(hb1)
vbMain.addLayout(hb2)
vbMain.addLayout(hb3)
vbMain.addLayout(hb4)
vbMain.addLayout(hb5)

self.connect(self.PushButtonLogin,SIGNAL("clicked()"),self.slotLogin)
self.connect(self.PushButtonPut,SIGNAL("clicked()"),self.slotPut)
self.connect(self.PushButtonGet,SIGNAL("clicked()"),self.slotGet)

self.curStatus = ""
self.localFile = QFile()
self.remoteFileName = QFile()

self.PushButtonPut.setEnabled(False)
self.PushButtonGet.setEnabled(False)

    def slotLogin(self):
        serverAddress = self.LineEditServer.text()
        if serverAddress.isEmpty():
            QMessageBox.warning(self,self.tr("error"),self.tr("Please input server address!"))
            return
        userName = self.LineEditUser.text()
        if userName.isEmpty():
            QMessageBox.warning(self,self.tr("error"),self.tr("Please input user name!"))
            return
        password = self.LineEditPassword.text()
        if password.isEmpty():
            QMessageBox.warning(self,self.tr("error"),self.tr("Please input Passwrod!"))
            return
        self.ftpClient = QFtp()
        self.ftpClient.connectToHost(serverAddress)
        
        self.connect(self.ftpClient,SIGNAL("stateChanged(int)"),self.slotStateChanged)
        self.connect(self.ftpClient,SIGNAL("done(bool)"),self.slotDone)
        
        self.ftpClient.login(userName,password)
        self.curStatus =  "LOGIN"
        
    
    def slotStateChanged(self,state):
        if state == QFtp.LoggedIn:
            self.PushButtonPut.setEnabled(True)
            self.PushButtonGet.setEnabled(True)
    
    def slotPut(self):
        OK = False
        fileName,OK = QInputDialog.getText(self,self.tr("Put File:"),self.tr("Please input file name:"),QLineEdit.Normal,QString())
        if OK and not fileName.isEmpty():
            self.remoteFileName = QFile(fileName)
            self.ftpClient.put(self.remoteFileName,fileName)
        self.curStatus = "PUT"
    
    def slotGet(self):
        OK = False
        fileName,OK = QInputDialog.getText(self,self.tr("Get File:"),self.tr("Please input file name:"),QLineEdit.Normal,QString())
        if OK and not fileName.isEmpty():
            self.localFile = QFile(fileName)
            self.localFile.open(QIODevice.WriteOnly)
            self.ftpClient.get(fileName,self.localFile)
        self.curStatus = "GET"
        
    
    def slotDone(self,error):
        if error:
            QMessageBox.warning(self,self.tr("error"),self.ftpClient.errorString())
            return
        if self.curStatus == "LOGIN":
            self.PushButtonLogin.setEnabled(False)
            self.curStatus = "None"
        if self.curStatus == "PUT":
            QMessageBox.warning(self,self.tr("succeed"),self.tr("Put file succeed!"))
            self.curStatus = "None"
        if self.curStatus == "GET":
            self.localFile.close()
            QMessageBox.warning(self,self.tr("succeed"),self.tr("Get file succeed!"))
            self.curStatus = "None"
        
        
app=QApplication(sys.argv)
dialog=FtpClient()
dialog.show()
app.exec_()