Initial version of update_version.py script to bump CVAT version (#4889)
Andrey Zhavoronkov
3 years ago
committed by
GitHub
parent
6410b86a4e
commit
4167a4e40c
@ -0,0 +1,211 @@
|
||||
import argparse
|
||||
from dataclasses import dataclass
|
||||
from pathlib import Path
|
||||
import re
|
||||
|
||||
|
||||
CVAT_VERSION_PATTERN = r'VERSION\s*=\s*\((\d+),\s*(\d*),\s*(\d+),\s*[\',\"](\w+)[\',\"],\s*(\d+)\)'
|
||||
COMPOSE_VERSION_PATTERN = r'(\$\{CVAT_VERSION:-)([\w.]+)(\})'
|
||||
|
||||
@dataclass()
|
||||
class Version:
|
||||
major: int = 0
|
||||
minor: int = 0
|
||||
patch: int = 0
|
||||
prerelease: str = ''
|
||||
prerelease_number: int = 0
|
||||
|
||||
def __str__(self) -> str:
|
||||
return f'{self.major}.{self.minor}.{self.patch}-{self.prerelease}.{self.prerelease_number}'
|
||||
|
||||
def cvat_repr(self):
|
||||
return f"({self.major}, {self.minor}, {self.patch}, '{self.prerelease}', {self.prerelease_number})"
|
||||
|
||||
def compose_repr(self):
|
||||
return f'v{self.major}.{self.minor}.{self.patch}'
|
||||
|
||||
def increment_prerelease_number(self) -> None:
|
||||
self.prerelease_number += 1
|
||||
|
||||
def increment_prerelease(self) -> None:
|
||||
flow = ('alpha', 'beta', 'rc', 'final')
|
||||
idx = flow.index(self.prerelease)
|
||||
if idx == len(flow) - 1:
|
||||
raise ValueError(f"Cannot increment current '{self.prerelease}' prerelease version")
|
||||
|
||||
self.prerelease = flow[idx + 1]
|
||||
self._set_default_prerelease_number()
|
||||
|
||||
def set_prerelease(self, value: str) -> None:
|
||||
values = ('alpha', 'beta', 'rc', 'final')
|
||||
if value not in values:
|
||||
raise ValueError(f'{value} is a wrong, must be one of {values}')
|
||||
|
||||
self.prerelease = value
|
||||
self._set_default_prerelease_number()
|
||||
|
||||
def increment_patch(self) -> None:
|
||||
self.patch += 1
|
||||
self._set_default_prerelease()
|
||||
|
||||
def increment_minor(self) -> None:
|
||||
self.minor += 1
|
||||
self._set_default_patch()
|
||||
|
||||
def increment_major(self) -> None:
|
||||
self.major += 1
|
||||
self._set_default_minor()
|
||||
|
||||
def _set_default_prerelease_number(self) -> None:
|
||||
self.prerelease_number = 0
|
||||
|
||||
def _set_default_prerelease(self) -> None:
|
||||
self.prerelease = 'alpha'
|
||||
self._set_default_prerelease_number()
|
||||
|
||||
def _set_default_patch(self) -> None:
|
||||
self.patch = 0
|
||||
self._set_default_prerelease()
|
||||
|
||||
def _set_default_minor(self) -> None:
|
||||
self.minor = 0
|
||||
self._set_default_patch()
|
||||
|
||||
def update_compose_config(new_version: Version) -> None:
|
||||
compose_file = get_compose_filename()
|
||||
with open(compose_file, 'r') as fp:
|
||||
compose_text = fp.read()
|
||||
|
||||
if new_version.prerelease == 'final':
|
||||
new_version_repr = new_version.compose_repr()
|
||||
else:
|
||||
new_version_repr = 'dev'
|
||||
|
||||
match = re.search(COMPOSE_VERSION_PATTERN, compose_text)
|
||||
if not match:
|
||||
raise RuntimeError('Cannot match version pattern')
|
||||
|
||||
if match[2] != new_version_repr:
|
||||
compose_text = re.sub(COMPOSE_VERSION_PATTERN, f'\\g<1>{new_version_repr}\\g<3>', compose_text)
|
||||
with open(compose_file, 'w') as fp:
|
||||
fp.write(compose_text)
|
||||
|
||||
print(f'\u2714 {compose_file} was updated. {match[2]} -> {new_version_repr}\n')
|
||||
|
||||
else:
|
||||
print(f'\u2714 {compose_file} no need to update.')
|
||||
|
||||
def update_cvat_version(old_version: str, new_version: Version) -> None:
|
||||
version_file = get_cvat_version_filename()
|
||||
with open(version_file, 'r') as fp:
|
||||
version_text = fp.read()
|
||||
|
||||
new_version_str = f'VERSION = {new_version.cvat_repr()}'
|
||||
version_text = version_text.replace(old_version, new_version_str)
|
||||
|
||||
with open(version_file, 'w') as fp:
|
||||
fp.write(version_text)
|
||||
|
||||
print(f'\u2714 {version_file} was updated. {old_version} -> {new_version_str}\n')
|
||||
|
||||
def verify_input(version_types: dict, args: dict) -> None:
|
||||
versions_to_bump = [args[v_type] for v_type in version_types]
|
||||
i = iter(versions_to_bump)
|
||||
if not any(i):
|
||||
raise ValueError(f'One of {list(version_types)} options must be specified')
|
||||
if any(i):
|
||||
raise ValueError(f'Only one of {list(version_types)} options accepted')
|
||||
|
||||
def get_cvat_version_filename() -> Path:
|
||||
return Path(__file__).resolve().parents[2] / 'cvat' / '__init__.py'
|
||||
|
||||
def get_compose_filename() -> Path:
|
||||
return Path(__file__).resolve().parents[2] / 'docker-compose.yml'
|
||||
|
||||
def get_current_version() -> tuple[str, Version]:
|
||||
version_file = get_cvat_version_filename()
|
||||
|
||||
with open(version_file, 'r') as fp:
|
||||
version_text = fp.read()
|
||||
|
||||
match = re.search(CVAT_VERSION_PATTERN, version_text)
|
||||
if not match:
|
||||
raise RuntimeError(f'Failed to find version in {version_file}')
|
||||
|
||||
version = Version(int(match[1]), int(match[2]), int(match[3]), match[4], int(match[5]))
|
||||
return match[0], version
|
||||
|
||||
def main() -> None:
|
||||
version_types = {
|
||||
'major': {
|
||||
'action': 'store_true',
|
||||
'help': 'Increment the existing major version by 1',
|
||||
},
|
||||
'minor': {
|
||||
'action': 'store_true',
|
||||
'help': 'Increment the existing minor version by 1',
|
||||
},
|
||||
'patch': {
|
||||
'action': 'store_true',
|
||||
'help': 'Increment the existing patch version by 1',
|
||||
},
|
||||
'prerelease': {
|
||||
'nargs': '?',
|
||||
'const': 'increment',
|
||||
'help': '''Increment prerelease version alpha->beta->rc->final,
|
||||
Also it's possible to pass value explicitly''',
|
||||
},
|
||||
'prerelease_number': {
|
||||
'action': 'store_true',
|
||||
'help': 'Increment prerelease number by 1',
|
||||
},
|
||||
}
|
||||
|
||||
parser = argparse.ArgumentParser(description='Bump CVAT version')
|
||||
|
||||
for v, v_config in version_types.items():
|
||||
parser.add_argument(f'--{v}', **v_config)
|
||||
|
||||
parser.add_argument('--current', '--show-current',
|
||||
action='store_true', help='Display current version')
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
version_str, version = get_current_version()
|
||||
|
||||
if args.current:
|
||||
print(version)
|
||||
return
|
||||
|
||||
try:
|
||||
verify_input(version_types, vars(args))
|
||||
except ValueError as e:
|
||||
print(f'\u2716 ERROR: {e}\n')
|
||||
parser.print_help()
|
||||
return
|
||||
|
||||
if args.prerelease_number:
|
||||
version.increment_prerelease_number()
|
||||
|
||||
elif args.prerelease:
|
||||
if args.prerelease == 'increment':
|
||||
version.increment_prerelease()
|
||||
else:
|
||||
version.set_prerelease(args.prerelease)
|
||||
|
||||
elif args.patch:
|
||||
version.increment_patch()
|
||||
|
||||
elif args.minor:
|
||||
version.increment_minor()
|
||||
|
||||
elif args.major:
|
||||
version.increment_major()
|
||||
|
||||
print(f'\u2714 Bump version to {version}\n')
|
||||
|
||||
update_cvat_version(version_str, version)
|
||||
update_compose_config(version)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in New Issue