pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-03-30 10:16:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow multiple branches to be protected

Browse source 
Original patch by @moas
This commit is contained in:
Anthony Sottile 2018-06-09 11:16:14 -07:00
parent 2aa7aeb572
commit baec308367
3 changed files with 22 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pre_commit_hooks/no_commit_to_branch.py
View file

@ -12,18 +12,19 @@ def is_on_branch(protected):
except CalledProcessError:
return False
chunks = branch.strip().split('/')
return '/'.join(chunks[2:]) == protected
return '/'.join(chunks[2:]) in protected
def main(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument(
'-b', '--branch', default='master',
help='branch to disallow commits to',
'-b', '--branch', action='append',
help='branch to disallow commits to, may be specified multiple times',
)
args = parser.parse_args(argv)
return int(is_on_branch(args.branch))
protected = set(args.branch or ('master',))
return int(is_on_branch(protected))
if __name__ == '__main__':