用Python写了个日志清理的脚本

欢迎吐槽！

#! /usr/bin/env python

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

import os,time,shutil

path = os.getcwd()

target_dir = []

cal = []

target_dir.append('/var/lib/mysql/log/')

target_dir.append('/newlogs/')

target_dir.append('/oldlogs/')

#初始化删除文件名

del_ext = ['frm','MYD','MYI']

#你要保留几天

days_before = 5

#初始化删除日期

for x in xrange(days_before+1,days_before+30):

    cal.append(time.strftime('%Y_%m_%d',time.localtime(time.time() - x*24*3600)))

#初始化删除文件名，使用笛卡尔积！

fl = map('.'.join, [[x,y] for x in cal for y in del_ext])

#以日期命名的文件夹

cal = []

for x in xrange(days_before+1,days_before+30):

    cal.append(time.strftime('%Y-%m-%d',time.localtime(time.time() - x*24*3600)))

fl += cal

#获取文件扩展名

def get_ext(filename):

    return filename.split('.')[-1]

for dirs in target_dir:

    #files = [i for i in box_list if(get_ext(i) in del_ext)]  #按扩展名筛选文件

    #修改当前工作目录

    i = 0

    print 'opening -> ' + dirs

    #判断文件夹存在

    if os.path.exists(dirs):

        #切换目录

        os.chdir(dirs)

        for files in fl:

            #判断文件夹或文件

            if(os.path.isfile(str(files))):

                i += 1

                try:

                    os.remove(files)

                    print files + '  deleted'

                except:

                    print 'error occurs while deleting file'

            elif(os.path.isdir(str(files))):

                i += 1

                try:

                #shutil.rmtree删除非空文件夹

                    shutil.rmtree(files)

                    print files + '  deleted'

                except:

                    print 'error occurs while deleting file'

    else:

        print 'Error on opening ' + dirs

    if i:

        print str(i) + ' files deleted'

    else:

        print 'nothing to remove'

    print 'leaving -> ' + dirs

    print '\n'