119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
"""
|
|
Adapted from https://github.com/pygame/pygameweb/blob/master/pygameweb/builds/update_version_from_git.py
|
|
|
|
For updating the version from git.
|
|
__init__.py contains a __version__ field.
|
|
Update that.
|
|
If we are on master, we want to update the version as a pre-release.
|
|
git describe --tags
|
|
With these:
|
|
__init__.py
|
|
__version__= '0.0.2'
|
|
git describe --tags
|
|
0.0.1-22-g729a5ae
|
|
We want this:
|
|
__init__.py
|
|
__version__= '0.0.2.dev22.g729a5ae'
|
|
Get the branch/tag name with this.
|
|
git symbolic-ref -q --short HEAD || git describe --tags --exact-match
|
|
"""
|
|
|
|
import io
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
|
|
def migrate_source_attribute(attr, to_this, target_file, regex):
|
|
"""Updates __magic__ attributes in the source file"""
|
|
change_this = re.compile(regex, re.S)
|
|
new_file = []
|
|
found = False
|
|
|
|
with open(target_file, 'r') as fp:
|
|
lines = fp.readlines()
|
|
|
|
for line in lines:
|
|
if line.startswith(attr):
|
|
found = True
|
|
line = re.sub(change_this, to_this, line)
|
|
new_file.append(line)
|
|
|
|
if found:
|
|
with open(target_file, 'w') as fp:
|
|
fp.writelines(new_file)
|
|
|
|
def migrate_version(target_file, new_version):
|
|
"""Updates __version__ in the source file"""
|
|
regex = r"['\"](.*)['\"]"
|
|
migrate_source_attribute('__version__', "'%s'" % new_version, target_file, regex)
|
|
|
|
|
|
def is_master_branch():
|
|
cmd = ('git rev-parse --abbrev-ref HEAD')
|
|
tag_branch = subprocess.check_output(cmd, shell=True)
|
|
return tag_branch in [b'master\n']
|
|
|
|
def git_tag_name():
|
|
cmd = ('git describe --tags')
|
|
tag_branch = subprocess.check_output(cmd, shell=True)
|
|
tag_branch = tag_branch.decode().strip()
|
|
return tag_branch
|
|
|
|
def get_git_version_info():
|
|
cmd = 'git describe --tags'
|
|
ver_str = subprocess.check_output(cmd, shell=True)
|
|
ver, commits_since, githash = ver_str.decode().strip().split('-')
|
|
return ver, commits_since, githash
|
|
|
|
def prerelease_version():
|
|
""" return what the prerelease version should be.
|
|
https://packaging.python.org/tutorials/distributing-packages/#pre-release-versioning
|
|
0.0.2.dev22
|
|
"""
|
|
ver, commits_since, githash = get_git_version_info()
|
|
initpy_ver = get_version()
|
|
|
|
assert len(initpy_ver.split('.')) in [3, 4], 'moto/__init__.py version should be like 0.0.2 or 0.0.2.dev'
|
|
assert initpy_ver > ver, 'the moto/__init__.py version should be newer than the last tagged release.'
|
|
return f'{initpy_ver}.dev{commits_since}'
|
|
|
|
def read(*parts):
|
|
""" Reads in file from *parts.
|
|
"""
|
|
try:
|
|
return io.open(os.path.join(*parts), 'r', encoding='utf-8').read()
|
|
except IOError:
|
|
return ''
|
|
|
|
def get_version():
|
|
""" Returns version from moto/__init__.py
|
|
"""
|
|
version_file = read('moto', '__init__.py')
|
|
version_match = re.search(r'^__version__ = [\'"]([^\'"]*)[\'"]',
|
|
version_file, re.MULTILINE)
|
|
if version_match:
|
|
return version_match.group(1)
|
|
raise RuntimeError('Unable to find version string.')
|
|
|
|
|
|
def release_version_correct():
|
|
"""Makes sure the:
|
|
- prerelease verion for master is correct.
|
|
- release version is correct for tags.
|
|
"""
|
|
if is_master_branch():
|
|
# update for a pre release version.
|
|
initpy = os.path.abspath("moto/__init__.py")
|
|
|
|
new_version = prerelease_version()
|
|
print(f'updating version in __init__.py to {new_version}')
|
|
migrate_version(initpy, new_version)
|
|
else:
|
|
# check that we are a tag with the same version as in __init__.py
|
|
assert get_version() == git_tag_name(), 'git tag/branch name not the same as moto/__init__.py __verion__'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
release_version_correct()
|