Python编程：获取ftp目录下文件夹和文件

原文标题《python ftplib.FTP 获取当前路径下所有目录》

python内置库ftplib中，FTP 模块里有一个dir函数，可以打印出当前路径下所有文件，但是这个函数没有返回值，只是打印出来。

还有一个nlst函数，可以返回一个文件名的列表，但是只有文件名，没有详细信息，无法判断是否是目录。

目前我只有两个笨办法，
一个继承FTP类，自己实现一个getSubdir()方法，可以直接copy nlst()函数的源码，把cmd的字符串替换成“LIST”，再加上一个判断语句。

另外就是自己写一个类，包含一个list成员和一个getSubdir方法，方法中将每个文件append到list中，该方法作为dir函数的参数传入。

参考：
http://www.cnblogs.com/vawter/p/5987432.html
https://www.cnblogs.com/hltswd/p/6228992.html

类MyFTP就是按照上述所说第一种方法，继承。修改了源代码，将使用变得简单了，可以单独存为一个文件，导入使用，当然参考上面第2个链接就可以实现上传下载功能，这里不再赘述。

# myftp.py

from ftplib import FTP

class MyFTP(FTP):
    encoding = "gbk"  # 默认编码
    def getSubdir(self, *args):
        '''拷贝了 nlst() 和 dir() 代码修改，返回详细信息而不打印'''
        cmd = 'LIST'
        func = None
        if args[-1:] and type(args[-1]) != type(''):
            args, func = args[:-1], args[-1]
        for arg in args:
            cmd = cmd + (' ' + arg)
        files = []
        self.retrlines(cmd, files.append)
        return files

    def getdirs(self, dirname=None):
        """返回目录列表，包括文件简要信息"""
        if dirname != None:
            self.cwd(dirname)
        files = self.getSubdir()

        # 处理返回结果，只需要目录名称
        r_files = [file.split(" ")[-1] for file in files]

        # 去除. .. 
        return [file for file in r_files if file != "." and file !=".."]

    def getfiles(self, dirname=None):
        """返回文件列表，简要信息"""
        if dirname != None:
            self.cwd(dirname)  # 设置FTP当前操作的路径
        return self.nlst()  # 获取目录下的文件 

    # 这个感觉有点乱，后面再说,
    # def getalldirs(self, dirname=None):
    #     """返回文件列表，获取整个ftp所有文件夹和文件名称简要信息"""
    #     if dirname != None:
    #         self.cwd(dirname)  # 设置FTP当前操作的路径
    #     files = []
    #     dirs = set(self.getdirs()) - set(self.getfiles())
    #     if dirs != {}:
    #         for name in dirs:
    #             self.cwd("..")  # 返回上级
    #             files += self.getalldirs(name)
    #     return files

def test():
    ftp = MyFTP()  # 实例化
    ftp.connect("ip", port)  # 连接
    ftp.login("username", "password")  # 登录

    # 获取第一层目录下的文件
    # lst =ftp.getdirs()
    # print(lst)
    # for name in lst:
    #     ftp.cwd("..")  # 返回上级
    #     names = ftp.getdirs(name)
    #     print(names)

    lst = ftp.getdirs()  # 返回目录下文件夹和文件列表
    print(lst)
    ftp.quit()  # 退出

if __name__ == '__main__':
    test()

当然，使用的时候可以新建一个类，封装起来，这样便于使用

from myftp import MyFTP

class FtpData(object):
    def __init__(self):
        self.ftp = MyFTP()
        self.ftp.connect("ip", port)
        self.ftp.login("username", "password")

    def getfiles(self, dirname):
        lst =self.ftp.getdirs(dirname)
        return lst

    def __del__(self):
        self.ftp.quit()

if __name__ == '__main__':
    ftpdata = FtpData()
    lst = ftpdata.getfiles("dirname")
    print(lst)