WIP cleanup utils

bioimageio/core/utils/__init__.py

@@ -1,23 +1,7 @@
-# todo: cleanup __init__: move stuff to util submodules or elsewhere
 from __future__ import annotations
 
-import hashlib
-import importlib.util
-import os
 import sys
-from contextlib import AbstractContextManager
-from functools import singledispatch
 from pathlib import Path
-from types import TracebackType
-from typing import Any, Callable
-from urllib.parse import urlsplit, urlunsplit
-
-from pydantic import AnyUrl, HttpUrl
-from typing_extensions import Unpack
-
-from bioimageio.core.io import FileSource, HashKwargs, download
-from bioimageio.spec.model.v0_4 import CallableFromDepencency, CallableFromFile
-from bioimageio.spec.model.v0_5 import ArchitectureFromFileDescr, ArchitectureFromLibraryDescr
 
 if sys.version_info < (3, 9):
 
@@ -29,58 +13,39 @@ def files(package_name: str):
     from importlib.resources import files as files
 
 
-class TemporaryInsertionIntoPythonPath(AbstractContextManager[None]):
-    def __init__(self, path: Path):
-        super().__init__()
-        self.path = str(path)
-
-    def __enter__(self):
-        super().__enter__()
-        sys.path.insert(0, self.path)
-
-    def __exit__(
-        self,
-        __exc_type: "type[BaseException] | None",
-        __exc_value: "BaseException | None",
-        __traceback: "TracebackType | None",
-    ) -> "bool | None":
-        assert sys.path[0] == self.path
-        _ = sys.path.pop(0)
-        return super().__exit__(__exc_type, __exc_value, __traceback)
-
-
-@singledispatch
-def import_callable(node: type, /) -> Callable[..., Any]:
-    raise TypeError(type(node))
+# TODO: import helpers
+# @singledispatch
+# def import_callable(node: type, /) -> Callable[..., Any]:
+#     raise TypeError(type(node))
 
 
-@import_callable.register
-def import_from_dependency(node: CallableFromDepencency) -> Callable[..., Any]:
-    module = importlib.import_module(node.module_name)
-    c = getattr(module, node.callable_name)
-    if not callable(c):
-        raise ValueError(f"{node} (imported: {c}) is not callable")
+# @import_callable.register
+# def import_from_dependency(node: CallableFromDepencency) -> Callable[..., Any]:
+#     module = importlib.import_module(node.module_name)
+#     c = getattr(module, node.callable_name)
+#     if not callable(c):
+#         raise ValueError(f"{node} (imported: {c}) is not callable")
 
-    return c
+#     return c
 
 
-@import_callable.register
-def import_from_file04(node: CallableFromFile, **kwargs: Unpack[HashKwargs]):
-    return _import_from_file_impl(node.file, node.callable_name, **kwargs)
+# @import_callable.register
+# def import_from_file04(node: CallableFromFile, **kwargs: Unpack[HashKwargs]):
+#     return _import_from_file_impl(node.file, node.callable_name, **kwargs)
 
 
-@import_callable.register
-def import_from_file05(node: CallableFromFile05, **kwargs: Unpack[HashKwargs]):
-    return _import_from_file_impl(node.source_file, node.callable_name, **kwargs)
+# @import_callable.register
+# def import_from_file05(node: CallableFromFile05, **kwargs: Unpack[HashKwargs]):
+#     return _import_from_file_impl(node.source_file, node.callable_name, **kwargs)
 
 
-def _import_from_file_impl(source: FileSource, callable_name: str, **kwargs: Unpack[HashKwargs]):
-    local_file = download(source, **kwargs)
-    module_name = local_file.path.stem
-    importlib_spec = importlib.util.spec_from_file_location(module_name, local_file.path)
-    if importlib_spec is None:
-        raise ImportError(f"Failed to import {module_name} from {source}.")
+# def _import_from_file_impl(source: FileSource, callable_name: str, **kwargs: Unpack[HashKwargs]):
+#     local_file = download(source, **kwargs)
+#     module_name = local_file.path.stem
+#     importlib_spec = importlib.util.spec_from_file_location(module_name, local_file.path)
+#     if importlib_spec is None:
+#         raise ImportError(f"Failed to import {module_name} from {source}.")
 
-    dep = importlib.util.module_from_spec(importlib_spec)
-    importlib_spec.loader.exec_module(dep)  # type: ignore  # todo: possible to use "loader.load_module"?
-    return getattr(dep, callable_name)
+#     dep = importlib.util.module_from_spec(importlib_spec)
+#     importlib_spec.loader.exec_module(dep)  # type: ignore  # todo: possible to use "loader.load_module"?
+#     return getattr(dep, callable_name)