pre-commit-hooks/balkian_pre_commit/branchversion.py

import argparse
import semver
import os
from subprocess import check_output

def main(argv=None):
    parser = argparse.ArgumentParser()
    parser.add_argument('filenames', nargs='*')
    args = parser.parse_args(argv)
    branch_name = check_output(['git', 'rev-parse', '--abbrev-ref', 'HEAD']).strip().decode()
    for fn in args.filenames:
        if os.path.basename(fn) == 'VERSION':
            with open(fn) as f:
                currentcontent = f.read().strip()
                prevcontent = check_output(['git', 'show', 'HEAD:{}'.format(fn)]).strip().decode()
                current = semver.parse_version_info(currentcontent)
                print('Checking versions: {} -> {}'.format(prevcontent, currentcontent))
                if (branch_name == 'master' and current.prerelease) or\
                   (semver.compare(currentcontent, prevcontent)<1):
                    return 1
    return 0

if __name__ == '__main__':
    exit(main())