From 301ff8ce83498a6bdc5985979c320cc8db416cf8 Mon Sep 17 00:00:00 2001 From: LeeW Date: Mon, 31 Aug 2015 21:25:17 +0800 Subject: [PATCH] Ignore irreverent directory when generating requirement.txt * Currently ignore .git, __pycache__, env --- pipreqs/pipreqs.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/pipreqs/pipreqs.py b/pipreqs/pipreqs.py index 15b748c..a1f369e 100755 --- a/pipreqs/pipreqs.py +++ b/pipreqs/pipreqs.py @@ -32,10 +32,18 @@ REGEXP = [ def get_all_imports(path): imports = [] candidates = [] + ignore_dirs = [".git", "__pycache__", "env"] for root, dirs, files in os.walk(path): + dirs[:] = [d for d in dirs if d not in ignore_dirs] + + # for d in ignore_dirs: + # if d in dirs: + # dirs.remove(d) + candidates.append(os.path.basename(root)) files = [fn for fn in files if os.path.splitext(fn)[1] == ".py"] + candidates += [os.path.splitext(fn)[0] for fn in files] for file_name in files: with open(os.path.join(root, file_name), "r") as f: