python实例31[文件夹清理]

 

使用：

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])

 

 

 

完！