terraformDummyRepo/devops_test.py
2023-02-09 11:43:10 +01:00

110 lines
No EOL
3.7 KiB
Python

import json
from enum import Enum
import re
def init_project():
# validate_values()
version = Version('build.gradle')
version.parse()
version.increment(ReleaseLevel.SNAPSHOT)
version.write()
def prepare_release():
pass
def release_in_git():
pass
class ReleaseLevel(Enum):
MAJOR = 0
MINOR = 1
PATCH = 2
SNAPSHOT = 3
class Version():
def __init__(self, config_file_path):
self.version = []
self.is_snapshot = False
self.config_file_path = config_file_path
self.config_file_type = config_file_path.split('.')[-1]
def parse(self):
match self.config_file_type:
case 'json':
self.__parse_json()
case 'gradle':
self.__parse_gradle()
def __parse_json(self):
with open(self.config_file_path, 'r') as json_file:
json_version = json.load(json_file)['version']
if '-SNAPSHOT' in json_version:
self.is_snapshot = True
json_version = json_version.replace('-SNAPSHOT', '')
self.version = [int(x) for x in json_version.split('.')]
def __parse_gradle(self):
with open(self.config_file_path, 'r') as gradle_file:
contents = gradle_file.read()
version_line = re.search("\nversion = .*\n", contents)
if version_line is None:
raise Exception("Version not found in gradle file")
version_line = version_line.group()
version_string = re.search('[0-9]*\.[0-9]*\.[0-9]*(-SNAPSHOT)?', version_line)
if version_string is None:
raise Exception("Version not found in gradle file")
version_string = version_string.group()
if '-SNAPSHOT' in version_string:
self.is_snapshot = True
version_string = version_string.replace('-SNAPSHOT', '')
self.version = [int(x) for x in version_string.split('.')]
def increment(self, level: ReleaseLevel):
self.is_snapshot = False
match level:
case ReleaseLevel.SNAPSHOT:
self.is_snapshot = True
case ReleaseLevel.PATCH:
self.version[ReleaseLevel.PATCH.value] += 1
case ReleaseLevel.MINOR:
self.version[ReleaseLevel.PATCH.value] = 0
self.version[ReleaseLevel.MINOR.value] += 1
case ReleaseLevel.MAJOR:
self.version[ReleaseLevel.PATCH.value] = 0
self.version[ReleaseLevel.MINOR.value] = 0
self.version[ReleaseLevel.MAJOR.value] += 1
def write(self):
match self.config_file_type:
case 'json':
self.__write_json()
case 'gradle':
self.__write_gradle()
case _:
raise Exception(f'The file type "{self.config_file_type}" is not implemented')
def __write_json(self):
with open(self.config_file_path, 'r+') as json_file:
json_data = json.load(json_file)
json_data['version'] = self.get()
json_file.seek(0)
json.dump(json_data, json_file, indent=4)
json_file.truncate()
def __write_gradle(self):
with open(self.config_file_path, 'r+') as gradle_file:
gradle_contents = gradle_file.read()
version_substitute = re.sub("\nversion = .*\n", f'\nversion = "{self.get()}"\n', gradle_contents)
gradle_file.seek(0)
gradle_file.write(version_substitute)
gradle_file.truncate()
def get(self) -> str:
version_string = ".".join([str(x) for x in self.version])
if self.is_snapshot:
version_string += "-SNAPSHOT"
return version_string