python中的实际导入名称列表

Question

用"pip冻结“，我会得到一个包裹名称的列表。例如：

Django==1.9.7
psycopg2==2.6.1
djangorestframework==3.3.3
djangorestframework-jwt==1.8.0
django-rest-swagger==0.3.7
django-environ==0.4.0
python-dateutil==2.5.3
django-sendfile==0.3.10

有任何方法可以接收到import的实际名称列表吗？代替djangorestframework => rest_framework

Answer 1

是的，top_level.txt将是一个正确的模块名。您可以使用pkg_resources模块从包中提取元数据。

Python代码用于此：

import pkg_resources


def read_requirements(requirements_file):
    with open(requirements_file, 'r') as f:
        return f.read().splitlines()


def get_package_name(package):
    return list(pkg_resources.get_distribution(package)._get_metadata('top_level.txt'))[0]


requirements = read_requirements('/path/to/requirements.txt')
packages = [get_package_name(p) for p in requirements]

Answer 2

您可以使用标准的pkgutil模块获取顶级导入的列表，如下所示：

import pkgutil
list(pkgutil.iter_modules())

这将只找到生活在普通文件中的模块，压缩文件或支持模块枚举的另一个加载程序。应该大部分都在一个标准的体系里。

结果是一个带有加载程序、模块名称的三元组列表，以及它是单个模块还是包。如果您只对模块名称感兴趣，只需执行以下操作：

list(item[1] for item in pkgutil.iter_modules())

Answer 3

另一种方法是查看该文件的第一行：

# If output of pip freeze is djangorestframework==3.3.3
# Then your dir prefix becomes "djangorestframework-3.3.3"

../lib/python2.7/site-packages/<dir-prefix>.dist-info/top_level.txt 

# e.g.
../lib/python2.7/site-packages/djangorestframework-3.3.3.dist-info/top_level.txt