python定時FTP同步文件

#coding=utf-8
'''
	ftp自動下載、自動上傳腳本,能夠遞歸目錄操做
'''

from ftplib import FTP
import os,sys,string,datetime,time
import socket

class MYFTP:
	def __init__(self,rootdir_local, hostaddr, username, password, remotedir, port=21):
		self.hostaddr = hostaddr
		self.username = username
		self.password = password
		self.remotedir  = remotedir
		self.port     = port
		self.ftp      = FTP()
		self.file_list = []

		self.rootdir_local = rootdir_local
		self.local_files = []
		self.remote_files = []
		self.appendFiles = []
		# self.ftp.set_debuglevel(2)
	def __del__(self):
		self.ftp.close()
		# self.ftp.set_debuglevel(0)
	def login(self):
		ftp = self.ftp
		try: 
			timeout = 300
			socket.setdefaulttimeout(timeout)
			ftp.set_pasv(True)
			print u'開始鏈接到 %s' %(self.hostaddr)
			ftp.connect(self.hostaddr, self.port)
			print u'成功鏈接到 %s' %(self.hostaddr)
			print u'開始登陸到 %s' %(self.hostaddr)
			ftp.login(self.username, self.password)
			print u'成功登陸到 %s' %(self.hostaddr)
			debug_print(ftp.getwelcome())
		except Exception:
			print u'鏈接或登陸失敗'
		try:
			ftp.cwd(self.remotedir)
		except(Exception):
			print u'切換目錄失敗'

	def margeFile(self):
		temp = []
		try:
			for f in self.remote_files:
				if f in self.local_files:
					temp.append(f)
			for f in temp:
				self.remote_files.remove(f)
		except Exception,e:
			print e

	def get_localFileList(self,fileTxt):
		print u'從 %s 中讀取本地文件列表'%(self.rootdir_local +'/'+fileTxt)
		try:
			fileTxt = open(rootdir_local+'/'+fileTxt,'r')
		except Exception,e:
			print e
		for line in fileTxt:
			line = line.strip('\n')
			line = line.strip(' ')
			self.local_files.append(line)

	def update_localFileList(self,fileTxt):
		print u'更新 %s 中文件列表'%(self.rootdir_local +'/'+fileTxt)
		try:
			fileTxt = open(rootdir_local+'/'+fileTxt,'a')
			fileTxt.writelines(['%s\n' %(x) for x in self.appendFiles])
		except Exception,e:
			print e

	def get_remoteFileList(self):
		print u'讀取服務器文件列表'
		try:
			self.remote_files = self.ftp.nlst()
		except Exception,e:
			print e

	def download_file(self, localfile, remotefile):
		debug_print(u'>>>>>>>>>>>>下載文件 %s <<<<<<<<<<<<' %localfile)
		#return
		file_handler = open(localfile, 'wb')
		self.ftp.retrbinary(u'RETR %s'%(remotefile), file_handler.write)
		file_handler.close()
		self.appendFiles.append(remotefile)

	def download_marge_files(self):
		for f in self.remote_files:
			self.download_file(rootdir_local+'/'+f,f)
'''
	#暫時不須要這些邏輯
	def download_files(self, localdir='./', remotedir='./'):
		try:
			self.ftp.cwd(remotedir)
		except:
			debug_print(u'目錄%s不存在,繼續...' %remotedir)
			return
		if not os.path.isdir(localdir):
			os.makedirs(localdir)
		debug_print(u'切換至目錄 %s' %self.ftp.pwd())
		self.file_list = []
		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('..')
		debug_print(u'返回上層目錄 %s' %self.ftp.pwd())

	def upload_file(self, localfile, remotefile):
		if not os.path.isfile(localfile):
			return
		if self.is_same_size(localfile, remotefile):
		 	debug_print(u'跳過[相等]: %s' %localfile)
		 	return
		file_handler = open(localfile, 'rb')
		self.ftp.storbinary('STOR %s' %remotefile, file_handler)
		file_handler.close()
		debug_print(u'已傳送: %s' %localfile)
	def upload_files(self, localdir='./', remotedir = './'):
		if not os.path.isdir(localdir):
			return
		localnames = os.listdir(localdir)
		self.ftp.cwd(remotedir)
		for item in localnames:
			src = os.path.join(localdir, item)
			if os.path.isdir(src):
				try:
					self.ftp.mkd(item)
				except:
					debug_print(u'目錄已存在 %s' %item)
				self.upload_files(src, item)
			else:
				self.upload_file(src, item)
		self.ftp.cwd('..')

	def get_file_list(self, line):
		ret_arr = []
		file_arr = self.get_filename(line)
		if file_arr[1] not in ['.', '..']:
			self.file_list.append(file_arr)
			
	def get_filename(self, line):
		pos = line.rfind(':')
		while(line[pos] != ' '):
			pos += 1
		while(line[pos] == ' '):
			pos += 1
		file_arr = [line[0], line[pos:]]
		return file_arr
'''		
def debug_print(s):
	print s

if __name__ == '__main__':
	timenow  = time.localtime()
	datenow  = time.strftime('%Y-%m-%d', timenow)
	# 配置以下變量
	hostaddr = '10.6.152.147' # ftp地址
	username = 'anonymous' # 用戶名
	password = 'anonymous@' # 密碼
	port  =  21   # 端口號 
	rootdir_local  = 'E:/mypiv' # 本地目錄
	rootdir_remote = '/'          # 遠程目錄
	local_file_list_txt = 'fileList.txt'
	while :
		f = MYFTP(rootdir_local,hostaddr, username, password, rootdir_remote, port)
		f.login()
		f.get_localFileList(local_file_list_txt)
		f.get_remoteFileList()
		f.margeFile()
		print f.remote_files
		f.download_marge_files()
		f.update_localFileList(local_file_list_txt)
		timenow  = time.localtime()
		datenow  = time.strftime('%Y-%m-%d %h:%M:%s', timenow)
		logstr = u"%s 成功執行了備份\n" %datenow
		debug_print(logstr)
相關文章
相關標籤/搜索