查找所有导入的外部库

时间:2016-02-18 20:31:17

标签: python pycharm python-3.4

我有一个庞大的python程序需要在新的虚拟环境中运行(在另一台机器上)。该程序导入几个外部模块(需要先在新的环境中安装)。

例如,我的一个模块具有以下导入:

import matplotlib
import os
from kivy.uix.label import Label
import my_file.my_module_5 as my_mod_5 

另一个模块有:

import my_module_7
import django

在这种情况下,我需要创建一个这样的列表:

['matplotlib', 'kivy', 'django']

请注意,我自己的模块不包含在内,因为它们是将迁移到新环境的程序的一部分,并且不必安装。这些模块都不是像os那样的python的一部分。

我创建了一个函数,可以在项目中查找所有导入的模块,并过滤掉属于项目本身的模块。但是,它还返回标准的python模块,如ossys等。

def all_modules_in_project():
    """
    Finds all modules imported in the current working directory tree.

    :return: Set of module names.
    """

    project_directories = set()
    project_files = set()

    modules_imported = set()

    for path, dirs_names, files_names_in_dir in os.walk(os.getcwd()):
        project_directories |= set(dirs_names)

        for file_name in files_names_in_dir:
            if file_name.endswith('.py'):

                project_files.add(file_name[:-3])

                with open(path + '/' + file_name, 'r') as opened_file:
                    file_lines = opened_file.readlines()

                    for line in file_lines:

                        # import XXX
                        match = re.match(r'import\s([\w\.]+)', line)
                        if match:
                            modules_imported.add(match.groups()[0])

                        # from XXX
                        match = re.match(r'from\s([\w\.]+)', line)
                        if match:
                            modules_imported.add(match.groups()[0])

    # Removes XXX that were matched as follows `import proj_dir. .. .XXX`
    for module in modules_imported.copy():
        matched = re.match(r'(\w+)\.', module)
        if matched:
            pre_dot = matched.groups()[0]

            if pre_dot in project_directories:
                modules_imported.remove(module)

            else:
                # Replaces `xxx.yyy` with `xxx`
                modules_imported.remove(module)
                modules_imported.add(pre_dot)

    return modules_imported - project_files - project_directories
  • 如何过滤掉不需要的标准python库 安装?
  • 或者,有没有更简单的方法来确定我的程序使用哪些外部库?

(我不需要all installed packages;我只需要程序导入的那些)

1 个答案:

答案 0 :(得分:0)

您可以使用npm WARN optional Skipping failed optional dependency /chokidar/fsevents: npm WARN notsup Not compatible with your operating system or architecture: fsevents@1.0.7 中的get_installed_distributions方法获取脚本中已安装的第三方发行版的列表:

pip

然后,您可以在函数中使用该列表来确定包是标准库还是第三方,例如:

from pip import get_installed_distributions

# Get a list of distribution objects (see pkg_resources documentation)
dist_list = get_installed_distributions()

# Get a list of distribution names as strings
dist_names = [dist.project_name for dist in dist_list]

有关分发对象的更多信息,请查看pkg_resources documentation

编辑:因为你在一个venv中,你会有点子,但是如果你没有pip,你可以使用你知道要安装的包来获得如下的类似列表(我正在使用在此示例中为if pre_dot in dist_names: modules_imported.add(module)

setuptools