- 论坛徽章:
- 2
|
本帖最后由 cnvbird 于 2014-01-16 11:28 编辑
- #!/usr/bin/env python
- # -*- coding:utf-8 -*-
- import os
- for item in os.walk('/data'):
- file_type = {}
- # 构造一个字典数据结构,key存放文件后缀,value存放文件列表
- for f in item[2]:
- i = -1
- suffix = ''
- while f[i] != '.':
- suffix += f[i]
- i -= 1
- if suffix[::-1] not in file_type:
- file_type[suffix[::-1]] = []
- file_type[suffix[::-1]].append(f)
- else:
- file_type[suffix[::-1]].append(f)
- if len(item[2]) > 0:
- size = 0
- for files in file_type.values()[:]:
- for file_name in files:
- size += os.path.getsize(item[0] + os.sep + file_name)
- print item[0], file_type, size
复制代码 |
|