server_version.py

from __future__ import annotations

import re


class InvalidServerVersionError(Exception):
    pass


class ServerVersion:
    """
    A representation of the version of the Neo4j Graph Data Science library installed on the server.
    """

    def __init__(self, major: int, minor: int, patch: int):
        self.major = major
        self.minor = minor
        self.patch = patch

    @classmethod
    def from_string(cls, version: str) -> ServerVersion:
        server_version_match = re.search(r"^(\d+)\.(\d+)\.(\d+)", version)
        if not server_version_match:
            raise InvalidServerVersionError(f"{version} is not a valid GDS library version")

        return cls(*map(int, server_version_match.groups()))

    def __lt__(self, other: ServerVersion) -> bool:
        if self.major != other.major:
            return self.major < other.major

        if self.minor != other.minor:
            return self.minor < other.minor

        if self.patch != other.patch:
            return self.patch < other.patch

        return False

    def __ge__(self, other: ServerVersion) -> bool:
        return not self.__lt__(other)

    def __str__(self) -> str:
        return f"{self.major}.{self.minor}.{self.patch}"