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:])