arrayview 0.26.1__tar.gz → 0.26.2__tar.gz

{arrayview-0.26.1 → arrayview-0.26.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arrayview
-Version: 0.26.1
+Version: 0.26.2
 Summary: Fast multi-dimensional array viewer
 Project-URL: Home, https://github.com/oscarvanderheide/arrayview
 Project-URL: Source, https://github.com/oscarvanderheide/arrayview

{arrayview-0.26.1 → arrayview-0.26.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "arrayview"
-version = "0.26.1"
+version = "0.26.2"
 description = "Fast multi-dimensional array viewer"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.12"

{arrayview-0.26.1 → arrayview-0.26.2}/src/arrayview/_stdio_server.py

@@ -115,14 +115,19 @@ def _handle_register(msg: dict) -> None:
     name = msg.get("name") or __import__("os").path.basename(file_path)
     options = msg.get("options", {})
 
-    from ._io import default_array_key
+    from ._io import list_array_keys
 
+    array_keys = list_array_keys(file_path)
+    key = array_keys[0]["key"] if array_keys else None
     data, spatial_meta = load_data_with_meta(
         file_path,
-        key=default_array_key(file_path),
+        key=key,
     )
     session = Session(data=data, filepath=file_path, name=name)
     session.spatial_meta = spatial_meta
+    if len(array_keys) > 1:
+        session.array_keys = array_keys
+        session.array_filepath = file_path
     if spatial_meta is not None:
         session.original_volume = data
 
@@ -154,6 +159,48 @@ def _handle_metadata(msg: dict) -> None:
     _write_json(_build_metadata(session))
 
 
+def _handle_reload_key(sid: str, msg: dict) -> None:
+    """Reload a session's data from a multi-array file with a different key."""
+    session = SESSIONS.get(sid)
+    if not session:
+        _write_error("session not found")
+        return
+    array_filepath = getattr(session, "array_filepath", None)
+    if not array_filepath:
+        _write_error("session has no array keys")
+        return
+
+    try:
+        body = json.loads(msg.get("body") or "{}")
+    except Exception:
+        _write_error("invalid JSON body")
+        return
+    key = body.get("key")
+    if not key:
+        _write_error("key is required")
+        return
+
+    try:
+        from ._io import load_data
+
+        data = load_data(array_filepath, key=key)
+    except Exception as e:
+        _write_error(str(e))
+        return
+
+    session.data = data
+    session.shape = data.shape
+    session.data_version += 1
+    session.raw_cache.clear()
+    session.rgba_cache.clear()
+    session.mosaic_cache.clear()
+    session._raw_bytes = 0
+    session._rgba_bytes = 0
+    session._mosaic_bytes = 0
+    session._estimated_mem = session._estimate_memory()
+    _write_json({"ok": True})
+
+
 def _handle_sessions() -> None:
     """Return list of active sessions."""
     _write_json([_session_summary(s) for s in SESSIONS.values()])
@@ -409,6 +456,8 @@ def _handle_fetch_proxy(msg: dict) -> None:
         _handle_metadata({"sid": sid})
     elif route == "clearcache" and sid:
         _handle_clearcache({"sid": sid})
+    elif route == "session" and len(parts) >= 3 and parts[2] == "reload-key":
+        _handle_reload_key(parts[1], msg)
     elif route == "sessions":
         _handle_sessions()
     elif route == "histogram" and sid:

{arrayview-0.26.1 → arrayview-0.26.2}/tests/test_io.py

@@ -52,7 +52,7 @@ def test_mat_loader_returns_key_list_for_multiple_numeric_arrays(tmp_path):
     )
 
 
-def test_stdio_register_uses_default_mat_key(tmp_path, monkeypatch):
+def test_stdio_mat_register_exposes_keys_and_reload_switches_array(tmp_path, monkeypatch):
     import scipy.io
 
     import arrayview._stdio_server as stdio
@@ -75,8 +75,65 @@ def test_stdio_register_uses_default_mat_key(tmp_path, monkeypatch):
     try:
         stdio._handle_register({"path": str(path)})
         assert len(responses) == 1
+        sid = responses[0]["sid"]
         assert responses[0]["shape"] == [2, 3]
+        assert SESSIONS[sid].array_keys == [
+            {"key": "first", "shape": [2, 3], "dtype": "float32"},
+            {"key": "second", "shape": [4, 5], "dtype": "float32"},
+        ]
         assert "error" not in responses[0]
+
+        responses.clear()
+        stdio._handle_fetch_proxy(
+            {
+                "endpoint": f"/session/{sid}/reload-key",
+                "method": "POST",
+                "body": '{"key":"second"}',
+            }
+        )
+
+        assert responses == [{"ok": True}]
+        assert SESSIONS[sid].shape == (4, 5)
+    finally:
+        for sid in set(SESSIONS) - before:
+            SESSIONS.pop(sid, None)
+
+
+def test_stdio_npz_register_exposes_keys_and_reload_switches_array(tmp_path, monkeypatch):
+    import arrayview._stdio_server as stdio
+    from arrayview._session import SESSIONS
+
+    path = tmp_path / "stdio_multiple_arrays.npz"
+    np.savez(
+        path,
+        first=np.zeros((2, 3), dtype=np.float32),
+        second=np.ones((4, 5), dtype=np.float32),
+    )
+
+    responses = []
+    monkeypatch.setattr(stdio, "_write_json", responses.append)
+
+    before = set(SESSIONS)
+    try:
+        stdio._handle_register({"path": str(path)})
+        sid = responses[0]["sid"]
+        assert responses[0]["shape"] == [2, 3]
+        assert SESSIONS[sid].array_keys == [
+            {"key": "first", "shape": [2, 3], "dtype": "float32"},
+            {"key": "second", "shape": [4, 5], "dtype": "float32"},
+        ]
+
+        responses.clear()
+        stdio._handle_fetch_proxy(
+            {
+                "endpoint": f"/session/{sid}/reload-key",
+                "method": "POST",
+                "body": '{"key":"second"}',
+            }
+        )
+
+        assert responses == [{"ok": True}]
+        assert SESSIONS[sid].shape == (4, 5)
     finally:
         for sid in set(SESSIONS) - before:
             SESSIONS.pop(sid, None)

{arrayview-0.26.1 → arrayview-0.26.2}/uv.lock

@@ -198,7 +198,7 @@ wheels = [
 
 [[package]]
 name = "arrayview"
-version = "0.26.1"
+version = "0.26.2"
 source = { editable = "." }
 dependencies = [
     { name = "fastapi" },