toplev: Support .zst files for input/output

andikleen · Sep 23, 2023 · be9f3c1 · be9f3c1
1 parent 3b3fb2d
commit be9f3c1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tl_io.py b/tl_io.py
@@ -32,6 +32,8 @@ def flex_open_r(fn):
     if fn.endswith(".gz"):
         gzip = popen_stdout(["gzip", "-d", "-c", fn])
         return gzip.stdout
+    if fn.endswith(".zst"):
+        return popen_stdout(["zstd", "-d", "--stdout", fn]) .stdout
     return open(fn, 'r')
 
 def flex_open_w(fn):
@@ -42,6 +44,8 @@ def flex_open_w(fn):
     if fn.endswith(".gz"):
         gzip = popen_stdinout(["gzip", "-c"], f)
         return gzip.stdin
+    if fn.endswith(".zst"):
+        return popen_stdinout(["zstd", "--stdout"], f).stdin
     return f
 
 test_mode = os.getenv("TL_TESTER")