欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  后端开发

python实现的文件夹清理程序分享

程序员文章站 2022-04-25 17:06:49
...
使用:

代码如下:


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