前言
最近因為業務需求,就寫了這個腳本,腳本完成的任務是從FTP上下載一個目錄,大家都知道從FTP上下載一個文件可用用get命令,下載多個文件可以用mget,但你要下載一個目錄,對不起,不能,如果非要較勁,你說用lftp,然后用mirror命令也可以呀,這個我不較真,因為每個公司有每個公司的場景,選擇合適的就行,因為我們這的FTP都是用ssl的,所以lftp就不能用了,腳本基本思路是建立ftp連接,然后登陸獲取文件列表,根據返回的列表進行for循環,一個一個下載。
腳本如下:
#!/usr/bin/evn python from ftplib import FTP_TLS, FTPimport socketimport sslimport osimport 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
后面跟的參數基本就是研發提供給你要更新的文件名稱,然后運行就可以下載整個目錄中的文件,腳本其它內容大家看注釋就會明白。
以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作能帶來一定的幫助,如果有疑問大家可以留言交流。
新聞熱點
疑難解答