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

Detect imports from extensionless python files

Browse Source
This commit is contained in:
kxrd 2017-07-01 09:15:06 +02:00
parent 22e80c27c2
commit 521adaac13
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pipreqs/pipreqs.py
View File

@ -38,7 +38,9 @@ from pipreqs import __version__
REGEXP = [
re.compile(r'^import (.+)$'),
re.compile(r'^from ((?!\.+).*?) import (?:.*)$')
re.compile(r'^from ((?!\.+).*?) import (?:.*)$'),
re.compile('(.*?python[\d.]*)') # Regexp for capturing python versions,
# used in function get_all_imports
]
if sys.version_info[0] > 2:
@ -50,10 +52,11 @@ else:
py2_exclude = ["concurrent", "concurrent.futures"]
def get_all_imports(path, encoding=None, extra_ignore_dirs=None):
def get_all_imports(path, examine_all, encoding=None, extra_ignore_dirs=None):
imports = set()
raw_imports = set()
candidates = []
files = []
ignore_errors = False
ignore_dirs = [".hg", ".svn", ".git", ".tox", "__pycache__", "env", "venv"]
@ -63,13 +66,23 @@ def get_all_imports(path, encoding=None, extra_ignore_dirs=None):
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]
candidates.append(os.path.basename(root))
files = [fn for fn in files if os.path.splitext(fn)[1] == ".py"]
if examine_all:
for file in files_:
check_type = os.popen("file {}".format(file)).read()
ext = os.path.splitext(file)[1]
if ext == ".py" or re.match(REGEXP[2], check_type.lower()):
files.append(file)
else:
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_func(os.path.join(root, file_name), "r", encoding=encoding) as f:
contents = f.read()