python中实际导入名称的列表

Question

通过"pip freeze",我将获得一个包名列表.例如:

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

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

import pkgutil
list(pkgutil.iter_modules())

这将只找到生活在常规文件,zip文件或支持模块枚举的其他加载器的模块.应该是大多数标准系统.

结果是一个3元组的列表,包含加载器,模块名称,以及它是单个模块还是包.如果您只对模块名称感兴趣,只需执行以下操作:

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

Answer 2

是的，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]