在 python 脚本中使用 conda install

Question

根据answer，您可以从 Python 脚本中导入 pip 并使用它来安装模块。 conda install可以做到这一点吗？

conda 文档仅显示命令行中的示例，但我正在寻找可以从 Python 脚本中执行的代码。

是的，我可以从脚本中执行 shell 命令，但我试图避免这种情况，因为它基本上假设 conda 不能被导入并且它的函数被调用。

Answer 1

在 Python 脚本中使用 conda 已经有一段时间了，我认为使用 subprocess 模块调用 conda 总体上效果最好。在 Python 3.7+ 中，您可以执行以下操作：

import json
from subprocess import run


def conda_list(environment):
    proc = run(["conda", "list", "--json", "--name", environment],
               text=True, capture_output=True)
    return json.loads(proc.stdout)


def conda_install(environment, *package):
    proc = run(["conda", "install", "--quiet", "--name", environment] + packages,
               text=True, capture_output=True)
    return json.loads(proc.stdout)

正如我在评论中指出的那样，conda.cli.main() 不适合外部使用。它直接解析sys.argv，所以如果你尝试在你自己的脚本中使用你自己的命令行参数，它们也会被馈送到conda.cli.main()。

@YenForYang 的回答建议 conda.cli.python_api 更好，因为这是一个公开记录的 API，用于调用 conda 命令。但是，我发现它仍然有粗糙的边缘。 conda 在执行命令时建立内部状态（例如缓存）。 conda 通常使用和测试的方式是作为命令行程序。在这种情况下，这个内部状态在conda 命令结束时被丢弃。使用conda.cli.python_api，您可以在单个进程中执行多个conda 命令。在这种情况下，内部状态会继续存在，有时会导致意外结果（例如，在执行命令时缓存变得过时）。当然，conda 应该可以直接处理这个内部状态。我的观点是，以这种方式使用conda 并不是开发人员的主要关注点。如果您想要最可靠的方法，请按照开发人员希望使用的方式使用conda——作为它自己的进程。

conda 是一个相当慢的命令，所以我认为不必担心调用子进程对性能的影响。正如我在另一条评论中指出的，pip 是与conda 类似的工具，并在its documentation that it should be called as a subprocess, not imported into Python 中明确声明。

Answer 2

试试这个：

!conda install xyzpackage

请记住，这必须在 Python 脚本而不是操作系统提示中完成。

或者你可以尝试以下方法：

import sys
from conda.cli import main

    sys.exit(main())

    try:
        import conda
        from conda.cli import main
        sys.argv = ['conda'] + list(args)
        main()

Answer 3

我发现conda.cli.python_api 和conda.api 是有限的，从某种意义上说，它们都没有执行这样的命令的选项：

conda export env > requirements.txt

所以我使用带有 shell=True 标志的子进程来完成工作。

subprocess.run(f"conda env export --name {env} > {file_path_from_history}",shell=True)

其中env 是要保存到requirements.txt 的环境的名称。

Answer 4

我尝试并为我工作的更简单的事情是：

import os

try:
    import graphviz
except:
    print ("graphviz not found, Installing graphviz ")
    os.system("conda install -c anaconda graphviz")
    import graphviz

并确保以管理员身份运行脚本。

Answer 5

我在查看最新的Conda Python API 并注意到实际上只有 2 个具有“非常长期稳定性”的公共模块：

1. conda.cli.python_api
2. conda.api

对于您的问题，我将使用第一个：

注意：下面的run_command() 将总是添加一个-y/--yes 选项（即它不会要求确认 )

import conda.cli.python_api as Conda
import sys

###################################################################################################
# The below is roughly equivalent to:
#   conda install -y 'args-go-here' 'no-whitespace-splitting-occurs' 'square-brackets-optional'

(stdout_str, stderr_str, return_code_int) = Conda.run_command(
    Conda.Commands.INSTALL, # alternatively, you can just say "install"
                            # ...it's probably safer long-term to use the Commands class though
                            # Commands include:
                            #  CLEAN,CONFIG,CREATE,INFO,INSTALL,HELP,LIST,REMOVE,SEARCH,UPDATE,RUN
    [ 'args-go-here', 'no-whitespace-splitting-occurs', 'square-brackets-optional' ],
    use_exception_handler=True,  # Defaults to False, use that if you want to handle your own exceptions
    stdout=sys.stdout, # Defaults to being returned as a str (stdout_str)
    stderr=sys.stderr, # Also defaults to being returned as str (stderr_str)
    search_path=Conda.SEARCH_PATH  # this is the default; adding only for illustrative purposes
)
###################################################################################################

---

使用上面的好处是它解决了使用conda.cli.main()时出现的问题（在上面的cmets中提到）：

...conda 试图解释命令行参数而不是 conda.cli.main() 的参数，因此像这样使用 conda.cli.main() 可能不适用于某些事情。

---

上面cmets中的另一个问题是：

频道不是默认频道时如何【安装包】？

import conda.cli.python_api as Conda
import sys

###################################################################################################
# Either:
#   conda install -y -c <CHANNEL> <PACKAGE>
# Or (>= conda 4.6)
#   conda install -y <CHANNEL>::<PACKAGE>

(stdout_str, stderr_str, return_code_int) = Conda.run_command(
    Conda.Commands.INSTALL,
    '-c', '<CHANNEL>',
    '<PACKAGE>'
    use_exception_handler=True, stdout=sys.stdout, stderr=sys.stderr
)
###################################################################################################

Answer 6

您可以使用conda.cli.main。例如，这会安装numpy：

import conda.cli

conda.cli.main('conda', 'install',  '-y', 'numpy')

使用-y 参数来避免交互式问题：

-y, --yes 不要求确认。