Add support for handling zarr urls

lib/iris/fileformats/__init__.py

@@ -155,6 +155,20 @@ def _load_grib(*args, **kwargs):
 )
 
 
+#
+# Zarr files.
+#
+FORMAT_AGENT.add_spec(
+    FormatSpecification(
+        "zarr",
+        FileExtension(requires_fh=False),
+        ".zarr#mode=nczarr,file",
+        netcdf.load_cubes,
+        priority=3,
+    )
+)
+
+
 #
 # UM Fieldsfiles.
 #

lib/iris/fileformats/netcdf/saver.py

@@ -418,7 +418,11 @@ def __init__(self, filename, netcdf_format, compute=True):
         else:
             # Given a filepath string/path : create a dataset from that
             try:
-                self.filepath = Path(filename).absolute()
+                filepath = Path(filename)
+                if ".zarr" not in filepath.suffix:
+                    self.filepath = filepath.absolute()
+                else:
+                    self.filepath = filepath
                 self._dataset = _thread_safe_nc.DatasetWrapper(
                     self.filepath, mode="w", format=netcdf_format
                 )

lib/iris/io/__init__.py

@@ -129,7 +129,10 @@ def decode_uri(uri, default="file"):
         # put - last in the brackets so it refers to the character, not a range
         # reference on valid schemes: https://tools.ietf.org/html/std66#section-3.1
         match = re.match(r"^([a-zA-Z][a-zA-Z0-9+.-]+):(.+)", uri)
-        if match:
+        if ".zarr" in uri:
+            scheme = "zarr"
+            part = uri
+        elif match:
             scheme = match.group(1)
             part = match.group(2)
         else:

lib/iris/loading.py

@@ -97,6 +97,10 @@ def _generate_cubes(uris, callback, constraints):
             urls = [":".join(x) for x in groups]
             for cube in iris.io.load_http(urls, callback):
                 yield cube
+        elif scheme == "zarr":
+            urls = [x[1] for x in groups]
+            for cube in iris.io.load_http(urls, callback):
+                yield cube
         elif scheme == "data":
             data_objects = [x[1] for x in groups]
             for cube in iris.io.load_data_objects(data_objects, callback):

lib/iris/tests/results/file_load/known_loaders.txt

@@ -14,4 +14,5 @@
  * UM Fieldsfile (FF) converted with ieee to 32 bit (priority 3)
  * UM Fieldsfile (FF) pre v3.1 (priority 3)
  * UM Post Processing file (PP) little-endian (priority 3)
+ * zarr (priority 3)
  * GRIB (priority 1)