Question

＆＃34; pip freeze＆＃34;我将获得包名列表。 e.g：

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 =＆gt; 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]

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

python中实际导入名称的列表

3 个答案: