climate-ref-esmvaltool 0.8.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

climate_ref_esmvaltool/__init__.py

@@ -2,14 +2,65 @@
 Rapid evaluating CMIP data with ESMValTool.
 """
 
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING
+
+import pooch
+from loguru import logger
+
 import climate_ref_esmvaltool.diagnostics
-from climate_ref_core.dataset_registry import DATASET_URL, dataset_registry_manager
+from climate_ref_core.dataset_registry import (
+    DATASET_URL,
+    dataset_registry_manager,
+    fetch_all_files,
+    validate_registry_cache,
+)
 from climate_ref_core.providers import CondaDiagnosticProvider
 from climate_ref_esmvaltool._version import __version__
 from climate_ref_esmvaltool.recipe import _ESMVALTOOL_COMMIT
 
+if TYPE_CHECKING:
+    from climate_ref.config import Config
+
+_REGISTRY_NAME = "esmvaltool"
+
+
+class ESMValToolProvider(CondaDiagnosticProvider):
+    """Provider for ESMValTool diagnostics."""
+
+    def fetch_data(self, config: Config) -> None:
+        """Fetch ESMValTool reference data."""
+        registry = dataset_registry_manager[_REGISTRY_NAME]
+        fetch_all_files(registry, _REGISTRY_NAME, output_dir=None)
+
+    def validate_setup(self, config: Config) -> bool:
+        """Validate conda environment and data checksums."""
+        # First check conda environment
+        if not super().validate_setup(config):
+            return False
+
+        # Then check data checksums
+        registry = dataset_registry_manager[_REGISTRY_NAME]
+        errors = validate_registry_cache(registry, _REGISTRY_NAME)
+        if errors:
+            for error in errors:
+                logger.error(f"{self.slug} validation failed: {error}")
+            logger.error(
+                f"Data for {self.slug} is missing or corrupted. "
+                f"Please run `ref providers setup --provider {self.slug}` to fetch data."
+            )
+            return False
+        return True
+
+    def get_data_path(self) -> Path | None:
+        """Get the path where ESMValTool data is cached."""
+        return Path(pooch.os_cache("climate_ref"))
+
+
 # Initialise the diagnostics manager.
-provider = CondaDiagnosticProvider(
+provider = ESMValToolProvider(
     "ESMValTool",
     __version__,
     repo="https://github.com/ESMValGroup/ESMValTool.git",

climate_ref_esmvaltool/recipe.py

@@ -144,8 +144,9 @@ def dataframe_to_recipe(
     return variables
 
 
-_ESMVALTOOL_COMMIT = "2c438d0e0cc8904790294c72450eb7f06552c52a"
-_ESMVALTOOL_VERSION = f"2.13.0.dev148+g{_ESMVALTOOL_COMMIT[:9]}"
+_ESMVALTOOL_COMMIT = "fa3e10ed09fc34d354f179a6a71aacd032d6ea9e"
+# _ESMVALTOOL_VERSION = f"2.13.0.dev148+g{_ESMVALTOOL_COMMIT[:9]}"
+_ESMVALTOOL_VERSION = "2.13.0"
 
 _RECIPES = pooch.create(
     path=pooch.os_cache("climate_ref_esmvaltool"),