王锋

王锋的博客

他的个人主页  他的博客

ftp的应用

王锋  2012年07月06日 星期五 15:19 | 1952次浏览 | 0条评论

python ftp

#!/usr/bin/python  
    #coding=gbk  
    ''''' 
        ftp自动下载、自动上传脚本,可以递归目录操作 
    '''  
      
    from ftplib import FTP  
    import os,sys,string,datetime,time  
    import socket  
      
    class MYFTP:  
        def __init__(self, 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.ftp.set_debuglevel(2)  
        def __del__(self):  
            self.ftp.close()  
            # self.ftp.set_debuglevel(0)  
        def login(self):  
            ftp = self.ftp  
            try:   
                timeout = 60  
                socket.setdefaulttimeout(timeout)  
                ftp.set_pasv(True)  
                print '开始连接到 %s' %(self.hostaddr)  
                ftp.connect(self.hostaddr, self.port)  
                print '成功连接到 %s' %(self.hostaddr)  
                print '开始登录到 %s' %(self.hostaddr)  
                ftp.login(self.username, self.password)  
                print '成功登录到 %s' %(self.hostaddr)  
                debug_print(ftp.getwelcome())  
            except Exception:  
                deal_error("连接或登录失败")  
            try:  
                ftp.cwd(self.remotedir)  
            except(Exception):  
                deal_error('切换目录失败')  
      
        def is_same_size(self, localfile, remotefile):  
            try:  
                remotefile_size = self.ftp.size(remotefile)  
            except:  
                remotefile_size = -1  
            try:  
                localfile_size = os.path.getsize(localfile)  
            except:  
                localfile_size = -1  
            debug_print('lo:%d  re:%d' %(localfile_size, remotefile_size),)  
            if remotefile_size == localfile_size:  
                return 1  
            else:  
                return 0  
        def download_file(self, localfile, remotefile):  
            if self.is_same_size(localfile, remotefile):  
                debug_print('%s 文件大小相同,无需下载' %localfile)  
                return  
            else:  
                debug_print('>>>>>>>>>>>>下载文件 %s ... ...' %localfile)  
            #return  
            file_handler = open(localfile, 'wb')  
            self.ftp.retrbinary('RETR %s'%(remotefile), file_handler.write)  
            file_handler.close()  
      
        def download_files(self, localdir='./', remotedir='./'):  
            try:  
                self.ftp.cwd(remotedir)  
            except:  
                debug_print('目录%s不存在,继续...' %remotedir)  
                return  
            if not os.path.isdir(localdir):  
                os.makedirs(localdir)  
            debug_print('切换至目录 %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('返回上层目录 %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('跳过[相等]: %s' %localfile)  
                return  
            file_handler = open(localfile, 'rb')  
            self.ftp.storbinary('STOR %s' %remotefile, file_handler)  
            file_handler.close()  
            debug_print('已传送: %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('目录已存在 %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)  
    def deal_error(e):  
        timenow  = time.localtime()  
        datenow  = time.strftime('%Y-%m-%d', timenow)  
        logstr = '%s 发生错误: %s' %(datenow, e)  
        debug_print(logstr)  
        file.write(logstr)  
        sys.exit()  
      
    if __name__ == '__main__':  
        file = open("log.txt", "a")  
        timenow  = time.localtime()  
        datenow  = time.strftime('%Y-%m-%d', timenow)  
        logstr = datenow  
        # 配置如下变量  
        hostaddr = 'localhost' # ftp地址  
        username = 'test' # 用户名  
        password = 'test' # 密码  
        port  =  21   # 端口号   
        rootdir_local  = '.' + os.sep + 'bak/' # 本地目录  
        rootdir_remote = './'          # 远程目录  
          
        f = MYFTP(hostaddr, username, password, rootdir_remote, port)  
        f.login()  
        f.download_files(rootdir_local, rootdir_remote)  
          
        timenow  = time.localtime()  
        datenow  = time.strftime('%Y-%m-%d', timenow)  
        logstr += " - %s 成功执行了备份\n" %datenow  
        debug_print(logstr)  
          
        file.write(logstr)  
        file.close()  

评论

我的评论:

发表评论

请 登录 后发表评论。还没有在Zeuux哲思注册吗?现在 注册 !

暂时没有评论

Zeuux © 2022

京ICP备05028076号