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

 

 

 

完!


作者:iTech
微信公众号: cicdops
出处:http://itech.cnblogs.com/
github:https://github.com/cicdops/cicdops

posted @ 2011-01-11 17:04  iTech  阅读(2061)  评论(0编辑  收藏  举报