Skip to content

CosmoTech_Acceleration_Library.Modelops.core.common.graph_handler

GraphHandler

Bases: RedisHandler

Class that handle Graph Redis information

Source code in CosmoTech_Acceleration_Library/Modelops/core/common/graph_handler.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class GraphHandler(RedisHandler):
    """
    Class that handle Graph Redis information
    """

    def __init__(self, host: str, port: int, name: str, password: str = None, source_url: str = "", graph_rotation: int = 3):
        super().__init__(host=host, port=port, name=name, password=password)
        logger.debug("GraphHandler init")
        self.graph = self.r.graph(name)
        self.name = name
        self.m_metadata = ModelMetadata(host=host, port=port, name=name, password=password)
        current_metadata = self.m_metadata.get_metadata()
        if not current_metadata:
            if graph_rotation is None:
                graph_rotation = 3
            logger.debug("Create metadata key")
            self.m_metadata.set_metadata(last_graph_version=0, graph_source_url=source_url,
                                         graph_rotation=graph_rotation)

RotatedGraphHandler

Bases: VersionedGraphHandler

Class that handle Rotated Graph Redis information

Source code in CosmoTech_Acceleration_Library/Modelops/core/common/graph_handler.py 
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
class RotatedGraphHandler(VersionedGraphHandler):
    """
    Class that handle Rotated Graph Redis information
    """

    def __init__(self, host: str, port: int, name: str, password: str = None, version: int = None, source_url: str = "",
                 graph_rotation: int = None):
        super().__init__(host=host, port=port, name=name, password=password, source_url=source_url, version=version,
                         graph_rotation=graph_rotation)
        logger.debug("RotatedGraphHandler init")
        self.graph_rotation = self.m_metadata.get_graph_rotation()

    def get_all_versions(self):
        matching_graph_keys = self.r.keys(ModelUtil.build_graph_key_pattern(self.name))
        versions = []
        for graph_key in matching_graph_keys:
            versions.append(graph_key.split(":")[-1])
        return versions

    def handle_graph_rotation(func):
        """
        Decorator to do stuff then handle graph rotation (delete the oldest graph if the amount of graph is greater than graph rotation)
        """

        def handle(self, *args, **kwargs):
            graph_versions = self.get_all_versions()

            if len(graph_versions) > 0:
                max_version = max([int(x) for x in graph_versions if x.isnumeric()])
            else:
                max_version = 0
            # upgrade current graph to max_version+1
            self.version = max_version + 1
            self.version_name = ModelUtil.build_graph_version_name(self.name, self.version)
            self.graph = self.r.graph(self.version_name)
            logger.debug(f'Using graph updated version {self.version_name}')

            # do function on new graph
            func(self, *args, **kwargs)

            # get max version to manage case func not using (hence creating) graph
            graph_versions = [int(v) for v in self.get_all_versions()]
            graph_versions.sort()
            graph_versions.reverse()
            to_remove = graph_versions[int(self.graph_rotation):]

            # remove all older versions
            for v in to_remove:
                oldest_graph_version_to_delete = ModelUtil.build_graph_version_name(self.name, v)
                self.r.delete(oldest_graph_version_to_delete)
                logger.debug(f"Graph {oldest_graph_version_to_delete} deleted")

            # upgrade metadata last version to +1 after function execution
            self.m_metadata.set_last_graph_version(self.version)

        return handle

handle_graph_rotation(func)

Decorator to do stuff then handle graph rotation (delete the oldest graph if the amount of graph is greater than graph rotation)

Source code in CosmoTech_Acceleration_Library/Modelops/core/common/graph_handler.py 
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
def handle_graph_rotation(func):
    """
    Decorator to do stuff then handle graph rotation (delete the oldest graph if the amount of graph is greater than graph rotation)
    """

    def handle(self, *args, **kwargs):
        graph_versions = self.get_all_versions()

        if len(graph_versions) > 0:
            max_version = max([int(x) for x in graph_versions if x.isnumeric()])
        else:
            max_version = 0
        # upgrade current graph to max_version+1
        self.version = max_version + 1
        self.version_name = ModelUtil.build_graph_version_name(self.name, self.version)
        self.graph = self.r.graph(self.version_name)
        logger.debug(f'Using graph updated version {self.version_name}')

        # do function on new graph
        func(self, *args, **kwargs)

        # get max version to manage case func not using (hence creating) graph
        graph_versions = [int(v) for v in self.get_all_versions()]
        graph_versions.sort()
        graph_versions.reverse()
        to_remove = graph_versions[int(self.graph_rotation):]

        # remove all older versions
        for v in to_remove:
            oldest_graph_version_to_delete = ModelUtil.build_graph_version_name(self.name, v)
            self.r.delete(oldest_graph_version_to_delete)
            logger.debug(f"Graph {oldest_graph_version_to_delete} deleted")

        # upgrade metadata last version to +1 after function execution
        self.m_metadata.set_last_graph_version(self.version)

    return handle

VersionedGraphHandler

Bases: GraphHandler

Class that handle Versioned Graph Redis information

Source code in CosmoTech_Acceleration_Library/Modelops/core/common/graph_handler.py 
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
class VersionedGraphHandler(GraphHandler):
    """
    Class that handle Versioned Graph Redis information
    """

    def __init__(self, host: str, port: int, name: str, version: int = None, password: str = None, source_url: str = "",
                 graph_rotation: int = None):
        super().__init__(host=host, port=port, name=name, password=password, source_url=source_url,
                         graph_rotation=graph_rotation)
        logger.debug("VersionedGraphHandler init")
        self.version = version
        if version is None:
            self.version = self.m_metadata.get_last_graph_version()
        self.versioned_name = ModelUtil.build_graph_version_name(self.name, self.version)
        self.graph = self.r.graph(self.versioned_name)