使用：

复制代码 代码如下:
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\’,\’–days\’,action=\’store\’,type=\’string\’,dest=\’days\’,help=\”keep the subfolders which are created in recent %days% days\”)
  p.add_option(\’-k\’,\’–keepfile\’,action=\’store\’,type=\’string\’,dest=\’keepfile\’,help=\”keep the subfolders which are recorded in text file %keepfile% \”)
  options, arguments = p.parse_args()
 
  if len(arguments) != 1:
    print(\”error: must input one directory as only one parameter \”)
    return
 
  return options.days, 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 isKeepByDay(dir, day):
  indays = 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)
    if(createdate >= indate):
      indays = True
  print (indays)
  return indays
 
def isKeepByKeepfile(dir, keepfile):
  needkeep = False
  print (dir)
  if (keepfile is not None):
    try :
      kf = open(keepfile,\”r\”)
      for f in kf.readlines():
        print (f)
        if (dir.upper().endswith(\”\\\\\” + f.strip().upper())):
          needkeep = True
      kf.close()
    except:
      print (\”error: keep file cannot be opened\”)
  print(needkeep)
  return needkeep
   
def removeSubFolders(dir, day, keepfile):
  subdirs = os.listdir(dir)
  for subdir in subdirs:
    subdir = os.path.join(dir,subdir)
    if ( not os.path.isdir(subdir)):
      continue
    print(\”———————-\”)
    if( (not isKeepByDay(subdir, day))and (not isKeepByKeepfile(subdir, keepfile))):
      print(\”remove subfolder: \” + 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()

对目录下保留最后的zip文件：

复制代码 代码如下:
def KeepLastNumZips(num)
    def extractTime(f):
        return os.path.getctime(f)

    zipfiles = [os.path.join(zipdir, f)
                for f in os.listdir(zipdir)
                if os.path.splitext(f)[1] == \”.zip\”]
    if len(zipfiles) > num:
        zipfiles.sort(key=extractTime, reverse=True)
        for i in range(num, len(zipfiles)):
            os.remove(zipfiles[i])