trustgraph/trustgraph_configurator/index.py

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]