python31实例[文件夹清理]

时间：2011-01-11 来源：iTech

使用：

foldercleanup.py -d 10 -k c:\test\keepfile.txt c:\test

表示对c:\test目录只保留最近10天的子文件夹和keepfile.txt中指定的子文件夹。

代码：

import os
import os.path
import datetime

def getOption():
  from optparse import OptionParser

  des = "clean up the folder with some options"
  prog = "foldercleanup"
  ver = "%prog 0.0.1"
  usage = "%prog [options] foldername"

  p = OptionParser(description=des, prog=prog, version=ver, usage=usage,add_help_option=True)
  p.add_option('-d','--day',action='store',type='string',dest='day',help="keep the files in recent %day days")
  p.add_option('-k','--keepfile',action='store',type='string',dest='keepfile',help="keep the files showed in %keepfile")
  options, arguments = p.parse_args()

  if len(arguments) != 1:
    p.error("the number of arguments is not correct")
    return

  return options.day, options.keepfile, arguments[0]

def preCheckDir(dir):
  if(not os.path.exists(dir)):
    print("error: the directory your input is not existed")
    return
  if(not os.path.isdir(dir)):
    print ("error: the parameter your input is not a directory")
    return

  return os.path.abspath(dir)

def isKeep(dir, day, keepfile):
  inday = False
  if( day is not None) :
    t = os.path.getctime(dir)
    today = datetime.date.today()
    createdate = datetime.date.fromtimestamp(t)
    indate = today - datetime.timedelta(days = int(day))
    print (createdate)
    print (indate)
    if(createdate >= indate):
      inday = True
  print (inday)

  needkeep = False
  if (keepfile is not None):
    try :
      kf = open(keepfile,"r")
      for f in kf.readlines():
        if (dir.upper().endswith("\\" + f.strip().upper())):
          needkeep = True
      kf.close()
    except:
      print ("error: keep file cannot be opened")
  print(needkeep)

  if(inday or needkeep):
    return True
  else:
    return False

def removeSubFolders(dir, day, keepfile):
  subdirs = os.listdir(dir)
  for subdir in subdirs:
    subdir = os.path.join(dir,subdir)
    if (os.path.isdir(subdir)):
      print("----------------------")
      print(subdir)
      if(not isKeep(subdir, day, keepfile)):
        print(subdir)
        import shutil
        shutil.rmtree(subdir,True)

def FolderCleanUp():
  (day, keepfile, dir) = getOption()
  dir = preCheckDir(dir)
  if dir is None:
    return
  removeSubFolders(dir,day,keepfile)

if __name__=='__main__':
  FolderCleanUp()

完！