Mercurial > repos > jpayne > tableops
diff 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 diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/table-sort.py Mon Dec 08 15:03:06 2025 +0000 @@ -0,0 +1,26 @@ +#!/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:])
