Exclude folders from file list

The previous implementation did not exclude folders, which could be included in the case of submodules
This commit is contained in:
Zachary Barryte 2018-09-07 14:22:28 -04:00
parent 219c70a9c1
commit 582c917016

View file

@ -1,6 +1,7 @@
from __future__ import print_function from __future__ import print_function
import argparse import argparse
import os
import sys import sys
BLACKLIST = [ BLACKLIST = [
@ -22,6 +23,7 @@ def detect_private_key(argv=None):
private_key_files = [] private_key_files = []
for filename in args.filenames: for filename in args.filenames:
if os.path.isfile(filename):
with open(filename, 'rb') as f: with open(filename, 'rb') as f:
content = f.read() content = f.read()
if any(line in content for line in BLACKLIST): if any(line in content for line in BLACKLIST):