我有一个程序将所有文件放在不同的文件夹中,例如这些文件:
20120809_1204567_cat_pic1.jpg
20120810_1204567_cat_pic1.jpg
20120811_1204567_cat_pic1.jpg
该程序以第一个数字的名称创建 3 个不同的文件夹和一个名为 cat 的子文件夹。现在我想压缩这些文件夹,但在启动程序之前我不知道每个文件夹的名称。创建这些文件夹后如何立即压缩文件夹?
import zipfile
import glob
import datetime
import os
from collections import defaultdict
from shutil import copyfile
src = 'D:/Testing/src/'
for name in glob.glob('D:/Testing/src/*'):
print name
dict_date = defaultdict(lambda : defaultdict(list))
for fil in os.listdir(src):
if os.path.isfile(os.path.join(src, fil)):
date, animal = fil.split('_')[0], fil.split('_')[2]
dict_date[date][animal].append(fil)
for date in dict_date:
for animal in dict_date[date]:
try:
os.makedirs(os.path.join(src, date, animal))
except os.error:
pass
for fil in dict_date[date][animal]:
copyfile(os.path.join(src, fil), os.path.join(src, date, animal, fil))
directory = src
os.chdir(directory)
files = glob.glob('*.txt')
for filename in files:
os.unlink(filename)
files2 = glob.glob('*.png')
for filename in files2:
os.unlink(filename)