mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 19:46:54 +00:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
import argparse
|
|
import os
|
|
from typing import Optional
|
|
from typing import Sequence
|
|
import re
|
|
|
|
|
|
def check_for_no_commit_tokens(
|
|
filename: str,
|
|
tokens: Optional[Sequence[str]],
|
|
) -> str:
|
|
with open(filename, mode='r', encoding='utf-8') as file_processed:
|
|
lines = file_processed.read()
|
|
for token in tokens:
|
|
match = re.search(token, lines)
|
|
if match:
|
|
# no-commit token is present
|
|
# return offending token
|
|
return token
|
|
return None
|
|
|
|
|
|
def main(argv: Optional[Sequence[str]] = None) -> int:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
'--tokens',
|
|
help=(
|
|
'The set of tokens, separated by comma, used to block commits '
|
|
'if present. For example, "nocommit,NOCOMMIT"'
|
|
),
|
|
)
|
|
parser.add_argument('filenames', nargs='*', help='Filenames to fix')
|
|
args = parser.parse_args(argv)
|
|
|
|
return_code = 0
|
|
if args.tokens:
|
|
tokens = args.tokens.split(",")
|
|
else:
|
|
# default value
|
|
tokens = ["nocommit"]
|
|
|
|
for filename in args.filenames:
|
|
_, extension = os.path.splitext(filename.lower())
|
|
token = check_for_no_commit_tokens(filename, tokens)
|
|
if token:
|
|
print(f'Found no-commit token in {filename}: {token}')
|
|
return_code = 1
|
|
return return_code
|
|
|
|
|
|
if __name__ == '__main__':
|
|
exit(main())
|