原文标题《python ftplib.FTP 获取当前路径下所有目录》
python内置库ftplib中,FTP 模块里有一个dir函数,可以打印出当前路径下所有文件,但是这个函数没有返回值,只是打印出来。
还有一个nlst函数,可以返回一个文件名的列表,但是只有文件名,没有详细信息,无法判断是否是目录。
目前我只有两个笨办法,
一个继承FTP类,自己实现一个getSubdir()方法,可以直接copy nlst()函数的源码,把cmd的字符串替换成“LIST”,再加上一个判断语句。另外就是自己写一个类,包含一个list成员和一个getSubdir方法,方法中将每个文件append到list中,该方法作为dir函数的参数传入。
类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)