Script to sort and print output from `du -sh` Bash command

sort_du_output_by_dir_size.py

## Script to sort and print output from `du -sh` Bash command
## xram | 5/29/25

# -- Output from `du -sh` command -------------------------
du = """
1.2M    /bin/bash
776K    /bin/ssh
7.1M    /boot/grub
97M     /lib/gcc
2.5G    /var/log
"""
# ---------------------------------------------------------

from functools import cmp_to_key

# Define comparison-style sort function
def sortfn(a, b):
    # Returns 1 if a>b, -1 if a<b, and 0 if a=b
    
    a_val = a.split(' ')[0]
    b_val = b.split(' ')[0]
    a_mag = a_val[-1]
    b_mag = b_val[-1]
    
    m = {'K':1, 'M':2, 'G':3}  # magnitude suffixes
    
    if m[a_mag] < m[b_mag]: return -1    # not(a > b)
    elif m[a_mag] > m[b_mag]: return 1   # (a > b)
    else:
        return 1 if float(a_val[:-1]) > float(b_val[:-1]) else -1

# Split list by lines (ignore empty lines)
du_list = [line for line in du.split('\n') if len(line.strip()) > 1]

# Sort lines
du_list.sort(key=cmp_to_key(sortfn), reverse=True)  # translate comparison-style function into key-style function

for line in du_list:
    print(line)