Helper function to return keys in the sdata.attrs slot in an specific format

[timtreis]

Gets a specific key from sdata.attrs and then returns it either

a) as is (return_as=None) or
b) converted (return_as="dict" | "json" | "df") and potentially flattened (flatten=True, separated by sep)

Needed for spatialdata-db project, but tried to keep it general.

import spatialdata
from spatialdata.datasets import blobs

sdata = blobs()

sdata.attrs  # empty dictionnary
sdata.attrs["test"] = {"a": {"b": 12}, "c": 8}

sdata.write("blobs.zarr", overwrite=True) 
sdata = spatialdata.read_zarr("blobs.zarr") 

sdata.get_attrs(key="test", return_as=None, flatten=False)

{'a': {'b': 12}, 'c': 8}

import spatialdata
from spatialdata.datasets import blobs

sdata = blobs()

sdata.attrs  # empty dictionnary
sdata.attrs["test"] = {"a": {"b": 12}, "c": 8}

sdata.write("blobs.zarr", overwrite=True) 
sdata = spatialdata.read_zarr("blobs.zarr") 

sdata.get_attrs(key="test", return_as="df", flatten=False)

import spatialdata
from spatialdata.datasets import blobs

sdata = blobs()

sdata.attrs  # empty dictionnary
sdata.attrs["test"] = {"a": {"b": 12}, "c": 8}

sdata.write("blobs.zarr", overwrite=True) 
sdata = spatialdata.read_zarr("blobs.zarr") 

sdata.get_attrs(key="test", return_as="df", flatten=True)

import spatialdata
from spatialdata.datasets import blobs

sdata = blobs()

sdata.attrs  # empty dictionnary
sdata.attrs["test"] = {"a": {"b": 12}, "c": 8}

sdata.write("blobs.zarr", overwrite=True) 
sdata = spatialdata.read_zarr("blobs.zarr") 

sdata.get_attrs(key="test", return_as="json", flatten=True)

FYI @Zethson