Class ZarrArrayWrapper

xarray/backends/zarr.py:242–333

class ZarrArrayWrapper(BackendArray):
    __slots__ = ("_array", "dtype", "shape")

    def __init__(self, zarr_array):
        # some callers attempt to evaluate an array if an `array` property exists on the object.
        # we prefix with _ to avoid this inference.

        # TODO type hint this?
        self._array = zarr_array
        self.shape = self._array.shape

        # preserve vlen string object dtype (GH 7328)
        if (
            not _zarr_v3()
            and self._array.filters is not None
            and any(filt.codec_id == "vlen-utf8" for filt in self._array.filters)
        ) or (
            _zarr_v3()
            and self._array.serializer
            and self._array.serializer.to_dict()["name"] == "vlen-utf8"
        ):
            dtype = coding.strings.create_vlen_dtype(str)
        else:
            dtype = self._array.dtype

        self.dtype = dtype

    def get_array(self):
        return self._array

    def _oindex(self, key):
        return self._array.oindex[key]

    def _vindex(self, key):
        return self._array.vindex[key]

    def _getitem(self, key):
        return self._array[key]

    async def _async_getitem(self, key):
        if not _zarr_v3():
            raise NotImplementedError(
                "For lazy basic async indexing with zarr, zarr-python=>v3.0.0 is required"
            )

        async_array = self._array._async_array
        return await async_array.getitem(key)

    async def _async_oindex(self, key):
        if not has_zarr_async_index():
            raise NotImplementedError(
                "For lazy orthogonal async indexing with zarr, zarr-python=>v3.1.2 is required"
            )

        async_array = self._array._async_array
        return await async_array.oindex.getitem(key)

    async def _async_vindex(self, key):