Python - 批量获取文件夹的大小输出为文件格式化保存

Posted 2021-03-03 Johnthegreat

很多时候，查看一个文件夹下的每个文件大小可以轻易的做到，因为文件后面就是文件尺寸，但是如果需要查看一个文件夹下面所有的文件夹对应的尺寸，就发现需要把鼠标放到对应的文件夹上，稍等片刻才会出结果。

有时候，我们需要查看几十个甚至于上百个文件夹，找出包含文件最多，空间占用最大的那个，就比较麻烦了。这段代码是我以前的代码，可以按大小排序输出文件夹大小到txt文件，供使用的方便。

格式化当时花了很长时间，最后发现使用‘YaHei.Consolas’字体可以解决，对齐后输出结果看起来还算舒服。

上代码：

import os
import datetime


def get_folder_size(path):
    folder_size = 0

    if not os.path.exists(path):
        return folder_size

    if os.path.isfile(path):
        folder_size = os.path.getsize(path)
        return folder_size
    try:
        if os.path.isdir(path):
            with os.scandir(path) as directory_lists:
                for directory_list in directory_lists:
                    if directory_list.is_dir():
                        sub_folder_size = get_folder_size(directory_list.path)  # 递归获取大小
                        folder_size += sub_folder_size
                    elif directory_list.is_file():
                        file_size = os.path.getsize(directory_list.path)
                        folder_size += file_size

                return folder_size
    except:
        pass


# 以下主要是为了格式化输出
def get_file_length(file_name):
    characters = list(file_name)
    ascii_length = 0
    utf8_length = 0

    for character in characters:
        if ord(character) < 128:
            ascii_length += 1
        else:
            utf8_length += 2

    return ascii_length + utf8_length


def main(basedir):
    with os.scandir(basedir) as dirs:
        directory_size = []
        for dir in dirs:
            try:
                if not dir.is_file():
                    dirsize = round(get_folder_size(dir.path) / 1000000)  # return the file size in Mb
                    resformat = [dir.name, dirsize]
                    directory_size.append(resformat)
            except:
                pass
        results = sorted(directory_size, key=lambda x: x[1], reverse=True)  # return a list ordered by size
        results = [[i[0], \'文件夹大小：\' + str(i[1]) + \' Mb\'] for i in results]

        with open(basedir + os.sep + datetime.date.today().isoformat() + \'.txt\', \'a+\') as f:
            for result in results:
                # 按照50的宽度格式化输出结果
                len1 = 50 - get_file_length(result[0]) + len(result[0])
                len2 = 25 - get_file_length(result[1]) + len(result[1])
                f.writelines(\'{:<{len1}s} {:>{len2}s}\\n\'.format(result[0], result[1], len1=len1, len2=len2))
            print(\'The result was successfully saved in the directory with date as file name.\')


if __name__ == "__main__":
    basedir = input("Please input the directory you would like to know the sizes: ")
    main(basedir)

如果输入相应的文件夹路径，输出结果如下：

有时间我再简化一下代码，目前先这样。