Python GUI开发之FTP客户端续

这一篇博客是继上一篇博客，想了解的可以直走右拐。

上一篇主要讲了客户端的界面，这一篇主要讲解FTP客户端的两个功能，文件下载，以及压缩下载。

文件下载

调用处

self.downbt = Button(self.right, text='下载', command=self.download)

定义download函数

def download(self):
    self.getrootpath()
    check = self.list.get(self.list.curselection())
    self.dpath = []
    self.downloadfiles(check, self.rootpath)
    self.f.cwd('/'.join(self.path))

---

1. self.getrootpath来获取下载的存放路径，具体获取代码如下：

def getrootpath(self):
    self.rootpath = os.path.join(os.path.curdir, 'download')
    if not os.path.exists(self.rootpath):
        os.mkdir(self.rootpath)

直接在当前目录下创建download文件夹，用来存放下载的文件。

2. check = self.list.get(self.list.curselection())用来获取选中的目录或文件。
3. self.dpath用来存储下载的目录的深度，主要为了下载的时候能够准确定位当前下载的准确位置
4. self.downloadfiles(check, self.rootpath)，具体的下载代码

def downloadfiles(self,dirs,rootpath):
     if dirs:
        try:
            self.f.cwd(dirs)
            self.dpath.append(dirs)
            newdir = os.path.join(rootpath,'/'.join(self.dpath))
            if not os.path.exists(newdir):
                os.mkdir(newdir)
            for dir in self.f.nlst():
                self.downloadfiles(dir,rootpath)
            self.dpath.pop()
            nowpath = '/'.join(self.path)+'/'+'/'.join(self.dpath)
            self.f.cwd(nowpath)
        except ftplib.error_perm:
            newfile = os.path.join(rootpath,'/'.join(self.dpath)+'/'+dirs)
            with open(newfile,'wb') as f:
                self.f.retrbinary('RETR %s' % (dirs), f.write)

下载代码也是很简单，进入到该目录(cwd)，这个时候如果是文件夹，就不会发生异常，如果不是文件夹，那么就可以认定是文件了
i. 先假定是文件夹，那么记录目录深度(self.dpath.append(dirs))，并在本地创建目录，然后递归调用下载代码，遍历该目录下的所有文件。这一层文件遍历完了之后，更新目录深度(self.dpath.pop())，这个时候需要ftp也要同时更新其所在目录，self.path在上一个博客已经说明，其代表用户进入目录，这里可以看做是用户最后一次点击进行下载的那个目录，加上下载的目录深度，也就是当前的目录了，举个例子

1. 用户在‘home/user’点击，那么我们的self.path记录的是['home','user']
2. 然后下载‘name/name1’中的文件，当前的dirs也就是name1，
这个时候我们的self.dpath记录的是['name','name1']
3. 当遍历完name1下的文件之后，我们需要pop self.dpath顶部的路径，
这个时候self.dpath存储的就是['name']
4. 最后我们所处的路径就是'home/user/name'，也就是self.path和self.dpath的连结。

ii. 如果是文件的话，那么就会被异常捕获，这个时候只需要通过retrbinary，将ftp的文件以二进制的形式写入到本地即可。
5.远程跳转到选择的文件位置即可

---

压缩下载

调用处

self.ziplist = Button(self.right, text='备份', command=self.zipdownload)

定义zipdownload函数

def zipdownload(self):
    self.getrootpath()
    check = self.list.get(self.list.curselection())
    rootpath = os.path.join(self.rootpath, 'temp')
    if not os.path.exists(rootpath):
        os.mkdir(rootpath)
    self.dpath = []
    self.downloadfiles(check, rootpath)
    zipname = os.listdir(rootpath)[0]
    temppath = os.path.join(rootpath, zipname)
    z = zipfile.ZipFile(os.path.join(self.rootpath, zipname + '.zip'), 'w', zipfile.ZIP_DEFLATED)
    for dirpath, dirnames, filenames in os.walk(temppath):
        for file in filenames:
            z.write(os.path.join(dirpath, file))
    z.close()
    self.rmtemp(rootpath)
    self.f.cwd('/'.join(self.path))

---

1. 创建temp文件夹用来临时存放下载文件
2. zipname = os.listdir(rootpath)[0] 获取下载文件夹的名字，后面压缩的文件名即这个名字
3. 创建压缩包，通过zipfile.ZipFile进行创建，然后遍历temp中的文件，os.walk会找到目录下的所有文件，将每个文件写入到压缩文件中即可
4. 删除temp临时目录

def rmtemp(self,temppath):
     for root, dirs, files in os.walk(temppath, topdown=False):
        for name in files:
            os.remove(os.path.join(root, name))
        for name in dirs:
            os.rmdir(os.path.join(root, name))
     os.rmdir(temppath)

通过os.walk遍历temp临时文件夹，files中存放的是文件，dirs存放的是文件夹，先删除文件，可以避免删除文件夹失败，最后删除该临时文件夹，就完成了删除。
5.远程跳转到选择的文件位置即可

上述过程是否看懂了呢，FTP客户端就先到这里了。