pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-01 18:56:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add whitelist support for detect-privat-key hook

Browse source
This commit is contained in:
Carlos Bustillo 2022-12-05 18:50:51 +01:00
parent 6336b8e792
commit 706b591462
2 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pre_commit_hooks/detect_private_key.py
View file

@ -20,14 +20,24 @@ BLACKLIST = [
def main(argv: Sequence[str] | None = None) -> int:
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Filenames to check')
parser.add_argument(
'--whitelist',
help='The filename with with the files (relative path) to whitelist',
)
args = parser.parse_args(argv)
private_key_files = []
whitelisted_files = ''
if args.whitelist:
with open(args.whitelist) as f:
whitelisted_files = f.read()
for filename in args.filenames:
with open(filename, 'rb') as f:
content = f.read()
if any(line in content for line in BLACKLIST):
if any(line in content for line in BLACKLIST) \
and filename not in whitelisted_files:
private_key_files.append(filename)
if private_key_files: