本文主要是介绍【教程】自动检测和安装Python脚本依赖的第三方库,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
转载请注明出处:小锋学长生活大爆炸[xfxuezhang.cn]
背景说明
对于新python环境,要运行某个脚本,可能需要安装很多库,一般可以通过提供的requirements.txt来自动安装。但如果没有这个txt,那就得手动一个一个安装,非常的麻烦。
通过捕捉ImportError错误,实际上可以从错误消息中提取缺失的模块名称,而一旦确定了缺失的模块名称,就可以使用 pip
自动安装它们了。
参考脚本
autoDependencyInstaller.py
import argparse
import subprocess
import sys
import redef install_module(module_name, python_executable):try:subprocess.check_call([python_executable, "-m", "pip", "install", module_name])return Trueexcept subprocess.CalledProcessError:return Falsedef extract_imports(file_name):with open(file_name, 'r') as file:file_content = file.read()imports = re.findall(r'^(?:from\s+(\S+)|import\s+(\S+))(?:\s+|$)', file_content, re.MULTILINE)top_level_modules = {imp[0].split('.')[0] if imp[0] else imp[1].split('.')[0] for imp in imports}return list(top_level_modules)def check_module(module, python_executable):try:subprocess.check_call([python_executable, "-c", f"import {module}"])return Trueexcept subprocess.CalledProcessError:return Falsedef check_and_install_modules(modules, python_executable):for module in modules:if check_module(module, python_executable):print(f"模块 '{module}' 已存在.")else:print(f"尝试安装模块: {module}")if not install_module(module, python_executable):correct_name = input(f"安装 '{module}' 失败。请输入正确的包名,或按 Enter 跳过: ").strip()if correct_name:install_module(correct_name, python_executable)def main():parser = argparse.ArgumentParser(description="自动检测和安装 Python 脚本依赖.")parser.add_argument("script", help="要检查依赖的 Python 脚本文件名")parser.add_argument("-p", "--python-path", help="Python 解释器的路径(可选)", default=sys.executable)args = parser.parse_args()modules_to_check = extract_imports(args.script)check_and_install_modules(modules_to_check, args.python_path)if __name__ == "__main__":main()
参数第一项表示目标py文件。第二项表示要用道德python路径,如果没有给,则默认使用当前环境下的python
这个脚本只提取 Python 脚本中的 import
和 from ... import
语句,然后在当前脚本中尝试导入它们。这样做的好处是,不会执行原始脚本的其他部分,只会检查依赖项是否存在。如果遇到任何 ImportError
,则可以安装相应的缺失模块。这种方法更加安全和高效,因为它避免了不必要的脚本执行。
使用示例
python autoDependencyInstaller.py <目标py文件> <目标python路径>
比如:
python autoDependencyInstaller.py young_http.py
python autoDependencyInstaller.py young_http.py -p /home/sxf/miniconda3/envs/py39/bin/python
进阶使用
把这个脚本打包:
pip install pyinstaller
pyinstaller -F -w autoDependencyInstaller.py
生成的二进制文件在dist目录下,所以就可以非常方便的使用了:
./dist/autoDependencyInstaller young_http.py -p /home/sxf/miniconda3/envs/py39/bin/python
注意:对于打包的文件,必须提供-p,不然用的是二进制文件自己的环境,会出问题。
或者可以直接这样,用$(which python):
./dist/autoDependencyInstaller young_http.py -p $(which python)
更进阶使用
把这个文件放到系统目录下,就不用每次都找了:
sudo mv dist/autoDependencyInstaller /usr/sbin/
以后就可以直接用了:
autoDependencyInstaller young_http.py -p /home/sxf/miniconda3/envs/py39/bin/python
这篇关于【教程】自动检测和安装Python脚本依赖的第三方库的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!