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): Walk through all sys.path instead of only get_python_lib()

Browse Source
This commit is contained in:
Vadim Kravcenko 2015-05-11 23:05:47 +02:00
parent f3011d6487
commit f3ec4d5d2a
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pipreqs/pipreqs.py
View File

@ -88,18 +88,20 @@ def get_imports_info(imports):
def get_locally_installed_packages():
packages = {}
for root, dirs, files in os.walk(get_python_lib()):
for item in files:
if "top_level" in item:
with open(os.path.join(root, item), "r") as f:
package = root.split("/")[-1].split("-")
package_import = f.read().strip().split("\n")
for item in package_import:
if item not in ["tests", "_tests"]:
packages[item] = {
'version': package[1].replace(".dist", ""),
'name': package[0]
}
ignore = ["tests", "_tests", "egg", "EGG", "info"]
for path in sys.path:
for root, dirs, files in os.walk(path):
for item in files:
if "top_level" in item:
with open(os.path.join(root, item), "r") as f:
package = root.split("/")[-1].split("-")
package_import = f.read().strip().split("\n")
for item in package_import:
if item not in ignore and package[0] not in ignore:
packages[item] = {
'version': package[1].replace(".dist", ""),
'name': package[0]
}
return packages