Mercurial > repos > jpayne > tableops
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:])
