Add utility to get kwave bin paths and URLs (#377)

Author: ebrahimebrahim

src/openlifu/util/assets.py

@@ -134,3 +134,16 @@ def stmt_calls_banned(stmt: ast.stmt) -> bool:
 def _import_kwave_inertly() -> ModuleType:
     """Import kwave without allowing it to install binaries"""
     return _import_without_calls("kwave", banned_calls=["install_binaries"])
+
+def get_kwave_paths() -> list[tuple[Path, str]]:
+    """Get a list of paths and urls to kwave binaries.
+
+    Each item in the list is a pair consisting of the install of a needed binary, followed by a download url for that binary.
+    """
+    kwave = _import_kwave_inertly()
+    paths : list[tuple[str, str]] = []
+    for url_list in kwave.URL_DICT[kwave.PLATFORM].values():
+        for url in url_list:
+            _, filename = url.split("/")[-2:]
+            paths.append((Path(kwave.BINARY_PATH) / filename, url))
+    return paths

tests/test_assets.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
 import shutil
+from pathlib import Path
 from unittest.mock import MagicMock
 
 import pytest
 import requests
 
-from openlifu.util.assets import install_asset
+from openlifu.util.assets import get_kwave_paths, install_asset
 
 
 def test_destination_already_exists(tmp_path, mocker):
@@ -94,3 +95,14 @@ def test_raises_error_if_no_source_provided(tmp_path):
     destination = tmp_path / "asset.dat"
     with pytest.raises(ValueError, match="Either path_to_asset or url_to_asset must be provided."):
         install_asset(destination, path_to_asset=None, url_to_asset=None)
+
+def test_get_kwave_paths():
+    """Check that get_kwave_paths returns a nonempty list of (Path, str) pairs.
+    This may break if kwave changes how it represents download urls and binary paths."""
+    paths = get_kwave_paths()
+    assert isinstance(paths, list)
+    assert len(paths) > 0
+    for p, url in paths:
+        assert isinstance(p, Path)
+        assert isinstance(url, str)
+        assert len(url) > 0