disk_report.py

#!/usr/bin/python3
import sys
import os
import pandas as pd

def get_size(path):
    total = 0
    for entry in os.scandir(path):
        try:
            if (entry.is_dir(follow_symlinks=False)):
                total += get_size(entry.path)
            else:
                total += entry.stat(follow_symlinks=False).st_size
        except:
            print("Exception: ", Exception)
            total += 0
    return total

if __name__ == '__main__':
    print("total arguments passed: ", len(sys.argv))

    if len(sys.argv) >= 2:
        directory = sys.argv[1]
    else:
        directory = '/home'

    usage = []
    paths = []

    for entry in os.scandir(directory):
        print(entry.path)
        # if is a directory and isn't a simbolic link
        if(entry.is_dir(follow_symlinks=False)):
            # print(entry.path + " is a directory")
            # print(get_size(entry.path))
            total = get_size(entry.path)
            print(total)
            paths.append(entry.path)
            usage.append(total)
        usage_dict = {'directory' : paths, 
                      'usage'     : usage }
        df = pd.DataFrame(usage_dict)
        print(df)
        df.to_csv("disk_home_usage.csv")