fix(read): handle filenames with dots confusing with extension

OxWearables · Feb 5, 2024 · 1e89ceb · 1e89ceb
1 parent 85ee34d
commit 1e89ceb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/stepcount/stepcount.py b/src/stepcount/stepcount.py
@@ -333,8 +333,10 @@ def nanint(x):
 def read(filepath, resample_hz='uniform', sample_rate=None, verbose=True):
 
     p = pathlib.Path(filepath)
-    ftype = p.suffixes[0].lower()
     fsize = round(p.stat().st_size / (1024 * 1024), 1)
+    ftype = p.suffix.lower()
+    if ftype in (".gz", ".xz", ".lzma", ".bz2", ".zip"):  # if file is compressed, check the next extension
+        ftype = pathlib.Path(p.stem).suffix.lower()
 
     if ftype in (".csv", ".pkl"):
 
@@ -386,6 +388,9 @@ def read(filepath, resample_hz='uniform', sample_rate=None, verbose=True):
             verbose=verbose,
         )
 
+    else:
+        raise ValueError(f"Unknown file format: {ftype}")
+
     if 'ResampleRate' not in info:
         info['ResampleRate'] = info['SampleRate']