52 lines
1.4 KiB
Python
Executable File
52 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import os
|
|
import re
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
|
|
def main():
|
|
default_branch = "main"
|
|
branch = subprocess.check_output("git rev-parse --abbrev-ref HEAD", shell=True).decode().strip()
|
|
if branch != default_branch:
|
|
print(f"Aborting, not on '{default_branch}' branch.")
|
|
return
|
|
|
|
project_path = Path(__file__).resolve().parent
|
|
version_filepath = project_path / project_path.name / "__init__.py"
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("level", type=str, choices=["patch", "minor", "major"])
|
|
args = parser.parse_args()
|
|
|
|
with open(version_filepath) as f:
|
|
file_content = f.read()
|
|
|
|
version = re.search(r'__version__ = "(.+?)"', file_content).group(1)
|
|
major, minor, patch = map(int, version.split(sep="."))
|
|
|
|
if args.level == "patch":
|
|
patch += 1
|
|
elif args.level == "minor":
|
|
minor += 1
|
|
patch = 0
|
|
elif args.level == "major":
|
|
major += 1
|
|
minor = 0
|
|
patch = 0
|
|
|
|
new_version = f"{major}.{minor}.{patch}"
|
|
|
|
with open(version_filepath, "w") as f:
|
|
f.write(re.sub(r'__version__ = "(.+?)"', f'__version__ = "{new_version}"', file_content))
|
|
|
|
os.system(f"git commit {version_filepath} -m 'Updating for version {new_version}'")
|
|
os.system(f"git tag -a {new_version} -m 'Release {new_version}'")
|
|
os.system("git push --follow-tags")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|