Поправить скрипт Размер файлов в каталоге.

[satfan]

IDLE (Python 3.10 64-bit)
Win 10
—————————
Есть такой рабочий скрипт Catalog size:

import os
import matplotlib.pyplot as plt


def get_size_format(b, factor=1024, suffix="B"):
    for unit in ["", "K", "M", "G", "T", "P", "E", "Z"]:
        if b < factor:
            return f"{b:.2f}{unit}{suffix}"
        b /= factor
    return f"{b:.2f}Y{suffix}"


def get_directory_size(directory):
    total = 0
    try:
        for entry in os.scandir(directory):
            if entry.is_file():
                total += entry.stat().st_size
            elif entry.is_dir():
                total += get_directory_size(entry.path)
    except NotADirectoryError:
        return os.path.getsize(directory)
    except PermissionError:
        return 0
    return total


def plot_pie(sizes, names):
    plt.pie(sizes, labels=names, autopct=lambda pct: f"{pct:.2f}%")
    plt.title("Размеры подкаталогов и файлов")
    plt.show()


folder_path = 'E:\IPTV\__Готовые m3u'

directory_sizes = []
names = []

for directory in os.listdir(folder_path):
    directory = os.path.join(folder_path, directory)
    directory_size = get_directory_size(directory)

    if directory_size == 0:
        continue

    directory_sizes.append(directory_size)
    names.append(os.path.basename(directory) + ": " + get_size_format(directory_size))


print("Общий размер каталога:", get_size_format(sum(directory_sizes)))
plot_pie(directory_sizes, names)

Как его поправить чтобы список и размеры файлов в квталоге сохранялись построчно в текстовый файл ?