asp-chef-cli 0.4.20__tar.gz → 0.4.22__tar.gz

{asp_chef_cli-0.4.20 → asp_chef_cli-0.4.22}/PKG-INFO

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

{asp_chef_cli-0.4.20 → asp_chef_cli-0.4.22}/asp_chef_cli/server/routers/dumbo.py

@@ -158,6 +158,7 @@ async def _(json):
         cmd.append("--enumerate")
     pyqasp_process[uuid] = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     out, err = pyqasp_process[uuid].communicate(program.encode())
+    timeout_reached: Final = pyqasp_process[uuid].returncode == 124
     pyqasp_process[uuid] = None
 
     # report errors
@@ -165,6 +166,8 @@ async def _(json):
     lines = output.split('\n')
     if any(line.startswith("Error") for line in lines):
         raise ValueError(output)
+    if timeout_reached:
+        lines = [line for line in lines if not line.startswith("Sig term")]
 
     # return models
     models = [
@@ -174,6 +177,52 @@ async def _(json):
     return {"models": models}
 
 
+casper_process: Final[Dict[str, Optional[subprocess.Popen]]] = defaultdict(lambda: None)
+casper_path: Final[str | None] = shutil.which("casper")
+
+
+def casper_terminate(uuid):
+    if casper_process[uuid] is not None:
+        casper_process[uuid].kill()
+
+
+@endpoint(router, "/casper/")
+async def _(json):
+    global casper_process
+
+    uuid = json["uuid"]
+    program = json["program"]
+    enumerate = json["enumerate"]
+    timeout = json["timeout"]
+    if type(timeout) is not int or timeout < 1 or timeout >= 24 * 60 * 60:
+        timeout = 5
+
+    casper_terminate(uuid)
+
+    cmd = ["/bin/timeout", str(timeout), casper_path, "--problem", "/dev/stdin", "--json"]
+    if enumerate:
+        cmd.append("-n0")
+    casper_process[uuid] = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    out, err = casper_process[uuid].communicate(program.encode())
+    timeout_reached: Final = pyqasp_process[uuid].returncode == 124
+    casper_process[uuid] = None
+    
+    # report errors
+    output = out.decode()
+    lines = output.split('\n')
+    if any(line.startswith("Error") for line in lines):
+        raise ValueError(output)
+    if timeout_reached:
+        lines = [line for line in lines if not line.startswith("Sig term")]
+
+    # return models
+    models = [
+        [lit for lit in json_module.loads(line)['literals'] if not lit.startswith('not ')]
+        for line in lines if line
+    ]
+    return {"models": models}    
+
+
 @endpoint(router, "/template/core-template/")
 async def _(json):
     if not json:
@@ -215,4 +264,4 @@ async def _(json):
             "program": str(templates[key].program)
         })
 
-    return res
+    return res

{asp_chef_cli-0.4.20 → asp_chef_cli-0.4.22}/pyproject.toml

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