python实现支持目录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程序设计有所帮助。