Sort sacctmgr lines

sort_sacctmgr_structure.py

#!/usr/bin/env python3
import sys
from collections import defaultdict, OrderedDict

def main():
  # Read all lines and track original order
  lines = [line.strip() for line in sys.stdin if line.strip()]

  # Build hierarchy while preserving order
  hierarchy = defaultdict(OrderedDict)
  account_lines = {}
  root_accounts = OrderedDict()

  for line in lines:
    parts = line.split()
    account = parts[3]
    parent = parts[-1].split('=')[1]
    account_lines[account] = line

    if parent == 'root':
      root_accounts[account] = None  # OrderedDict preserves insertion order
    else:
      hierarchy[parent][account] = None  # OrderedDict preserves insertion order

  # Print in correct order
  def print_account(account):
    print(account_lines[account])
    for child in hierarchy.get(account, []):
      print_account(child)

  for account in root_accounts:
    print_account(account)

if __name__ == '__main__':
    main()