github/pipreqs
1
1
Fork 0
mirror of https://github.com/bndr/pipreqs.git synced 2025-06-06 03:25:21 +00:00
Code Issues 3 Packages Projects Releases Wiki Activity

fix(pipreqs): allow comma-separated --ignore parameter

Browse Source
This commit is contained in:
Vadim Kravcenko 2016-02-05 18:00:34 +01:00
parent 632206f9f4
commit 075436d8e1
3 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pipreqs/pipreqs.py
View File

@ -13,7 +13,7 @@ Options:
$ export HTTP_PROXY="http://10.10.1.10:3128" $ export HTTP_PROXY="http://10.10.1.10:3128"
$ export HTTPS_PROXY="https://10.10.1.10:1080" $ export HTTPS_PROXY="https://10.10.1.10:1080"
--debug Print debug information --debug Print debug information
--ignore <dirs>... Ignore extra directories --ignore <dirs>... Ignore extra directories, each separated by a comma
--encoding <charset> Use encoding parameter for file open --encoding <charset> Use encoding parameter for file open
--savepath <file> Save the list of requirements in the given file --savepath <file> Save the list of requirements in the given file
--force Overwrite existing requirements.txt --force Overwrite existing requirements.txt
@ -44,13 +44,18 @@ else:
open_func = codecs.open open_func = codecs.open
def get_all_imports(path, encoding=None, extra_ignore_dirs = []): def get_all_imports(path, encoding=None, extra_ignore_dirs=None):
imports = set() imports = set()
raw_imports = set() raw_imports = set()
candidates = [] candidates = []
ignore_errors = False ignore_errors = False
ignore_dirs = [".hg", ".svn", ".git", "__pycache__", "env", "venv"] ignore_dirs = [".hg", ".svn", ".git", "__pycache__", "env", "venv"]
ignore_dirs.extend(extra_ignore_dirs)
if extra_ignore_dirs:
ignore_dirs_parsed = []
for e in extra_ignore_dirs:
ignore_dirs_parsed.append(os.path.basename(os.path.realpath(e)))
ignore_dirs.extend(ignore_dirs_parsed)
for root, dirs, files in os.walk(path): for root, dirs, files in os.walk(path):
dirs[:] = [d for d in dirs if d not in ignore_dirs] dirs[:] = [d for d in dirs if d not in ignore_dirs]
@ -195,7 +200,10 @@ def join(f):
def init(args): def init(args):
encoding = args.get('--encoding') encoding = args.get('--encoding')
extra_ignore_dirs = args.get('--ignore', []) extra_ignore_dirs = args.get('--ignore')
if extra_ignore_dirs:
extra_ignore_dirs = extra_ignore_dirs.split(',')
candidates = get_all_imports(args['<path>'], candidates = get_all_imports(args['<path>'],
encoding=encoding, encoding=encoding,

2
tests/_data_ignore/.ignore_second/ignored.py
View File

@ -1,2 +1,2 @@
# Everything in here should be ignored # Everything in here should be ignored
from getpass import getpass from pattern.web import Twitter, plaintext

2
tests/test_pipreqs.py
View File

@ -154,7 +154,7 @@ class TestPipreqs(unittest.TestCase):
'--use-local': None, '--force': True, '--use-local': None, '--force': True,
'--proxy':None, '--proxy':None,
'--pypi-server':None, '--pypi-server':None,
'--ignore':['.ignored_dir', '.ignore_second'] '--ignore':'.ignored_dir,.ignore_second'
} }
) )
with open(os.path.join(self.project_with_ignore_directory, "requirements.txt"), "r") as f: with open(os.path.join(self.project_with_ignore_directory, "requirements.txt"), "r") as f: