terraformDummyRepo/git_repository.py

import os
import subprocess as sub
from pathlib import Path
from system_repository import SystemRepository
from release_type import ReleaseType

class GitRepository():

    def __init__(self):
        self.latest_commit = None
        self.system_repository = SystemRepository()

    @classmethod
    def create_from_commit_string(cls, commit_string):
        inst = cls()
        inst.latest_commit = commit_string
        return inst

    def get_latest_n_commits(self, n: int):
        self.system_repository.run_checked('git', 'log', '--oneline', '--format="%s %b"', f'-n {n}')
        return self.system_repository.stdout

    def get_latest_commit(self):
        output = self.get_latest_n_commits(1)
        self.latest_commit = " ".join(output) # returns a list of strings otherwise
        return self.latest_commit

    def get_release_type_from_latest_commit(self):
        if self.latest_commit is None:
            self.get_latest_commit()

        if ReleaseType.MAJOR.name in self.latest_commit.upper():
            return ReleaseType.MAJOR
        elif ReleaseType.MINOR.name in self.latest_commit.upper():
            return ReleaseType.MINOR
        elif ReleaseType.PATCH.name in self.latest_commit.upper():
            return ReleaseType.PATCH
        elif ReleaseType.SNAPSHOT.name in self.latest_commit.upper():
            return ReleaseType.SNAPSHOT
        else:
            return None

    def tag_annotated(self, annotation: str, message: str):
        self.system_repository.run_checked('git', 'tag', '-a', annotation, '-m', message)
        return self.system_repository.stdout

    def tag_annotated(self, annotation: str, message: str, count: int):
        self.system_repository.run_checked('git', 'tag', '-a', annotation, '-m', message, f'HEAD~{count}')
        return self.system_repository.stdout

    def get_current_branch(self):
        self.system_repository.run_checked('git', 'branch', '--show-current')
        return ''.join(self.system_repository.stdout).rstrip()

    def add_file(self, file_path: Path):
        self.system_repository.run_checked('git', 'add', file_path)
        return self.system_repository.stdout

    def commit(self, commit_message: str):
        self.system_repository.run_checked('git', 'commit', '-m', commit_message)
        return self.system_repository.stdout

    def push(self):
        self.system_repository.run_checked('git', 'push')
        return self.system_repository.stdout

    def checkout(self, branch: str):
        self.system_repository.run_checked('git', 'checkout', branch)
        return self.system_repository.stdout