Version updater in Python

version_updater.py

#!/usr/bin/env python

import argparse
import re

pattern = "^(.*)(\d+).(\d+).(\d+)(.*)$"

def isVersion(arg, pattern=pattern):
  if not re.match(pattern, arg):
    raise argparse.ArgumentTypeError("invalid version pattern. Should match: [prefix]x.y.z[suffix]")
  return arg

parser = argparse.ArgumentParser(
  prog='version_updater.py',
  description='Update a version number with given update type.')
parser.add_argument('version', help='version to update, should match [prefix]x.y.z[suffix]', type=isVersion)
parser.add_argument('update', help='number to update. Default is patch', choices=['major', 'minor', 'patch'], default='patch', nargs='?')

args = parser.parse_args()

splitted = re.match(pattern, args.version)
prefix, major, minor, patch, suffix = splitted.groups()

if args.update == 'major':
  major, minor, patch = int(major) + 1, 0, 0
elif args.update == 'minor':
  minor, patch = int(minor) + 1, 0
elif args.update == 'patch':
  patch = int(patch) + 1

print(f'{prefix}{major}.{minor}.{patch}{suffix}')