pre-commit-hooks/pre_commit_hooks/no_commit_to_branch.py at main · pre-commit/pre-commit-hooks

History

48 lines (39 loc) · 1.33 KB

Raw

from __future__ import annotations

import argparse

import re

from collections.abc import Sequence

from typing import AbstractSet

from pre_commit_hooks.util import CalledProcessError

from pre_commit_hooks.util import cmd_output

def is_on_branch(

protected: AbstractSet[str],

patterns: AbstractSet[str] = frozenset(),

) -> bool:

try:

ref_name = cmd_output('git', 'symbolic-ref', 'HEAD')

except CalledProcessError:

return False

chunks = ref_name.strip().split('/')

branch_name = '/'.join(chunks[2:])

return branch_name in protected or any(

re.match(p, branch_name) for p in patterns

)

def main(argv: Sequence[str] | None = None) -> int:

parser = argparse.ArgumentParser()

parser.add_argument(

'-b', '--branch', action='append',

help='branch to disallow commits to, may be specified multiple times',

)

parser.add_argument(

'-p', '--pattern', action='append',

help=(

'regex pattern for branch name to disallow commits to, '

'may be specified multiple times'

)

args = parser.parse_args(argv)

protected = frozenset(args.branch or ('master', 'main'))

patterns = frozenset(args.pattern or ())

return int(is_on_branch(protected, patterns))

if __name__ == '__main__':

raise SystemExit(main())

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Uh oh!

Expand file tree

Search code, repositories, users, issues, pull requests...

Uh oh!

Uh oh!

FilesExpand file tree

no_commit_to_branch.py

Latest commit

History

no_commit_to_branch.py

File metadata and controls

Expand file tree