fm-weck 1.4.8__py3-none-any.whl → 1.5.0__py3-none-any.whl

fm_weck/grpc_service/run_store.py

@@ -0,0 +1,38 @@
+# This file is part of fm-weck: executing fm-tools in containerized environments.
+# https://gitlab.com/sosy-lab/software/fm-weck
+#
+# SPDX-FileCopyrightText: 2024 Dirk Beyer <https://www.sosy-lab.org>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import logging
+from threading import Lock
+
+from .request_handling import RunHandler
+
+RUNS_IN_PROGRESS = {}
+EXCEPTIONS = {}
+LOCK = Lock()
+logger = logging.getLogger(__name__)
+
+
+def add_run(run_handler: RunHandler) -> str:
+    with LOCK:
+        RUNS_IN_PROGRESS[run_handler.run_id] = run_handler
+
+    return run_handler.run_id
+
+
+def remove_run(run_id: str) -> None:
+    with LOCK:
+        RUNS_IN_PROGRESS.pop(run_id, None)
+
+
+def get_run(run_id: str) -> RunHandler:
+    with LOCK:
+        return RUNS_IN_PROGRESS.get(run_id)
+
+
+def active_runs() -> frozenset:
+    with LOCK:
+        return frozenset(RUNS_IN_PROGRESS.keys())

fm_weck/grpc_service/server_utils.py

@@ -0,0 +1,27 @@
+# This file is part of fm-weck: executing fm-tools in containerized environments.
+# https://gitlab.com/sosy-lab/software/fm-weck
+#
+# SPDX-FileCopyrightText: 2024 Dirk Beyer <https://www.sosy-lab.org>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import logging
+import tempfile
+from pathlib import Path
+
+TMP_DIR = Path(tempfile.gettempdir()) / "fm_weck"
+TMP_DIR.mkdir(parents=True, exist_ok=True)
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+
+
+def read_file(file_path: Path) -> bytes:
+    """
+    Returns the content of a file as bytes.
+
+    :param file_path: The path to the file.
+    :return: The content of the file as bytes.
+    """
+
+    with open(file_path, "rb") as file:
+        return file.read()

fm_weck/image_mgr.py

@@ -26,6 +26,8 @@ if TYPE_CHECKING:
 
 CONTAINERFILE = Path(__file__).parent / "resources" / "Containerfile"
 
+logger = logging.getLogger(__name__)
+
 
 class ImageMgr(object):
     """
@@ -49,7 +51,7 @@ class ImageMgr(object):
         if not image.base_images:
             raise NoImageError("No base image specified")
 
-        logging.info(
+        logger.info(
             "Building image from from base image %s with packages %s", image.base_images[0], image.required_packages
         )
         image_cmd = engine.image_from(CONTAINERFILE)

fm_weck/resources/Containerfile

@@ -8,6 +8,5 @@
 
 ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
-RUN apt-get update
 ARG REQUIRED_PACKAGES
-RUN apt-get install -y ${REQUIRED_PACKAGES} && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y ${REQUIRED_PACKAGES} && rm -rf /var/lib/apt/lists/*

fm_weck/resources/__init__.py

@@ -6,6 +6,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import importlib.resources as pkg_resources
+from functools import cache
 from pathlib import Path
 
 from . import fm_tools, properties
@@ -19,16 +20,33 @@ RUNEXEC_SCRIPT = "runexec"
 
 
 def iter_fm_data():
-    for fm_data in pkg_resources.contents(fm_tools):
-        with pkg_resources.path(fm_tools, fm_data) as fake_context_path:
-            fm_data_path = Path(fake_context_path)
-            if fm_data_path.is_file() and (fm_data_path.name.endswith(".yml") or fm_data_path.name.endswith(".yaml")):
-                yield fm_data_path
+    for entry in pkg_resources.files(fm_tools).iterdir():
+        if entry.name.endswith((".yml", ".yaml")):
+            with pkg_resources.as_file(entry) as path:
+                fm_data_path = Path(path)
+                if fm_data_path.is_file():
+                    yield fm_data_path
 
 
 def iter_properties():
-    for prop in pkg_resources.contents(properties):
-        with pkg_resources.path(properties, prop) as fake_context_path:
-            prop_path = Path(fake_context_path)
+    for entry in pkg_resources.files(properties).iterdir():
+        with pkg_resources.as_file(entry) as path:
+            prop_path = Path(path)
             if prop_path.is_file():
                 yield prop_path
+
+
+@cache
+def fm_tools_choice_map():
+    ignore = {
+        "schema.yml",
+    }
+
+    actors = {actor_def.stem: actor_def for actor_def in iter_fm_data() if (actor_def.name not in ignore)}
+
+    return actors
+
+
+@cache
+def property_choice_map():
+    return {spec_path.stem: spec_path for spec_path in iter_properties() if spec_path.suffix != ".license"}