dda-devops-build/version.py

from enum import Enum
from file_handlers import FileHandler

class Version():

    def __init__(self, version_list, release_type):
        self.version_list = None
        self.version_string = None
        self.is_snapshot = None        

    def increment(self, release_type: ReleaseType):
        self.is_snapshot = False
        match release_type:
            case ReleaseType.BUMP:
                self.is_snapshot = True
                self.version_list[ReleaseType.PATCH.value] += 1
            case ReleaseType.SNAPSHOT:
                self.is_snapshot = True
            case ReleaseType.PATCH:
                self.version_list[ReleaseType.PATCH.value] += 1
            case ReleaseType.MINOR:
                self.version_list[ReleaseType.PATCH.value] =  0
                self.version_list[ReleaseType.MINOR.value] += 1
            case ReleaseType.MAJOR:
                self.version_list[ReleaseType.PATCH.value] =  0
                self.version_list[ReleaseType.MINOR.value] =  0
                self.version_list[ReleaseType.MAJOR.value] += 1

    def get_version_string(self) -> str:
        self.version_string = ".".join([str(x) for x in self.version_list])
        if self.is_snapshot:
            self.version_string += "-SNAPSHOT"

    @classmethod
    def create_release_version(cls, version_list, release_type):
        inst = cls(version_list, release_type)
        if release_type == ReleaseType.PATCH:
            inst.is_snapshot = False
        else:
            inst.increment(release_type)
        return inst

    @classmethod
    def create_bump_version(cls, version_list):
        inst = cls(version_list, ReleaseType.BUMP)
        inst.increment(release_type)
        return inst