44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import Protocol
|
||
|
|
|
||
|
|
SCRIPT_TIMEOUT = 60
|
||
|
|
OUTPUT_LIMIT = 4000
|
||
|
|
SAFE_PATH = "/usr/local/bin:/usr/bin:/bin"
|
||
|
|
RADIO = "kvb-org-radio/scripts/radio.py"
|
||
|
|
RASCI = "kvb-org-rasci/scripts/rasci.py"
|
||
|
|
NIGHT = "kvb-org-sunset/scripts/night.py"
|
||
|
|
OPERATOR_SCRIPT = "kvb-org-operator/scripts/operator.py"
|
||
|
|
|
||
|
|
|
||
|
|
class Runner(Protocol):
|
||
|
|
def run(self, script: str, args: list[str]) -> dict: ...
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class Scripts:
|
||
|
|
skills: Path
|
||
|
|
council: Path
|
||
|
|
home: Path
|
||
|
|
|
||
|
|
def run(self, script: str, args: list[str]) -> dict:
|
||
|
|
command = " ".join([Path(script).name, *args])
|
||
|
|
env = {"PATH": SAFE_PATH, "HOME": str(self.home), "AGENTS_COUNCIL_ROOT": str(self.council), "PYTHONDONTWRITEBYTECODE": "1"}
|
||
|
|
try:
|
||
|
|
done = subprocess.run(
|
||
|
|
[sys.executable, str(self.skills / script), *args],
|
||
|
|
capture_output=True,
|
||
|
|
encoding="utf-8",
|
||
|
|
errors="replace",
|
||
|
|
timeout=SCRIPT_TIMEOUT,
|
||
|
|
env=env,
|
||
|
|
stdin=subprocess.DEVNULL,
|
||
|
|
check=False,
|
||
|
|
)
|
||
|
|
except (OSError, subprocess.TimeoutExpired) as error:
|
||
|
|
return {"command": command, "code": -1, "stdout": "", "stderr": str(error)}
|
||
|
|
return {"command": command, "code": done.returncode, "stdout": done.stdout[-OUTPUT_LIMIT:], "stderr": done.stderr[-OUTPUT_LIMIT:]}
|