add .bz2 decompression support

Author: valgur

README.md

@@ -27,6 +27,7 @@ where ease of cross-platform install and correctness are primary goals.
 * Plain ASCII or seamlessly read compressed ASCII in:
   * `.gz` GZIP
   * `.Z` LZW
+  * `.bz2` bzip2
   * `.zip`
 * Hatanaka compressed RINEX (plain `.crx` or `.crx.gz` etc.)
 * Python `io.StringIO` text stream RINEX'

src/georinex/rio.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 import typing as T
 import gzip
+import bz2
 import zipfile
 from pathlib import Path
 from contextlib import contextmanager
@@ -48,6 +49,14 @@ def opener(fn: T.TextIO | Path, header: bool = False) -> T.Iterator[T.TextIO]:
                     # even with 'rt' doesn't decompress until read.
                     f = io.StringIO(crx2rnx(f.read()))
                 yield f
+        elif suffix == ".bz2":
+            with bz2.open(fn, "rt") as f:
+                version, is_crinex = rinex_version(first_nonblank_line(f))
+                f.seek(0)
+
+                if is_crinex and not header:
+                    f = io.StringIO(crx2rnx(f.read()))
+                yield f
         elif suffix == ".zip":
             with zipfile.ZipFile(fn, "r") as z:
                 flist = z.namelist()

src/georinex/tests/test_hatanaka.py

@@ -79,8 +79,9 @@ def test_obs3_gz():
 
 
 @pytest.mark.timeout(30)
-def test_obs3():
-    fn = R / "P43300USA_R_20190012056_17M_15S_MO.crx"
+@pytest.mark.parametrize("suffix", [".crx", ".crx.bz2"])
+def test_obs3(suffix):
+    fn = R / ("P43300USA_R_20190012056_17M_15S_MO" + suffix)
 
     info = gr.rinexinfo(fn)