pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-05 19:46:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Auto-detect repository's default branch instead of hard-coding list

Browse source
This commit is contained in:
emanb29 2026-04-04 13:56:22 -07:00
parent f1dff44d3a
commit ea5f72dd61
No known key found for this signature in database
3 changed files with 48 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
pre_commit_hooks/no_commit_to_branch.py
View file

@ -9,6 +9,17 @@ from pre_commit_hooks.util import CalledProcessError
from pre_commit_hooks.util import cmd_output
def _default_branch() -> frozenset[str]:
try:
ref = cmd_output('git', 'rev-parse', '--abbrev-ref', 'origin/HEAD')
branch = ref.strip().removeprefix('origin/')
if branch:
return frozenset((branch,))
except CalledProcessError:
pass
return frozenset(('master', 'main'))
def is_on_branch(
protected: AbstractSet[str],
patterns: AbstractSet[str] = frozenset(),
@ -39,7 +50,7 @@ def main(argv: Sequence[str] | None = None) -> int:
)
args = parser.parse_args(argv)
protected = frozenset(args.branch or ('master', 'main'))
protected = frozenset(args.branch) if args.branch else _default_branch()
patterns = frozenset(args.pattern or ())
return int(is_on_branch(protected, patterns))