Added bump project.version in pyproject.toml

Author: adamlui

remove-json-keys/utils/bump.py

@@ -1,35 +1,61 @@
+import argparse, re, sys
 from os import path
-import re, sys
 from types import SimpleNamespace as sns
 import tomli, tomli_w
 
 # Init logger
 sys.path.insert(0, path.join(path.dirname(__file__), '../src'))
 from remove_json_keys.lib import log # type: ignore
+msgs = sns(err_invalid_arg='You must pass --<major|minor|patch> as an argument.')
 
 # Init project data
 pyproject_path = path.join(path.dirname(__file__), '../pyproject.toml')
 log.info(f'Loading {pyproject_path}...')
 with open(pyproject_path, 'rb') as file : pyproject = tomli.load(file)
 project = sns(**pyproject['project'])
 
-def update_changelog_url():
-    new_ver_tag = f'{project.name}-{project.version}'
+def parse_args():
+    parser = argparse.ArgumentParser(description="Bump versions in pyproject.toml + README.md")
+    parser.add_argument('--major', action='store_true', help='Bump the major version')
+    parser.add_argument('--minor', action='store_true', help='Bump the minor version')
+    parser.add_argument('--patch', action='store_true', help='Bump the patch version')
+    return parser.parse_args()
+
+def bump_pyproject_versions(bump_type): # project.version + .urls['Changelog']
+
+    # Bump project.version
+    current_ver = project.version
+    major, minor, patch = map(int, current_ver.split('.'))
+    if   bump_type == 'major' : major += 1 ; minor = 0 ; patch = 0
+    elif bump_type == 'minor' : minor += 1 ; patch = 0
+    elif bump_type == 'patch' : patch += 1
+    else : raise ValueError(msgs.err_invalid_arg)
+    new_ver = f'{major}.{minor}.{patch}'
+    pyproject['project']['version'] = new_ver
+    with open(pyproject_path, 'wb') as file : tomli_w.dump(pyproject, file)
+    log.success(f"Bumped project.version in pyproject.toml from [{current_ver}] to [{new_ver}]")
+
+    # Bump version tag in Changelog URL
+    new_ver_tag = f'{project.name}-{new_ver}'
     new_changelog_url = f"{project.urls['Releases']}/tag/{new_ver_tag}"
-    log.data(f'Generated changelog URL: {new_changelog_url}')
+    log.data(f'Generated Changelog URL: {new_changelog_url}')
     log.info(f"{ 'Updating' if 'Changelog' in project.urls else 'Adding new' } Changelog URL in pyproject.toml...")
     project.urls['Changelog'] = new_changelog_url
-    pyproject['project'] = vars(project) # update og dict for dumping
     with open(pyproject_path, 'wb') as file : tomli_w.dump(pyproject, file)
     log.success(f'Bumped Changelog URL version tag to [{new_ver_tag}]!')
 
-def update_readme_versions():
+    return new_ver
+
+def update_readme_versions(new_ver): # in shield URLs
     readme_path = path.join(path.dirname(__file__), '../README.md')
     log.info('Updating versions in README.md...')
     with open(readme_path, 'r', encoding='utf-8') as file : readme_content = file.read()
-    updated_readme_content = re.sub(r'\d+\.\d+\.\d+', project.version, readme_content)
+    updated_readme_content = re.sub(r'\d+\.\d+\.\d+', new_ver, readme_content)
     with open(readme_path, 'w', encoding='utf-8') as file : file.write(updated_readme_content)
-    log.success(f'Updated versions in README URLs to [{project.version}]!')
+    log.success(f'Updated versions in README URLs to [{new_ver}]!')
 
-update_changelog_url()
-update_readme_versions()
+# Run MAIN routine
+args = parse_args()
+bump_type = 'major' if args.major else 'minor' if args.minor else 'patch' if args.patch else None
+if not bump_type : raise ValueError(msgs.err_invalid_arg)
+update_readme_versions(bump_pyproject_versions(bump_type))

translate-messages/utils/bump.py

@@ -1,35 +1,61 @@
+import argparse, re, sys
 from os import path
-import re, sys
 from types import SimpleNamespace as sns
 import tomli, tomli_w
 
 # Init logger
 sys.path.insert(0, path.join(path.dirname(__file__), '../src'))
 from translate_messages.lib import log # type: ignore
+msgs = sns(err_invalid_arg='You must pass --<major|minor|patch> as an argument.')
 
 # Init project data
 pyproject_path = path.join(path.dirname(__file__), '../pyproject.toml')
 log.info(f'Loading {pyproject_path}...')
 with open(pyproject_path, 'rb') as file : pyproject = tomli.load(file)
 project = sns(**pyproject['project'])
 
-def update_changelog_url():
-    new_ver_tag = f'{project.name}-{project.version}'
+def parse_args():
+    parser = argparse.ArgumentParser(description="Bump versions in pyproject.toml + README.md")
+    parser.add_argument('--major', action='store_true', help='Bump the major version')
+    parser.add_argument('--minor', action='store_true', help='Bump the minor version')
+    parser.add_argument('--patch', action='store_true', help='Bump the patch version')
+    return parser.parse_args()
+
+def bump_pyproject_versions(bump_type): # project.version + .urls['Changelog']
+
+    # Bump project.version
+    current_ver = project.version
+    major, minor, patch = map(int, current_ver.split('.'))
+    if   bump_type == 'major' : major += 1 ; minor = 0 ; patch = 0
+    elif bump_type == 'minor' : minor += 1 ; patch = 0
+    elif bump_type == 'patch' : patch += 1
+    else : raise ValueError(msgs.err_invalid_arg)
+    new_ver = f'{major}.{minor}.{patch}'
+    pyproject['project']['version'] = new_ver
+    with open(pyproject_path, 'wb') as file : tomli_w.dump(pyproject, file)
+    log.success(f"Bumped project.version in pyproject.toml from [{current_ver}] to [{new_ver}]")
+
+    # Bump version tag in Changelog URL
+    new_ver_tag = f'{project.name}-{new_ver}'
     new_changelog_url = f"{project.urls['Releases']}/tag/{new_ver_tag}"
-    log.data(f'Generated changelog URL: {new_changelog_url}')
+    log.data(f'Generated Changelog URL: {new_changelog_url}')
     log.info(f"{ 'Updating' if 'Changelog' in project.urls else 'Adding new' } Changelog URL in pyproject.toml...")
     project.urls['Changelog'] = new_changelog_url
-    pyproject['project'] = vars(project) # update og dict for dumping
     with open(pyproject_path, 'wb') as file : tomli_w.dump(pyproject, file)
     log.success(f'Bumped Changelog URL version tag to [{new_ver_tag}]!')
 
-def update_readme_versions():
+    return new_ver
+
+def update_readme_versions(new_ver): # in shield URLs
     readme_path = path.join(path.dirname(__file__), '../README.md')
     log.info('Updating versions in README.md...')
     with open(readme_path, 'r', encoding='utf-8') as file : readme_content = file.read()
-    updated_readme_content = re.sub(r'\d+\.\d+\.\d+', project.version, readme_content)
+    updated_readme_content = re.sub(r'\d+\.\d+\.\d+', new_ver, readme_content)
     with open(readme_path, 'w', encoding='utf-8') as file : file.write(updated_readme_content)
-    log.success(f'Updated versions in README URLs to [{project.version}]!')
+    log.success(f'Updated versions in README URLs to [{new_ver}]!')
 
-update_changelog_url()
-update_readme_versions()
+# Run MAIN routine
+args = parse_args()
+bump_type = 'major' if args.major else 'minor' if args.minor else 'patch' if args.patch else None
+if not bump_type : raise ValueError(msgs.err_invalid_arg)
+update_readme_versions(bump_pyproject_versions(bump_type))