python统计文件大小_python工具--01.统计当前目录下的文件的大小

环境

os: centos7

python : 3.7

实现功能

统计当目录下的文件夹有文件的大小，单位KB/MB/B；

代码实现

#!/usr/bin/env python

# _*_ coding:utf-8 _*_

import os,math

summary=0

def size_file(str2):

global summary

summary=summary+os.path.getsize(str2)

def size_dir(str1):

dlist=os.listdir(str1)

# print("正在统计目录："+str1)

for f in dlist:

file1=os.path.join(str1,f)

if os.path.isfile(file1):

size_file(file1)

else:

size_dir(file1)

def init_os(path_init):

if os.path.exists(path_init):

# 文件存在计算文件的大小；

size_dir(path_init)

if 1024 <= math.ceil(summary) <= 1000000:

print("[{}]目录： ".format(path_init)+str(summary/1024)+'KB')

elif math.ceil(summary) > 1000000:

print("[{}]目录： ".format(path_init)+str((summary/1024)/1024)+'MB')

else:

print("[{}]目录： ".format(path_init)+str(summary)+'B')

else:

print("目录不存在!,请重新输入.")

def main(local_path):

dblist=os.listdir(local_path)

for d in dblist:

file2 = os.path.join(local_path, d)

# 统计当前目录下文件的大小；

if os.path.isfile(file2):

size_file(file2)

if 1024 <= math.ceil(summary) <= 1000000:

print("[{}]： ".format(file2) + str(summary / 1024) + 'KB')

elif math.ceil(summary) > 1000000:

print("[{}]： ".format(file2) + str((summary / 1024) / 1024) + 'MB')

else:

print("[{}]： ".format(file2) + str(summary) + 'B')

# 统计当前目录下文件夹大小

else:

init_os(file2)

if __name__ == '__main__':

path=os.getcwd()

main(path)