这篇文章主要介绍了python实现支持目录FTP上传下载文件的方法,适用于windows及Linux平台FTP传输文件及文件夹,需要的朋友可以参考下
本文实例讲述了python实现支持目录FTP上传下载文件的方法。分享给大家供大家参考。具体如下:
该程序支持ftp上传下载文件和目录、适用于windows和linux平台。
#!/usr/bin/env python # -*- coding: utf-8 -*- import ftplib import os import sys class FTPSync(object): conn = ftplib.FTP() def __init__(self,host,port=21): self.conn.connect(host,port) def login(self,username,password): self.conn.login(username,password) self.conn.set_pasv(False) print self.conn.welcome def test(self,ftp_path): print ftp_path print self._is_ftp_dir(ftp_path) #print self.conn.nlst(ftp_path) #self.conn.retrlines( 'LIST ./a/b') #ftp_parent_path = os.path.dirname(ftp_path) #ftp_dir_name = os.path.basename(ftp_path) #print ftp_parent_path #print ftp_dir_name def _is_ftp_file(self,ftp_path): try: if ftp_path in self.conn.nlst(os.path.dirname(ftp_path)): return True else: return False except ftplib.error_perm,e: return False def _ftp_list(self, line): list = line.split(' ') if self.ftp_dir_name==list[-1] and list[0].startswith('d'): self._is_dir = True def _is_ftp_dir(self,ftp_path): ftp_path = ftp_path.rstrip('/') ftp_parent_path = os.path.dirname(ftp_path) self.ftp_dir_name = os.path.basename(ftp_path) self._is_dir = False if ftp_path == '.' or ftp_path== './' or ftp_path=='': self._is_dir = True else: #this ues callback function ,that will change _is_dir value try: self.conn.retrlines('LIST %s' %ftp_parent_path,self._ftp_list) except ftplib.error_perm,e: return self._is_dir return self._is_dir def get_file(self,ftp_path,local_path='.'): ftp_path = ftp_path.rstrip('/') if self._is_ftp_file(ftp_path): file_name = os.path.basename(ftp_path) #如果本地路径是目录,下载文件到该目录 if os.path.isdir(local_path): file_handler = open(os.path.join(local_path,file_name), 'wb' ) self.conn.retrbinary("RETR %s" %(ftp_path), file_handler.write) file_handler.close() #如果本地路径不是目录,但上层目录存在,则按照本地路径的文件名作为下载的文件名称 elif os.path.isdir(os.path.dirname(local_path)): file_handler = open(local_path, 'wb' ) self.conn.retrbinary("RETR %s" %(ftp_path), file_handler.write) file_handler.close() #如果本地路径不是目录,且上层目录不存在,则退出 else: print 'EROOR:The dir:%s is not exist' %os.path.dirname(local_path) else: print 'EROOR:The ftp file:%s is not exist' %ftp_path def put_file(self,local_path,ftp_path='.'): ftp_path = ftp_path.rstrip('/') if os.path.isfile( local_path ): file_handler = open(local_path, "r") local_file_name = os.path.basename(local_path) #如果远程路径是个目录,则上传文件到这个目录,文件名不变 if self._is_ftp_dir(ftp_path): self.conn.storbinary('STOR %s'%os.path.join(ftp_path,local_file_name), file_handler) #如果远程路径的上层是个目录,则上传文件,文件名按照给定命名 elif self._is_ftp_dir(os.path.dirname(ftp_path)): print 'STOR %s'%ftp_path self.conn.storbinary('STOR %s'%ftp_path, file_handler) #如果远程路径不是目录,且上一层的目录也不存在,则提示给定远程路径错误 else: print 'EROOR:The ftp path:%s is error' %ftp_path file_handler.close() else: print 'ERROR:The file:%s is not exist' %local_path def get_dir(self,ftp_path,local_path='.',begin=True): ftp_path = ftp_path.rstrip('/') #当ftp目录存在时下载 if self._is_ftp_dir(ftp_path): #如果下载到本地当前目录下,并创建目录 #下载初始化:如果给定的本地路径不存在需要创建,同时将ftp的目录存放在给定的本地目录下。 #ftp目录下文件存放的路径为local_path=local_path+os.path.basename(ftp_path) #例如:将ftp文件夹a下载到本地的a/b目录下,则ftp的a目录下的文件将下载到本地的a/b/a目录下 if begin: if not os.path.isdir(local_path): os.makedirs(local_path) local_path=os.path.join(local_path,os.path.basename(ftp_path)) #如果本地目录不存在,则创建目录 if not os.path.isdir(local_path): os.makedirs(local_path) #进入ftp目录,开始递归查询 self.conn.cwd(ftp_path) ftp_files = self.conn.nlst() for file in ftp_files: local_file = os.path.join(local_path, file) #如果file ftp路径是目录则递归上传目录(不需要再进行初始化begin的标志修改为False) #如果file ftp路径是文件则直接上传文件 if self._is_ftp_dir(file): self.get_dir(file,local_file,False) else: self.get_file(file,local_file) #如果当前ftp目录文件已经遍历完毕返回上一层目录 self.conn.cwd( ".." ) return else: print 'ERROR:The dir:%s is not exist' %ftp_path return def put_dir(self,local_path,ftp_path='.',begin=True): ftp_path = ftp_path.rstrip('/') #当本地目录存在时上传 if os.path.isdir(local_path): #上传初始化:如果给定的ftp路径不存在需要创建,同时将本地的目录存放在给定的ftp目录下。 #本地目录下文件存放的路径为ftp_path=ftp_path+os.path.basename(local_path) #例如:将本地文件夹a上传到ftp的a/b目录下,则本地a目录下的文件将上传的ftp的a/b/a目录下 if begin: if not self._is_ftp_dir(ftp_path): self.conn.mkd(ftp_path) ftp_path=os.path.join(ftp_path,os.path.basename(local_path)) #如果ftp路径不是目录,则创建目录 if not self._is_ftp_dir(ftp_path): self.conn.mkd(ftp_path) #进入本地目录,开始递归查询 os.chdir(local_path) local_files = os.listdir('.') for file in local_files: #如果file本地路径是目录则递归上传目录(不需要再进行初始化begin的标志修改为False) #如果file本地路径是文件则直接上传文件 if os.path.isdir(file): ftp_path=os.path.join(ftp_path,file) self.put_dir(file,ftp_path,False) else: self.put_file(file,ftp_path) #如果当前本地目录文件已经遍历完毕返回上一层目录 os.chdir( ".." ) else: print 'ERROR:The dir:%s is not exist' %local_path return if __name__ == '__main__': ftp = FTPSync('192.168.1.110') ftp.login('test','test') #上传文件,不重命名 #ftp.put_file('111.txt','a/b') #上传文件,重命名 #ftp.put_file('111.txt','a/112.txt') #下载文件,不重命名 #ftp.get_file('/a/111.txt',r'D:\\') #下载文件,重命名 #ftp.get_file('/a/111.txt',r'D:\112.txt') #下载到已经存在的文件夹 #ftp.get_dir('a/b/c',r'D:\\a') #下载到不存在的文件夹 #ftp.get_dir('a/b/c',r'D:\\aa') #上传到已经存在的文件夹 ftp.put_dir('b','a') #上传到不存在的文件夹 ftp.put_dir('b','aa/B/')
希望本文所述对大家的Python程序设计有所帮助。
版权声明:
本站所有资源均为站长或网友整理自互联网或站长购买自互联网,站长无法分辨资源版权出自何处,所以不承担任何版权以及其他问题带来的法律责任,如有侵权或者其他问题请联系站长删除!站长QQ754403226 谢谢。