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 customizing the module denylist

Browse source
This commit is contained in:
Enrico Zini 2025-12-12 10:24:09 +01:00
parent 6e513a2e71
commit 12f60d656d
2 changed files with 30 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pre_commit_hooks/debug_statement_hook.py
View file

@ -74,8 +74,24 @@ def check_file(filename: str) -> int:
def main(argv: Sequence[str] | None = None) -> int:
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Filenames to run')
parser.add_argument(
'--forbid',
type=str, action='append',
help='Extra module name(s) to forbid',
)
parser.add_argument(
'--allow',
type=str,
action='append',
help='Extra module name(s) to allow',
)
args = parser.parse_args(argv)
for name in args.forbid or ():
DEBUG_STATEMENTS.add(name)
for name in args.allow or ():
DEBUG_STATEMENTS.discard(name)
retv = 0
for filename in args.filenames:
retv |= check_file(filename)