过了明天就是国庆了，周围的小伙伴也都早早安排好，该回家的回家，该旅游的也已出发，这几天朋友圈基本都是晒照片的，据官方预测，估计未来几天会持续刷屏，其实看看照片还挺好，真到了那未必有照片好看，想想去年10.1长城人山人海拥挤的那个短视频，我就果断放弃了北京周边游的计划，过了10.1， 这2016也就快翻篇了，爱学习的小伙伴们要抓紧了，不过话说回来，学习呢还真是个持之以恒的事，这事着急不得，不求一天进度多少，但求每天进步一点足以，国庆倒数第二天，今天我还在公司上班，因为业务需求，就写了这个脚本，脚本完成的任务是从FTP上下载一个目录，大家都知道从FTP上下载一个文件可用用get命令，下载多个文件可以用mget，但你要下载一个目录，对不起，不能，如果非要较劲，你说用lftp，然后用mirror命令也可以呀，这个我不较真，因为每个公司有每个公司的场景，选择合适的就行，因为我们这的FTP都是用ssl的，所以lftp就不能用了，脚本基本思路是建立ftp连接，然后登陆获取文件列表，根据返回的列表进行for循环，一个一个下载，脚本如下：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
#!/usr/bin/evn python from ftplib import FTP_TLS, FTP import socket import ssl import os import sys class IMPLICIT_FTP_TLS(FTP_TLS): #构造函数初始化父类 def __init__(self, host='', user='', passwd='', acct='', keyfile=None, certfile=None, timeout=60): FTP_TLS.__init__(self, host, user, passwd, acct, keyfile, certfile, timeout) #建立服务器的FTPS连接 def connect(self, host='', port=0, timeout=-999): if host != '': self.host = host if port > 0: self.port = port if timeout != -999: self.timeout = timeout try: #创建socket self.sock = socket.create_connection((self.host, self.port), self.timeout) self.af = self.sock.family #wrap_socket接收一个socket实例,返回SSLSocket实例，可以理解在普通socket上封装了一层ssl self.sock = ssl.wrap_socket(self.sock, self.keyfile, self.certfile) self.file = self.sock.makefile('rb') self.welcome = self.getresp() except Exception as e: print (e) return self.welcome def get_ftp_ver(version): #如果没有这个目录就新建 if not os.path.isdir(version): os.makedirs(version) ftps = IMPLICIT_FTP_TLS() ftps.connect(host='10.0.0.8', port=666) ftps.login(user="ftp_user", passwd="ftp_password") #建立安全的数据连接，之后才能返回数据。 ftps.prot_p() ftps.cwd(version) files = ftps.nlst() #进入本地目录 os.chdir(version) #循环下载每个文件 for file in files: fp = open(file, 'wb') ftps.retrbinary('RETR %s' % file, fp.write) ftps.close() if __name__ == '__main__': get_ftp_ver(sys.argv[1]) |
脚本使用：
#python get_data.py version_20160920
后面跟的参数基本就是研发提供给你要更新的文件名称，然后运行就可以下载整个目录中的文件，脚本其它内容大家看注释就会明白，国庆来临，我这也提前祝小伙伴们国庆快乐。