python ftp批量上傳文件下載文件

# encoding:utf-8

from ftplib import FTP
import os
import sys

_XFER_FILE = 'FILE'
_XFER_DIR = 'DIR'

class FtpClient(object):
def __init__(self):
self.ftp = None

def __del__(self):
pass

def setFtpParams(self, ip, uname, pwd, port=21, timeout=60):
self.ip = ip
self.uname = uname
self.pwd = pwd
self.port = port
self.timeout = timeout

def initEnv(self):
if self.ftp is None:
self.ftp = FTP()
print '### connect ftp server: %s ...' % self.ip
self.ftp.connect(self.ip, self.port, self.timeout)
self.ftp.login(self.uname, self.pwd)
print self.ftp.getwelcome()
self.ftp.set_pasv(False)

def clearEnv(self):
if self.ftp:
self.ftp.close()
print '### disconnect ftp server: %s!' % self.ip
self.ftp = None

def uploadDir(self, localDir='./', remoteDir='./'):
if not os.path.isdir(localDir):
return
self.ftp.cwd(remoteDir)
for file in os.listdir(localDir):
source = os.path.join(localDir, file)
if os.path.isfile(source):
self.uploadFile(source, file)
elif os.path.isdir(source):
try:
self.ftp.mkd(file)
except:
sys.stderr.write('The dir is exists %s' % file)
self.uploadDir(source, file)
self.ftp.cwd('..')

def uploadFile(self, localPath, remotePath='./'):
if not os.path.isfile(localPath):
return
print (' upload %s to %s' % (localPath, remotePath))
self.ftp.storbinary('STOR ' + remotePath, open(localPath, 'rb'))

def __filetype(self, src):
if os.path.isfile(src):
index = src.rfind('/')
# if the src not contains /
if index == -1:
index = src.rfind('/')
return _XFER_FILE, src[index + 1:]
elif os.path.isdir(src):
return _XFER_DIR, ''

def upload(self, lacaldir, remoteDir="./"):
filetype, filename = self.__filetype(lacaldir)

self.initEnv()
if filetype == _XFER_DIR:
self.srcDir = lacaldir
self.uploadDir(self.srcDir, remoteDir)
elif filetype == _XFER_FILE:
self.uploadFile(lacaldir, filename)
self.clearEnv()

def download_file(self, localfile, remotefile):

self.debug_print('>>>>>>>>>>>>下載文件 %s ... ...' % localfile)
# return
with open(localfile, 'wb') as f:
self.ftp.retrbinary('RETR %s' % (remotefile), f.write)

def download_files(self, localDir='./', remoteDir='./'):
"""
從遠程目錄下，下載全部的文件並建立對應的目錄
:param localdir: 本地文件夾
:param remotedir: 遠程服務文件夾
:return:
"""
try:
self.ftp.cwd(remoteDir)
print self.ftp.pwd()
except:
self.debug_print(u'目錄%s不存在，繼續...' % remoteDir)
return
if not os.path.isdir(localDir):
os.makedirs(localDir)
self.debug_print(u'切換至目錄 %s' % self.ftp.pwd())
self.file_list = []
#
# self.ftp.dir(self,*args) List a directory in long form
# 最後一個參數爲回調函數，前面的不爲空的參數，會跟list 組合成command
# 默認列出當前目錄下的內容
#
self.ftp.dir(self.get_file_list)
remotenames = self.file_list
# print(remotenames)
# return
for item in remotenames:
filetype = item[0]
filename = item[1]
local = os.path.join(localdir, filename)
if filetype == 'd':
self.download_files(local, filename)
elif filetype == '-':
self.download_file(local, filename)
self.ftp.cwd('..')
self.debug_print('返回上層目錄 %s' % self.ftp.pwd())

def get_file_list(self, line):
"""

:param lines: ftp.dir() provide the line
:return: return the file list ,that get file name from line
"""
# ret_arr = []
file_arr = self.get_filename(line)
if file_arr[1] not in ['.', '..']:
self.file_list.append(file_arr)

def debug_print(self, s):
print s

def get_filename(self, line):
"""
line maybe :-rw-r--r-- 1 501 501 320294 Aug 24 05:08 1.d2dc00a98a6eee589f2e.chunk.js
:param line:
:return: return the first word of line and get the file name.
"""
pos = line.rfind(':')
while (line[pos] != ' '):
pos += 1
while (line[pos] == ' '):
pos += 1
file_arr = [line[0], line[pos:]]
return file_arr

if __name__ == '__main__': localdir = r"d:\mysql\www" # remoteDir = r"/" xfer = FtpClient() xfer.setFtpParams('', 'tny', '') xfer.initEnv() xfer.uploadDir(localDir=localdir)