python 实现远端ftp文件上传下载

python 实现ftp上传下载

* 脚本需要传入两个参数，参数1为需要从远端ftp站点下载文件名称，参数2为已知需要下载的文件md5值，文件下载完成后会自动进行md5值校验

* 运行示例

[root@vm172-31-16-11 software]# python file.txt 2d6c30dba93eda3f70f662a390d8e49c

# -*- coding: utf-8 -*-

from ftplib import FTP
import time
import tarfile
import os
import sys
import ftplib
import hashlib

filename = sys.argv[1]
localpath = '/data/update/'
remotepath = '/server'
def ftpconnect(host, username, password):
ftp = FTP()
#    ftp.set_debuglevel(2)
ftp.connect(host, 21)
ftp.login(username, password)
print ftp.welcome
return ftp
def downloadfile(ftp, filename):
if  os.path.exists(localpath):
pass
else:
os.makedirs(localpath, 0755)
bufsize = 1024
ftp.cwd(remotepath)
ftp.dir()
fp = open(localpath + filename,'wb')
ftp.retrbinary('RETR %s' % os.path.basename(filename), fp.write, bufsize)
fp.close()

def getFileMD5():
filepath = localpath + filename
if os.path.exists(filepath):
f = open(filepath,'rb')
md5obj = hashlib.md5()
md5obj.update(f.read())
hash = md5obj.hexdigest()
f.close()
return str(hash).upper()
else:
print "\033[1;31;40mERROR:Get the %s MD5 failed\033[0m" % filepath
return None

def file_exists():
filemd5 = getFileMD5()
downloadftpfile = localpath + filename
if os.path.exists(downloadftpfile) and filemd5 == sys.argv[2]:
print "\033[1;32;40m====================ftp down ok and md5 value is ok======================\033[0m"
elif os.path.exists(downloadftpfile) and filemd5 != sys.argv[2]:
print "\033[1;31;40m====================ftp down ok and md5 value is bad======================\033[0m"
else:
print "\033[1;31;40mERROR:The %s is not exists\033[0m" % downloadftpfile

def uploalfile(ftp, remotepath, localpath):
bufsize = 1024
fp = open(localpath, 'rb')
ftp.storbinary('STOR ' + remotepath, fp, bufsize)
#    ftp.set_debuglevel(0)
ftp.close()

def download():
try:
ftp = ftpconnect("x.x.x.x", "x.x", "x.x")
downloadfile(ftp, filename)
ftp.quit()
file_exists()
except ftplib.error_perm:
os.chdir(localpath)
os.remove(filename)
file_exists()
if __name__ == "__main__":
download()