| 1 | #!/usr/bin/env python3 |
| 2 | import json |
| 3 | import os |
| 4 | from pathlib import Path |
| 5 | import subprocess |
| 6 | import sys |
| 7 | |
| 8 | |
| 9 | data = json.load(sys.stdin) |
| 10 | if Path(data["hostRoot"]).name == "local-ai": |
| 11 | source = Path(__file__).with_name("service.pkl") |
| 12 | path = Path(subprocess.check_output([ |
| 13 | "pkl", "eval", "-f", "json", "-p", "serviceId=local-ai", "-p", f"uid={data['uid']}", |
| 14 | "-p", "preview=false", "--expression", 'container.volumes["/cache/ssd"].src', str(source), |
| 15 | ], text=True).strip()) |
| 16 | path.mkdir(mode=0o700, parents=True, exist_ok=True) |
| 17 | os.chown(path, data["uid"], data["uid"]) |
| 18 | path.chmod(0o700) |