[experiment](feat) Add util for composing model from stored code and model dump

Author: kzajac97

pydentification/experiment/storage/compose.py

@@ -2,6 +2,7 @@
 import json
 import os
 import sys
+import zipfile
 from pathlib import Path
 from typing import Callable
 
@@ -20,15 +21,37 @@ def _import_function_from_path(module_path: str, function_name: str) -> Callable
     return function
 
 
-def compose_model(path: str | Path, name: str = "model_fn", parameters: str | Path | None = None):
+def _safe_unzip(path: Path):
+    if path.with_suffix("").exists():
+        raise FileExistsError(f"Can't overwrite {path.stem}!")
+
+    with zipfile.ZipFile(path, "r") as zip:
+        zip.extractall(str(path.parent))
+
+
+def compose_model(
+    path: str | Path,
+    name: str = "model_fn",
+    parameters: str | Path | None = None,
+    source: str | Path | None = None,
+):
     """
     Compose model from dump, which will contain model generating function, JSON with its parameters and source code
     for module definitions (ZIP of entire `pydentification`).
 
     :param path: filesystem Path to the model generating function, which will be imported by `import_function_from_path`
     :param name: name of the function to be imported, default is `model_fn`
     :param parameters: filesystem Path to the JSON file with parameters, if None, empty dictionary will be used
+    :param source: filesystem Path to the ZIP file with source code
+                   if None imports are attempted from the current working directory.
     """
+    if isinstance(source, str):
+        source = Path(source)
+
+    if source is not None:
+        _safe_unzip(source)
+        sys.path.append(str(source.parent))
+
     model_fn = _import_function_from_path(path, name)
 
     if parameters is not None:
@@ -37,4 +60,8 @@ def compose_model(path: str | Path, name: str = "model_fn", parameters: str | Pa
     else:
         parameters = {}
 
-    return model_fn(parameters)
+    model = model_fn(**parameters)
+
+    if source is not None:
+        sys.path.remove(str(source.parent))
+    return model