pyzebra/make_release.py

43 lines
1.1 KiB
Python

#!/usr/bin/env python3
import argparse
import os
import re
def main():
filepath = "pyzebra/__init__.py"
parser = argparse.ArgumentParser()
parser.add_argument("level", type=str, choices=["patch", "minor", "major"])
parser.add_argument("tag_msg", type=str, help="tag message")
args = parser.parse_args()
with open(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(filepath, "w") as f:
f.write(re.sub(r'__version__ = "(.*?)"', f'__version__ = "{new_version}"', file_content))
os.system(f"git commit {filepath} -m 'Updating for version {new_version}'")
os.system(f"git tag -a {new_version} -m '{args.tag_msg}'")
if __name__ == "__main__":
main()