Squashed 'ai-context/trustgraph-templates/' content from commit 42a5fd1b

git-subtree-dir: ai-context/trustgraph-templates
git-subtree-split: 42a5fd1b678f32be378062e30451e2052ccb95dd

trustgraph_configurator/index.py

@@ -0,0 +1,121 @@
+
+import dataclasses
+import importlib
+import json
+
+@dataclasses.dataclass
+class Platform:
+    name: str
+    description: str
+
+@dataclasses.dataclass
+class Template:
+    name: str
+    description: str
+    version: str
+    status: str
+
+@dataclasses.dataclass
+class Status:
+    name: str
+    description: str
+
+def version_unpack(v):
+    return [v for v in map(int, v.split('.'))]
+
+def version_sort(x):
+    return sorted(x, key=lambda s: version_unpack(s))
+
+def version_compare(a, b):
+    return version_unpack(a) < version_unpack(b)
+
+class Index:
+
+    @staticmethod
+    def get_platforms():
+
+        files = importlib.resources.files()
+        index = files.joinpath("templates").joinpath("index.json")
+
+        with open(index) as f:
+            ix = json.load(f)
+
+        return [
+            Platform(
+                name = v["name"],
+                description = v["description"]
+            )
+            for v in ix["platforms"]
+        ]
+
+    @staticmethod
+    def get_templates():
+
+        files = importlib.resources.files()
+        index = files.joinpath("templates").joinpath("index.json")
+
+        with open(index) as f:
+            ix = json.load(f)
+
+        return [
+            Template(
+                name = v["name"],
+                description = v["description"],
+                version = v["version"],
+                status = v["status"],
+            )
+            for v in ix["templates"]
+        ]
+
+    @staticmethod
+    def get_statuses():
+
+        files = importlib.resources.files()
+        index = files.joinpath("templates").joinpath("index.json")
+
+        with open(index) as f:
+            ix = json.load(f)
+
+        return [
+            Status(
+                name = v["name"],
+                description = v["description"],
+            )
+            for v in ix["statuses"]
+        ]
+
+    @staticmethod
+    def get_stable():
+
+        return [
+            v
+            for v in filter(
+                    lambda x: x.status == "stable", Index.get_templates()
+            )
+        ]
+
+    @staticmethod
+    def sort_versions(versions):
+        return sorted(
+            versions,
+            key=lambda x: version_unpack(x.version)
+        )
+
+    @staticmethod
+    def get_latest():
+        v = Index.sort_versions(Index.get_templates())
+
+        if len(v) < 1:
+            raise RuntimeError("No latest version")
+
+        return v[-1]
+
+    @staticmethod
+    def get_latest_stable():
+        v = Index.sort_versions(Index.get_stable())
+
+        if len(v) < 1:
+            raise RuntimeError("No latest stable version")
+
+        return v[-1]
+