Mercurial > repos > jpayne > tableops
comparison 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 |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:402b58f45844 |
|---|---|
| 1 #!/usr/bin/env python3 | |
| 2 | |
| 3 import csv | |
| 4 import sys | |
| 5 | |
| 6 | |
| 7 def main(headers): | |
| 8 rows = csv.DictReader(sys.stdin, delimiter="\t", dialect="excel-tab") | |
| 9 | |
| 10 # More efficient header check using set intersection | |
| 11 if not set(headers).intersection(rows.fieldnames): | |
| 12 raise ValueError( | |
| 13 f"Couldn't find any of supplied headers ({', '.join(map(repr, headers))}) in the table." | |
| 14 ) | |
| 15 | |
| 16 # Optimized sorting using tuple comparison (generally faster than list comparison) | |
| 17 items = sorted(rows, key=lambda d: tuple(d.get(h, "") for h in headers)) | |
| 18 | |
| 19 wr = csv.DictWriter(sys.stdout, dialect="excel-tab", fieldnames=rows.fieldnames) | |
| 20 wr.writeheader() | |
| 21 wr.writerows(items) | |
| 22 # sys.stdout.flush() | |
| 23 | |
| 24 | |
| 25 if __name__ == "__main__": | |
| 26 main(sys.argv[1:]) |