yoki5 0.1.0__py3-none-any.whl

yoki5/__init__.py

@@ -0,0 +1,10 @@
+"""Wrapper around h5py to give easier interface around complex files."""
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("yoki5")
+except PackageNotFoundError:
+    __version__ = "uninstalled"
+
+__author__ = "Lukasz G. Migas"
+__email__ = "lukas.migas@yahoo.com"

yoki5/_version.py

@@ -0,0 +1,16 @@
+# file generated by setuptools_scm
+# don't change, don't track in version control
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple, Union
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+else:
+    VERSION_TUPLE = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+
+__version__ = version = '0.1.0'
+__version_tuple__ = version_tuple = (0, 1, 0)

yoki5/attrs.py

@@ -0,0 +1,48 @@
+"""Attributes."""
+
+from __future__ import annotations
+
+import typing as ty
+from contextlib import suppress
+
+if ty.TYPE_CHECKING:
+    from yoki5.base import Store
+
+
+class Attributes:
+    """Attributes"""
+
+    def __init__(self, parent: Store):
+        self.parent = parent
+
+    def __getitem__(self, item: str) -> ty.Any:
+        """Get attribute."""
+        with self.parent.open() as h5:
+            return h5.attrs[item]
+
+    def __contains__(self, key: str) -> bool:
+        """Check attribute."""
+        with self.parent.open() as h5:
+            return key in h5.attrs
+
+    def __setitem__(self, key: str, value: ty.Any) -> None:
+        """Set attribute."""
+        self.parent.check_can_write()
+        with self.parent.open() as h5:
+            h5.attrs[key] = value
+            h5.flush()
+
+    def to_dict(self) -> dict[str, ty.Any]:
+        """Return all attributes."""
+        attrs = {}
+        with self.parent.open() as h5:
+            keys = list(h5.attrs.keys())
+            for key in keys:
+                attrs[key] = h5.attrs[key]
+        return attrs
+
+    def get(self, key: str, default: ty.Any = None) -> ty.Any:
+        """Get attribute if it exists."""
+        with suppress(KeyError):
+            return self[key]
+        return default