Disk_usage_stats.py

#!/usr/bin/python3

import sys
import pandas as pd
import os

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 Exception as e:
            print("Exception: ", e)
            total += 0
            
    return total

if __name__ == '__main__':
    
    # if you're running this in linux, change the default path to "/home"
    # also run the code as "sudo python3 Disk_usage.py" or "sudo ./Disk_usage.py"
    # i.e, with super user permissions
    
    path = "C:/Users/"
    print("Total arguments passed: ", len(sys.argv))

    directory = sys.argv[1] if len(sys.argv) >= 2 else path
    usage = []
    paths = []
    
    for entry in os.scandir(directory):
        if entry.is_dir(follow_symlinks = False):
            
            print(entry.path + " is a directory.")
            print(get_size(entry.path))
            total = get_size(entry.path)
            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")