asp-chef-cli 0.5.4__tar.gz → 0.5.6__tar.gz

{asp_chef_cli-0.5.4 → asp_chef_cli-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: asp-chef-cli
-Version: 0.5.4
+Version: 0.5.6
 Summary: A simple CLI to run ASP Chef recipes
 Author: Mario Alviano
 Author-email: mario.alviano@gmail.com

{asp_chef_cli-0.5.4 → asp_chef_cli-0.5.6}/asp_chef_cli/server/routers/dumbo.py

@@ -1,7 +1,9 @@
 import json as json_module
 import os
+import re
 import shutil
 import subprocess
+import tempfile
 from collections import defaultdict
 from typing import Optional, Dict, Final
 
@@ -18,6 +20,12 @@ from ..dependencies import *
 router = APIRouter()
 
 
+def strip_ansi_codes(text):
+    # This regex matches standard ANSI escape sequences
+    ansi_regex = re.compile(r'\x1b\[[0-9;]*[a-zA-Z]')
+    return ansi_regex.sub('', text)
+
+
 @endpoint(router, "/to-zero-simplification-version/")
 async def _(json):
     program = SymbolicProgram.parse(json["program"])
@@ -228,6 +236,78 @@ async def _(json):
     return {"models": models}    
 
 
+pasta_process: Final[Dict[str, Optional[subprocess.Popen]]] = defaultdict(lambda: None)
+pasta_path: Final[str | None] = shutil.which("pastasolver")
+
+
+def pasta_terminate(uuid):
+    if pasta_process[uuid] is not None:
+        pasta_process[uuid].kill()
+
+
+@endpoint(router, "/pasta/")
+async def _(json):
+    global pasta_process
+
+    uuid = json["uuid"]
+    program = json["program"]
+    output_predicate = json["output_predicate"] or "__bounds__"
+    bound_multiplier = 0
+    try:
+        bound_multiplier = int(json["bound_multiplier"])
+        bound_multiplier = min(max(bound_multiplier, 0), 1_000_000)
+    except ValueError:
+        bound_multiplier = 0
+
+    timeout = json["timeout"]
+    if type(timeout) is not int or timeout < 1 or timeout >= 24 * 60 * 60:
+        timeout = 5
+
+    pasta_terminate(uuid)
+
+    with tempfile.NamedTemporaryFile(mode='w', delete=True) as temp_file:
+        temp_file.write(program)
+        temp_file.flush()
+
+        cmd = ["/bin/timeout", str(timeout), pasta_path, temp_file.name]
+        pasta_process[uuid] = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        out, err = pasta_process[uuid].communicate()
+
+    timeout_reached: Final = pasta_process[uuid].returncode == 124
+    pasta_process[uuid] = None
+
+    # report errors
+    output = out.decode()
+    lines = output.split('\n')
+    lines = [strip_ansi_codes(line) for line in lines]
+    if any(line.startswith("Error") for line in lines):
+        raise ValueError('\n' + '\n'.join(lines))
+    if timeout_reached:
+        lines = [line for line in lines if not line.startswith("Sig term")]
+
+    lines = [line for line in lines if line]
+    lb, ub = None, None
+    if len(lines) == 1:
+        lb = ub = float(lines[0].split("Lower probability == upper probability for the query: ")[1])
+    elif len(lines) == 2:
+        lb = float(lines[0].split("Lower probability for the query: ")[1])
+        ub = float(lines[1].split("Upper probability for the query: ")[1])
+
+    if lb is None or ub is None:
+        raise ValueError("Could not parse output from PASTA solver:\n" + '\n'.join(lines))
+
+    if bound_multiplier:
+        lb = round(lb * bound_multiplier)
+        ub = round(ub * bound_multiplier)
+    else:
+        lb = f'real("{lb}")'
+        ub = f'real("{ub}")'
+
+    return {"models": [
+        [f'{output_predicate}({lb},{ub})']
+    ]}
+
+
 @endpoint(router, "/template/core-template/")
 async def _(json):
     if not json:

{asp_chef_cli-0.5.4 → asp_chef_cli-0.5.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "asp-chef-cli"
-version = "0.5.4"
+version = "0.5.6"
 description = "A simple CLI to run ASP Chef recipes"
 authors = ["Mario Alviano <mario.alviano@gmail.com>"]
 readme = "README.md"