python下载整个ftp目录

过了今天就是国庆了,周围的小伙伴也都早早安排好,该回家的回家,该旅游的也已出发,这几天朋友圈基本都是晒照片的,据官方预测,估计未来几天会持续刷屏,其实看看照片还挺好,真到了那未必有照片好看,想想去年10.1长城人山人海拥挤的那个短视频,我就果断放弃了北京周边游的计划,过了10.1, 这2016也就快翻篇了,爱学习的小伙伴们要抓紧了,不过话说回来,学习呢还真是个持之以恒的事,这事着急不得,不求一天进度多少,但求每天进步一点足以,国庆倒数第二天,今天我还在公司上班,因为业务需求,就写了这个脚本,脚本完成的任务是从FTP上下载一个目录,大家都知道从FTP上下载一个文件可用用get命令,下载多个文件可以用mget,但你要下载一个目录,对不起,不能,如果非要较劲,你说用lftp,然后用mirror命令也可以呀,这个我不较真,因为每个公司有每个公司的场景,选择合适的就行,因为我们这的FTP都是用ssl的,所以lftp就不能用了,脚本基本思路是建立ftp连接,然后登陆获取文件列表,根据返回的列表进行for循环,一个一个下载,脚本如下:


#!/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

后面跟的参数基本就是研发提供给你要更新的文件名称,然后运行就可以下载整个目录中的文件,脚本其它内容大家看注释就会明白,国庆来临,我这也提前祝小伙伴们国庆快乐。




猜你喜欢

转载自blog.51cto.com/15060540/2651679