【重装救急】批量安装python源文件中import了的库

wkdxz

备份当前已安装的库，本来应该执行以下步骤，先运行pip freeze > requirements.txt
然后运行pip install -r requirements.txt

---

然而，我今天手欠将python3.8.5升级到3.10.4，发现原先的库好多都不兼容，用不了。之前的库也忘了备份到requirements.txt。
想换回原先的版本，安装时一直提示：0x80070001错误。于是我慌了，一堆脚本需要调用的，如果一下子都用不了，我的工作量估计要增加好几倍。
于是有了下面的代码：

功能：

• 查找指定目录内已有的python源文件
• 找出文件里import了哪些库
• 然后使用代码判断这些库是否安装
• 未安装就使用pip install安装库
  

虽然代码简单，但帮了我大忙，只花了10多分钟时间，我的python又满血复活了。

[Python] 纯文本查看 复制代码

import os
import ast
import subprocess


def dir_files(directory, extension):
    """遍历目录内指定后缀"""
    file_list = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                if "tempCodeRunnerFile" not in file:  # CodeRunner生成的临时文件忽略掉
                    file_list.append(os.path.join(root, file))
    return file_list


def py_file_list(python_file_folders):
    """查找指定目录内容的python文件"""
    pylist = []
    for directory_to_search in python_file_folders:
        for f in [".py", ".pyw"]:  # python文件后缀
            tmp_files = dir_files(directory_to_search, f)
            pylist += tmp_files
    return pylist


def parse_imports(filename):
    """解析python源文件import了哪些库"""
    with open(filename, "r", encoding="utf-8") as f:
        tree = ast.parse(f.read())
    imports = set()
    for node in ast.iter_child_nodes(tree):
        if isinstance(node, ast.Import):
            for alias in node.names:
                imports.add(alias.name.split(".")[0])
        elif isinstance(node, ast.ImportFrom):
            module_name = node.module.split(".")[0] if node.module else ""
            for alias in node.names:
                imports.add(module_name + "." + alias.name.split(".")[0])

    return list(imports)


def pip_install(libraries):
    """判断是否安装了库 没有就直接pip安装"""
    for library in libraries:
        library = library.split(".")[0]
        try:
            __import__(library)
            # print(library, "已安装")
        except ImportError:
            print(f"{library} 没有安装，正在使用pip安装...")
            subprocess.call(["pip", "install", library])
            # print("未安装：", library)


if __name__ == "__main__":
    py_folders = [r"E:\Backup\脚本", r"E:\管理系统", r"E:\桌面文件"]

    py_lists = py_file_list(py_folders)
    # print(py_lists)

    all_imports = []
    # 输出列表中的文件路径
    for py_file in py_lists:
        imports = parse_imports(py_file)

        # print(py_file, imports)

        all_imports += [library.split(".")[0] for library in imports]

    if imports:
        imports = set(all_imports)
        pip_install(imports)

爱飞的猫

如果代码重要，就应该提前记录到 requirements 里面

另外一般都会用 venv 在项目目录建立一个独立的 Python 环境，隔离 Python 依赖

吾乃齐天大圣

遇事不要慌

ccber

我一直想整一个离线版的安装。

sai609

一、安装andconda，内置多个第三方库
二、最新的版本，得卸载旧版，install安装最新的。。。。但是有时会新旧or与其他库共存而冲突

666888tzq

我的Python自打安装了就没打开过几次。

lcg2014

学习经验，保持进步

fatsheepfy58

出现问题，解决问题，赞

三滑稽甲苯

如果其中有的库只支持 Python 3.8.5，阁下又当如何应对