swap usage per process

swap_usage.py

#!/usr/bin/env python3

import os
import re
import psutil
import tempfile

def get_swap_usage():
    swap_data = []
    total_swap = 0

    # Iterate through all process directories in /proc/
    for pid in [d for d in os.listdir('/proc') if d.isdigit()]:
        try:
            # Get the program name
            proc_name = psutil.Process(int(pid)).name()

            # Calculate the total swap used by this process
            swap_usage = 0
            smaps_file = f'/proc/{pid}/smaps'
            if os.path.exists(smaps_file):
                with open(smaps_file, 'r') as f:
                    for line in f:
                        if 'Swap:' in line:
                            swap_usage += int(line.split()[1])
            
            if swap_usage > 0:
                swap_data.append((int(pid), swap_usage, proc_name))
                total_swap += swap_usage

        except (psutil.NoSuchProcess, FileNotFoundError):
            # Process might have ended or the smaps file might not exist
            continue

    return swap_data, total_swap

def print_swap_usage(swap_data, total_swap, sort_by):
    if sort_by == "name":
        swap_data.sort(key=lambda x: x[2].lower(), reverse=True)
        print(f"{'Name':<25} {'kB':>10} {'PID':>10}")
    elif sort_by == "kb":
        swap_data.sort(key=lambda x: x[1], reverse=True)
        print(f"{'kB':>10} {'PID':>10} {'Name':<25}")
    else:
        swap_data.sort(key=lambda x: x[0], reverse=True)
        print(f"{'PID':>10} {'kB':>10} {'Name':<25}")

    print("=" * 50)
    for pid, swap_usage, proc_name in swap_data:
        if sort_by == "name":
            print(f"{proc_name:<25} {swap_usage:>10} {pid:>10}")
        elif sort_by == "kb":
            print(f"{swap_usage:>10} {pid:>10} {proc_name:<25}")
        else:
            print(f"{pid:>10} {swap_usage:>10} {proc_name:<25}")
    
    print("=" * 50)
    print(f"Overall swap used: {total_swap} kB")
    print("=" * 50)

def main():
    sort_by = "kb"  # Default sort by kb
    if len(os.sys.argv) > 1:
        sort_by = os.sys.argv[1].lower()

    swap_data, total_swap = get_swap_usage()
    print_swap_usage(swap_data, total_swap, sort_by)

if __name__ == "__main__":
    main()