使用：

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

 

 

 

完！