欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2024-01-05 10:14:10
...

原文标题《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)

上一篇:

下一篇: