ewoksdata 2.0.1__tar.gz → 2.0.2__tar.gz

{ewoksdata-2.0.1/src/ewoksdata.egg-info → ewoksdata-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ewoksdata
-Version: 2.0.1
+Version: 2.0.2
 Summary: Ewoks tasks for data access
 Author-email: ESRF <dau-pydev@esrf.fr>
 License: # MIT License

{ewoksdata-2.0.1 → ewoksdata-2.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ewoksdata"
-version = "2.0.1"
+version = "2.0.2"
 authors = [{ name = "ESRF", email = "dau-pydev@esrf.fr" }]
 description = "Ewoks tasks for data access"
 readme = { file = "README.md", content-type = "text/markdown" }

{ewoksdata-2.0.1 → ewoksdata-2.0.2}/src/ewoksdata/data/bliss.py

@@ -3,7 +3,9 @@ or from memory (Redis or device servers).
 """
 
 import logging
+import os
 import sys
+import time
 from numbers import Integral
 from numbers import Number
 from typing import Any
@@ -52,6 +54,8 @@ def get_data(
         data_url = url.as_dataurl(data)
         filename, h5path, idx = url.h5dataset_url_parse(data_url)
 
+        _wait_for_file(filename, **hdf5_options)
+
         if h5py.is_hdf5(filename):
             return _get_hdf5_data(filename, h5path, idx=idx, **hdf5_options)
 
@@ -73,16 +77,36 @@ def get_data(
     raise TypeError(f"Unsupported data type: {type(data)!r}")
 
 
+def _wait_for_file(
+    filename: str,
+    retry_timeout: Optional[float] = None,
+    retry_period: Optional[float] = None,
+    **_,
+) -> None:
+    if retry_period is None:
+        retry_period = 0.1
+    t0 = time.time()
+    while True:
+        if os.path.exists(filename):
+            return
+        if retry_timeout is not None:
+            if (time.time() - t0) > retry_timeout:
+                raise TimeoutError(
+                    f"File {filename} is not created within {retry_timeout} sec"
+                )
+        time.sleep(retry_period)
+
+
 def get_image(
     data: Union[str, DataUrl, ArrayLike, Number], **hdf5_options: Any
 ) -> numpy.ndarray:
     """
     Return a NumPy array or a scalar from *data*.
 
+    Retry options, such as `retry_timeout` and `retry_period` are forwarded to the internal HDF5 data reader.
+
     :param data: Either a data-URL, an array-like object or a scalar number.
-    :param retry_timeout: Timeout for the retry mechanism.
-    :param retry_period: Interval between retries.
-    :param **hdf5_options: Additional options forwarded to :class:`silx.io.h5py_utils.File`.
+    :param hdf5_options: Additional options forwarded to :class:`silx.io.h5py_utils.File`.
     :return: A 2-D image.
     """
     data = get_data(data, **hdf5_options)

{ewoksdata-2.0.1 → ewoksdata-2.0.2}/src/ewoksdata/tests/test_data_bliss.py

@@ -28,8 +28,8 @@ def test_get_data_hdf5(tmpdir):
     numpy.testing.assert_array_equal(img1, img3)
 
     filename_non_existing = str(tmpdir / "data_non_existing.h5")
-    with pytest.raises(IOError):
-        _ = bliss.get_data(f"{filename_non_existing}?path=/img")
+    with pytest.raises(TimeoutError):
+        _ = bliss.get_data(f"{filename_non_existing}?path=/img", retry_timeout=0)
 
 
 @pytest.mark.parametrize("lima_names", [(), ("p3",), ("p3", "p4")])

{ewoksdata-2.0.1 → ewoksdata-2.0.2/src/ewoksdata.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ewoksdata
-Version: 2.0.1
+Version: 2.0.2
 Summary: Ewoks tasks for data access
 Author-email: ESRF <dau-pydev@esrf.fr>
 License: # MIT License