find grib files

ecmwf · May 1, 2024 · 1fe9bbb · 1fe9bbb
1 parent b26ea56
commit 1fe9bbb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -4,7 +4,6 @@ repos:
     rev: 24.1.1
     hooks:
       - id: black
-        language_version: python3.10
 -   repo: https://github.com/pycqa/isort
     rev: 5.13.0
     hooks:

diff --git a/climetlab/readers/grib/__init__.py b/climetlab/readers/grib/__init__.py
@@ -16,3 +16,11 @@ def reader(source, path, magic=None, deeper_check=False):
         from .reader import GRIBReader
 
         return GRIBReader(source, path)
+
+    if deeper_check:
+        with open(path, "rb") as f:
+            magic = f.read(1024)
+            if b"GRIB" in magic:
+                from .reader import GRIBReader
+
+                return GRIBReader(source, path)