view table-sort.py @ 0:402b58f45844 draft default tip

planemo upload commit 9cc4dc1db55299bf92ec6bd359161ece4592bd16-dirty
author jpayne
date Mon, 08 Dec 2025 15:03:06 +0000
parents
children
line wrap: on
line source

#!/usr/bin/env python3

import csv
import sys


def main(headers):
    rows = csv.DictReader(sys.stdin, delimiter="\t", dialect="excel-tab")

    # More efficient header check using set intersection
    if not set(headers).intersection(rows.fieldnames):
        raise ValueError(
            f"Couldn't find any of supplied headers ({', '.join(map(repr, headers))}) in the table."
        )

    # Optimized sorting using tuple comparison (generally faster than list comparison)
    items = sorted(rows, key=lambda d: tuple(d.get(h, "") for h in headers))

    wr = csv.DictWriter(sys.stdout, dialect="excel-tab", fieldnames=rows.fieldnames)
    wr.writeheader()
    wr.writerows(items)
    # sys.stdout.flush()


if __name__ == "__main__":
    main(sys.argv[1:])