1
0
mirror of https://github.com/balkian/pre-commit-hooks.git synced 2024-10-31 22:11:43 +00:00
pre-commit-hooks/balkian_pre_commit/branchversion.py

22 lines
678 B
Python
Raw Normal View History

2017-01-10 14:12:52 +00:00
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()
if branch_name == 'master':
for fn in args.filenames:
if os.path.basename(fn) == 'VERSION':
with open(fn) as f:
version = semver.parse_version_info(f.read())
if version.prerelease:
return 1
return 0
if __name__ == '__main__':
exit(main())