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

Merge pull request #95 from jonafato/cleanup-get_all_imports

Browse Source 
Clean up set and file usage in get_all_imports
This commit is contained in:
Jon Banafato 2017-10-26 17:03:17 -04:00 committed by GitHub
commit 800abc43e3
1 changed files with 21 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pipreqs/pipreqs.py
View File

@ -114,13 +114,14 @@ def get_all_imports(
cleaned_name, _, _ = name.partition('.') cleaned_name, _, _ = name.partition('.')
imports.add(cleaned_name) imports.add(cleaned_name)
packages = set(imports) - set(set(candidates) & set(imports)) packages = imports - (set(candidates) & imports)
logging.debug('Found packages: {0}'.format(packages)) logging.debug('Found packages: {0}'.format(packages))
with open(join("stdlib"), "r") as f: with open(join("stdlib"), "r") as f:
data = [x.strip() for x in f.readlines()] data = {x.strip() for x in f}
data = [x for x in data if x not in py2_exclude] if py2 else data
return list(set(packages) - set(data)) data = {x for x in data if x not in py2_exclude} if py2 else data
return list(packages - data)
def filter_line(l): def filter_line(l):