【Python】CDay-1 优化！对自个的反省

编程实现以下功能并进行最大化的优化：遍历指定目录下的所有文件，找出其中占用空间最大的前3个文件。

#!/usr/bin/env python
#-*- coding: utf-8 -*-
'''selectbiggestfile.py
@note: 使用os.stat获取相关信息，os.walk遍历，
@see: help(os)
@authoer: U{119<mailto：119@gmail.com>}
@version: $id$
'''
import sys
import os
def get_top_three(path):
'''获取给定路径中文件大小最大的三个
@param path: 指定路径
@return 返回一个list，每项为(size, filename)
'''
all_file = {}
for root, dirs, files in os.walk(path): #遍历path
for onefile in files:
fname = os.path.join(root, onefile) #获得当前处理文件的完整名字
#print 'fname is %s' %fname
fsize = os.stat(fname).st_size #获得当前处理文件大小
#print 'all_file is %s' %all_file
if all_file.has_key(fsize): #按照文件大小存储
all_file[fsize].append(fname)
else:
all_file[fsize] = [fname]
fsize_key = all_file.keys() #得到所有的文件大小
print 'fsize_key is %s\n' %fsize_key
fsize_key.sort() #排序，从小到大
print 'fsize_key is %s\n' %fsize_key
result = []
for i in [-1, -2, -3]: #依次取最大的三个
print 'all_file[fsize_key[i]] is %s\n' %all_file[fsize_key[i]]
print 'fsize_key[i] is %s\n' %fsize_key[i]
for j in all_file[fsize_key[i]]: #保存
result.append((fsize_key[i], j))
print 'result is %s\n' %result
return result[:3] #返回前三个
if __name__ == "__main__":
if len(sys.argv) == 1:
print 'usage:\n\tpython selectbiggestfile.py path'
else:
print 'sys.argv is %s\n' %sys.argv
abs_path = os.path.abspath(sys.argv[1]) #得到绝对路径
print 'abs_path is %s\n' %abs_path
if not os.path.isdir(abs_path): #判断所给的路径是否存在
print '%s is not exist' %abs_path
else:
top = get_top_three(abs_path)
print 'top is %s\n' %top
for (s, f) in top:
print '%s\t -> \t%s' %(f,s)

运行截屏:



利用ConfigParser，将上述题目中产生的结果按照cdays+1-my.ini格式存储到文件cdays+1-result.txt中。

cdays+1-my.ini内容为：

[Number]
filesize = somefilesize
filename = somefilename

源代码

#!/usr/bin/env python
#-*- coding: utf-8 -*-
import os
import sys
from ConfigParser import RawConfigParser
from selectbiggestfile import get_top_three

'''parseresult.py
@note: 利用ConfigParser解析ini格式
@see: 文档参见http://pydoc.org/2.4.1/ConfigParser.html, 其他例子http://effbot.org/librarybook/configparser-example-1.py
@authoer: U{119<mailto：119@gmail.com>}
@version: $id$
'''
def iniTT(size_file):
'''按照.ini的格式，存储size_file
'''
cfg = RawConfigParser()
print size_file
index = 1
for (s, f) in size_file:
cfg.add_section("%d" %index)
cfg.set("%d" %index, 'Filename', f)
cfg.set('%d' %index, 'FileSize', s)
index += 1
cfg.write(open('parseresult.txt','w'))
if __name__ == '__main__':
if len(sys.argv) == 1:
print 'usage:\n\tpython parseresult.py path'
else:
abs_path = os.path.abspath(sys.argv[1])
if not os.path.isdir(abs_path):
print '%s is not exist' % abs_path
else:
iniTT(get_top_three(abs_path))

运行结果


parseresult.txt如下：