horus-singularity 0.0.1__py3-none-any.whl

horus_singularity/__init__.py

@@ -0,0 +1,6 @@
+#
+# horus_singularity
+# Copyright (c) 2026 Temple Compute
+#
+# MIT License
+#

horus_singularity/executor/__init__.py

@@ -0,0 +1,6 @@
+#
+# horus_singularity
+# Copyright (c) 2026 Temple Compute
+#
+# MIT License
+#

horus_singularity/executor/singularity.py

@@ -0,0 +1,190 @@
+#
+# horus_singularity
+# Copyright (c) 2026 Temple Compute
+#
+# MIT License
+#
+"""
+Singularity/Apptainer executor implementation for Horus.
+"""
+
+import asyncio
+import shlex
+from typing import TYPE_CHECKING, Any, ClassVar
+
+from horus_builtin.runtime.command import CommandRuntime
+from horus_runtime.core.executor.base import BaseExecutor, RuntimeFilterType
+from horus_runtime.core.task.exceptions import TaskExecutionError
+from horus_runtime.logging import horus_logger
+from pydantic import Field, PrivateAttr
+
+from horus_singularity.i18n import tr as _
+
+if TYPE_CHECKING:
+    from horus_runtime.core.task.base import BaseTask
+
+
+class SingularityExecutor(BaseExecutor):
+    """
+    Runs the task's command inside a Singularity/Apptainer container.
+    """
+
+    kind: str = "singularity"
+    kind_name: ClassVar[str] = "Singularity Executor"
+    kind_description: ClassVar[str] = _(
+        "Executes a command inside a Singularity/Apptainer container on the "
+        "task target."
+    )
+
+    runtimes: ClassVar[RuntimeFilterType] = (CommandRuntime,)
+
+    image: str
+    """
+    Path to the ``.sif`` image file on the target (e.g.
+    ``/opt/images/boltz.sif``).  Unlike Docker there is no build step: the
+    image must already exist on the target.
+    """
+
+    exe: str = "singularity"
+    """
+    Container CLI to invoke — ``singularity`` or ``apptainer``.
+    """
+
+    binds: dict[str, str] = Field(default_factory=dict)
+    """
+    Bind mounts as ``host_path -> container_path`` (``--bind`` flag).
+    """
+
+    nv: bool = False
+    """
+    Add ``--nv`` so the NVIDIA driver stack is exposed inside the container.
+    """
+
+    env: dict[str, str] = Field(default_factory=dict)
+    """
+    Environment variables to set inside the container, as ``NAME -> value``.
+    """
+
+    working_dir: str | None = None
+    """
+    Working directory inside the container (``--pwd`` flag).
+    """
+
+    extra_args: list[str] = Field(default_factory=list)
+    """
+    Extra flags passed verbatim to ``<exe> exec`` before the image path.
+    """
+
+    _proc: Any = PrivateAttr(default=None)
+    """
+    Reference to the process started in :meth:`_execute` so
+    :meth:`cancel_execution` can kill it.
+    """
+
+    def _singularity_exec_cmd(
+        self, prepared_command: str, task: "BaseTask | None" = None
+    ) -> str:
+        """Return the full ``singularity exec`` CLI command string."""
+        # ponytail: auto-mount artifact parent dirs; explicit binds win
+        auto_mounts: dict[str, str] = {}
+        if task is not None:
+            for artifact in (*task.inputs, *task.outputs):
+                host_dir = str(artifact.path.parent)
+                auto_mounts[host_dir] = host_dir
+        merged_binds = {**auto_mounts, **self.binds}
+
+        parts = [shlex.quote(self.exe), "exec"]
+        if self.nv:
+            parts.append("--nv")
+        for k, v in self.env.items():
+            parts += ["--env", shlex.quote(f"{k}={v}")]
+        for host, container in merged_binds.items():
+            parts += ["--bind", shlex.quote(f"{host}:{container}")]
+        if self.working_dir:
+            parts += ["--pwd", shlex.quote(self.working_dir)]
+        parts += [shlex.quote(arg) for arg in self.extra_args]
+        parts += [
+            shlex.quote(self.image),
+            "/bin/sh",
+            "-c",
+            shlex.quote(prepared_command),
+        ]
+        return " ".join(parts)
+
+    async def _execute(self, task: "BaseTask") -> None:
+        """
+        Run the task's command inside a Singularity/Apptainer container on
+        the target.
+        """
+        if not isinstance(task.runtime, CommandRuntime):
+            raise TaskExecutionError(
+                _("SingularityExecutor only supports CommandRuntime runtimes.")
+            )
+        prepared_command = await task.runtime.setup_runtime(task)
+
+        exec_cmd = self._singularity_exec_cmd(prepared_command, task)
+        horus_logger.log.debug(
+            _(
+                "Running task %(task_id)s in Singularity image %(image)s: "
+                "%(command)s"
+            )
+            % {
+                "task_id": task.id,
+                "image": self.image,
+                "command": prepared_command,
+            }
+        )
+
+        proc = await task.target.run_command(exec_cmd, cwd=task.working_dir)
+        self._proc = proc
+
+        try:
+            stdout, stderr = await proc.communicate()
+        except asyncio.CancelledError:
+            proc.kill()
+            await proc.wait()
+            raise
+
+        out = stdout.decode(errors="replace").strip() if stdout else ""
+        err = stderr.decode(errors="replace").strip() if stderr else ""
+        if out:
+            horus_logger.log.info(out)
+        if err:
+            horus_logger.log.warning(err)
+
+        try:
+            if proc.returncode != 0:
+                horus_logger.log.error(
+                    _(
+                        "Container for task %(task_id)s exited with code "
+                        "%(code)s. Output: %(out)s"
+                    )
+                    % {
+                        "task_id": task.id,
+                        "code": proc.returncode,
+                        "out": (out or err).strip(),
+                    }
+                )
+
+                raise TaskExecutionError(
+                    _("Container exited with code %(code)s")
+                    % {"code": proc.returncode}
+                )
+        finally:
+            self._proc = None
+
+    async def cancel_execution(self) -> None:
+        """Kill the running container process so it is not orphaned.
+
+        Singularity containers have no daemon-side name to stop, so the
+        process started in :meth:`_execute` is killed directly.  If nothing
+        is running (e.g. the task finished before the cancel arrived) this
+        is a safe no-op.
+        """
+        proc = self._proc
+        if proc is None:
+            return
+        self._proc = None  # clear before kill — idempotent
+        if proc.returncode is None:
+            proc.kill()
+            await proc.wait()

horus_singularity/i18n.py

@@ -0,0 +1,22 @@
+#
+# horus_singularity
+# Copyright (c) 2026 Temple Compute
+#
+# MIT License
+#
+"""
+Localization for horus_singularity.
+
+Import ``tr`` (aliased as ``_``) in any module that has user-visible strings::
+
+    from horus_singularity.i18n import tr as _
+
+    _("Something happened.")
+    _("%(n)s item processed", "%(n)s items processed", n=count)
+"""
+
+from pathlib import Path
+
+from horus_runtime.i18n import make_translator
+
+tr = make_translator("horus_singularity", Path(__file__).parent / "locale")

horus_singularity/locale/en/LC_MESSAGES/horus_singularity.po

@@ -0,0 +1,47 @@
+# English translations for horus_singularity.
+# Copyright (C) 2026 Temple Compute
+# This file is distributed under the same license as the horus_singularity
+# project.
+# Temple Compute, 2026.
+#
+msgid ""
+msgstr ""
+"Project-Id-Version: horus_singularity VERSION\n"
+"Report-Msgid-Bugs-To: EMAIL@ADDRESS\n"
+"POT-Creation-Date: 2026-07-22 23:17+0200\n"
+"PO-Revision-Date: 2026-07-22 23:32+0200\n"
+"Last-Translator: Temple Compute\n"
+"Language: en\n"
+"Language-Team: en <LL@li.org>\n"
+"Plural-Forms: nplurals=2; plural=(n != 1);\n"
+"MIME-Version: 1.0\n"
+"Content-Type: text/plain; charset=utf-8\n"
+"Content-Transfer-Encoding: 8bit\n"
+"Generated-By: Babel 2.18.0\n"
+
+#: src/horus_singularity/executor/singularity.py:35
+msgid ""
+"Executes a command inside a Singularity/Apptainer container on the task "
+"target."
+msgstr ""
+"Executes a command inside a Singularity/Apptainer container on the task "
+"target."
+
+#: src/horus_singularity/executor/singularity.py:121
+msgid "SingularityExecutor only supports CommandRuntime runtimes."
+msgstr "SingularityExecutor only supports CommandRuntime runtimes."
+
+#: src/horus_singularity/executor/singularity.py:128
+#, python-format
+msgid "Running task %(task_id)s in Singularity image %(image)s: %(command)s"
+msgstr "Running task %(task_id)s in Singularity image %(image)s: %(command)s"
+
+#: src/horus_singularity/executor/singularity.py:159
+#, python-format
+msgid "Container for task %(task_id)s exited with code %(code)s. Output: %(out)s"
+msgstr "Container for task %(task_id)s exited with code %(code)s. Output: %(out)s"
+
+#: src/horus_singularity/executor/singularity.py:170
+#, python-format
+msgid "Container exited with code %(code)s"
+msgstr "Container exited with code %(code)s"

horus_singularity/locale/messages.pot

@@ -0,0 +1,45 @@
+# Translations template for horus_singularity.
+# Copyright (C) 2026 Temple Compute
+# This file is distributed under the same license as the horus_singularity
+# project.
+# FIRST AUTHOR <EMAIL@ADDRESS>, 2026.
+#
+#, fuzzy
+msgid ""
+msgstr ""
+"Project-Id-Version: horus_singularity VERSION\n"
+"Report-Msgid-Bugs-To: EMAIL@ADDRESS\n"
+"POT-Creation-Date: 2026-07-22 23:17+0200\n"
+"PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
+"Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
+"Language-Team: LANGUAGE <LL@li.org>\n"
+"MIME-Version: 1.0\n"
+"Content-Type: text/plain; charset=utf-8\n"
+"Content-Transfer-Encoding: 8bit\n"
+"Generated-By: Babel 2.18.0\n"
+
+#: src/horus_singularity/executor/singularity.py:35
+msgid ""
+"Executes a command inside a Singularity/Apptainer container on the task "
+"target."
+msgstr ""
+
+#: src/horus_singularity/executor/singularity.py:121
+msgid "SingularityExecutor only supports CommandRuntime runtimes."
+msgstr ""
+
+#: src/horus_singularity/executor/singularity.py:128
+#, python-format
+msgid "Running task %(task_id)s in Singularity image %(image)s: %(command)s"
+msgstr ""
+
+#: src/horus_singularity/executor/singularity.py:159
+#, python-format
+msgid "Container for task %(task_id)s exited with code %(code)s. Output: %(out)s"
+msgstr ""
+
+#: src/horus_singularity/executor/singularity.py:170
+#, python-format
+msgid "Container exited with code %(code)s"
+msgstr ""
+

horus_singularity-0.0.1.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: horus_singularity
+Version: 0.0.1
+Summary: A Horus runtime plugin that runs task commands inside Singularity/Apptainer containers.
+License-File: LICENSE
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.13
+Requires-Dist: horus-runtime>=0.2.0

horus_singularity-0.0.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+horus_singularity/__init__.py,sha256=F3A65u__Zvq08OpMNe7mBjx6POUhQn2O1zX1T192Dt4,76
+horus_singularity/i18n.py,sha256=AkCLpS_gr5zKRL-7v44Cdwm9PfgBtFoU833tzdo75Cw,493
+horus_singularity/executor/__init__.py,sha256=F3A65u__Zvq08OpMNe7mBjx6POUhQn2O1zX1T192Dt4,76
+horus_singularity/executor/singularity.py,sha256=kTZipZSvAWJnW3vIfVhLADE2DjdqzgaogrEmwTt_WrQ,6044
+horus_singularity/locale/messages.pot,sha256=kOSubLenV0qtWAVX6hf0HmVS1UToqL-cMhWBaj1dZGQ,1371
+horus_singularity/locale/en/LC_MESSAGES/horus_singularity.mo,sha256=S50hvFrSf8Ubqu4LMhqTdZcKwihBwqcJMGp9EbVR11I,1160
+horus_singularity/locale/en/LC_MESSAGES/horus_singularity.po,sha256=15zPLythKao2_iF-JBXjFFJX6mMohvQo0C7JP49uVBs,1712
+horus_singularity-0.0.1.dist-info/METADATA,sha256=AoJqkmDkOFORW0oOQrGoqqvMvPAV_JHxERTPKLZ0Vm0,339
+horus_singularity-0.0.1.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+horus_singularity-0.0.1.dist-info/entry_points.txt,sha256=9Sk4Qz_nvtRgbpfHunnwDvdCt-SwvVwZQOCwwZH9yU8,70
+horus_singularity-0.0.1.dist-info/licenses/LICENSE,sha256=6YWukf1Wwk94ltA2Xcjpy8rBuXnA9Yc8-51OPO2DMYo,1071
+horus_singularity-0.0.1.dist-info/RECORD,,

horus_singularity-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.31.0
+Root-Is-Purelib: true
+Tag: py3-none-any

horus_singularity-0.0.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[horus.executor]
+singularity = horus_singularity.executor.singularity

horus_singularity-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Temple Compute
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.