linux 如何在运行时安装和导入Python模块

iyfjxgzm  于 2023-01-29  发布在  Linux
关注(0)|答案(5)|浏览(220)

我想写一个脚本来自动设置一个全新的ubuntu安装和安装一个基于django的应用程序,因为这个脚本将在一个新的服务器上运行,Python脚本需要自动安装一些必需的模块。
这是剧本。

#!/usr/bin/env python

import subprocess
import os
import sys

def pip_install(mod):
    print subprocess.check_output("pip install %s" % mod, shell=True)

if __name__ == "__main__":
    if os.getuid() != 0:
        print "Sorry, you need to run the script as root."
        sys.exit()

    try:
        import pexpect
    except:
        pip_install('pexpect') 
        import pexpect        

    # More code here...

pexpect的安装是成功的,但是下一行import pexpect失败了,我认为这是因为在运行时代码没有意识到新安装的pexpect
如何在运行时安装和导入Python模块?我对其他方法持开放态度。

nx7onnlm

nx7onnlm1#

您可以导入pip而不使用子进程:

import pip

def install(package):
    pip.main(['install', package])

# Example
if __name__ == '__main__':
    try:
        import pexpect
    except ImportError:
        install('pexpect')
        import pexpect

另一个镜头:

import pip

def import_with_auto_install(package):
    try:
        return __import__(package)
    except ImportError:
        pip.main(['install', package])
    return __import__(package)

# Example
if __name__ == '__main__':
    pexpect = import_with_auto_install('pexpect')
    print(pexpect)

[编辑]
您应该考虑使用requirements.txt和pip。看起来您正在尝试自动化部署(这很好!),在我的工具带中,我也有virtualenvwrapper,vagrantansible
这是我的输出:

(test)root@vagrant:~/test# pip uninstall pexpect
Uninstalling pexpect:
  /usr/lib/python-environments/test/lib/python2.6/site-packages/ANSI.py
  /usr/lib/python-environments/test/lib/python2.6/site-packages/ANSI.pyc
  /usr/lib/python-environments/test/lib/python2.6/site-packages/FSM.py
  /usr/lib/python-environments/test/lib/python2.6/site-packages/FSM.pyc
  /usr/lib/python-environments/test/lib/python2.6/site-packages/fdpexpect.py
  /usr/lib/python-environments/test/lib/python2.6/site-packages/fdpexpect.pyc
  /usr/lib/python-environments/test/lib/python2.6/site-packages/pexpect-2.4-py2.6.egg-info
  /usr/lib/python-environments/test/lib/python2.6/site-packages/pexpect.py
  /usr/lib/python-environments/test/lib/python2.6/site-packages/pexpect.pyc
  /usr/lib/python-environments/test/lib/python2.6/site-packages/pxssh.py
  /usr/lib/python-environments/test/lib/python2.6/site-packages/pxssh.pyc
  /usr/lib/python-environments/test/lib/python2.6/site-packages/screen.py
  /usr/lib/python-environments/test/lib/python2.6/site-packages/screen.pyc
Proceed (y/n)? y
  Successfully uninstalled pexpect
(test)root@vagrant:~/test# python test.py
Downloading/unpacking pexpect
  Downloading pexpect-2.4.tar.gz (113Kb): 113Kb downloaded
  Running setup.py egg_info for package pexpect
Installing collected packages: pexpect
  Running setup.py install for pexpect
Successfully installed pexpect
Cleaning up...
<module 'pexpect' from '/usr/lib/python-environments/test/lib/python2.6/site-packages/pexpect.pyc'>
(test)root@vagrant:~/test#
tag5nh1u

tag5nh1u2#

对于使用pip版本高于10.x的用户,pip没有main函数,因此替代方法是使用import pip._internal as pip而不是import pip,如下所示:

更新Paulo的回答

import pip._internal as pip

def install(package):
    pip.main(['install', package])

if __name__ == '__main__':
    try:
        import pexpect
    except ImportError:
        install('pexpect')
        import pexpect
nc1teljy

nc1teljy3#

我实际上为此目的制作了一个模块(impstall
它真的很容易用途:

import impstall
impstall.now('pexpect')
impstall.now('wx', pipName='wxPython')

Github link for issues/contributions

gkn4icbw

gkn4icbw4#

我使用imp模块解决了我的问题。

#!/usr/bin/env python

import pip
import imp

def install_and_load(package):
    pip.main(['install', package])

    path = '/usr/local/lib/python2.7/dist-packages'
    if path not in sys.path:
        sys.path.append(path)

    f, fname, desc = imp.find_module(package)
    return imp.load(package, f, fname, desc)

if __name__ == "__main__":
    try:
        import pexpect
    except:
        pexpect = install_and_load('pexpect')

    # More code...

实际上,代码并不理想,因为我需要硬编码Python模块目录,但由于脚本是针对已知的目标系统,我认为这是可以的。

ztigrdn8

ztigrdn85#

我也遇到了同样的问题,但是谷歌搜索都没有帮助。经过几个小时的调试,我发现这可能是因为sys.path没有重新加载新的安装目录。
在我使用Ubuntu Docker的情况下,我希望在运行时为Python3.8(预安装)使用import dns.resolver,我还创建了ubuntu用户,并使用此用户运行所有内容(包括Python脚本)。

  • 在安装之前,sys.path没有/home/ubuntu/.local/lib/python3.8/site-packages,因为我没有安装任何东西。
  • 当像上面那样使用subprocesspip.main安装时,它会创建/home/ubuntu/.local/lib/python3.8/site-packages(作为用户安装)。
  • 安装后,应刷新sys.path以包含此新位置。

由于sys.pathsite模块管理,因此应重新加载它(参考HERE):

import site
    from importlib import reload
    reload(site)

所有需要的人都可以使用完整的模块:

import subprocess
import sys

try:
    import dns.resolver
except ImportError:
    subprocess.check_call([sys.executable, "-m", "pip", "install", "dnspython"])
    import site
    from importlib import reload
    reload(site)
    import dns.resolver

我不是Python开发人员,所以这些代码可以简化得更多。这可能会对像我这样的DevOps工程师在新的CI/CD环境中有所帮助。

相关问题