什么是Pythonic基于已安装的模块编写条件语句的方法？

Question

来自C ++世界，我习惯于使用像CMake之类的工具在编译时确定的标志来编写条件编译。 我想知道模仿这个功能最恐怖的方式是什么。 例如，这是我目前设置的，具体取决于是否找到模块：

import imp

try:
    imp.find_module('petsc4py')
    HAVE_PETSC=True
except ImportError:
    HAVE_PETSC=False

然后我可以在我的其余Python代码中使用HAVE_PETSC 。 这有效，但我想知道它是否是在Python中使用它的正确方法。

Answer 1

是的，没关系。 您甚至可以直接发出导入，并使用modulename本身作为标志 - 如：

try:
    import petsc4py
except ImportError
    petsc4py = None

在任何使用之前，只需测试petsc4py本身的真实性。

实际上，检查它是否存在，然后只是尝试导入它，由于冗余而感觉是unpythonic，因为两个动作都会触发ImportError。 但是检查有一个HAVE_PETSC变量是可以的 - 它可以在上面的try / except之后使用HAVE_PETSC = bool(petsc4py)

Answer 2

你做这件事的方式或多或少都很好。 实际上，python标准库使用了一种类似的范例：“尝试导入一些东西，如果由于某种原因它无效，那么在 多个 地方设置一个变量”。 检查程序中稍后是否设置了布尔值将比每次执行单独的try / except块更快。

在你的情况下，这样做可能会更好：

try:
    import petsc4py
    HAVE_PETSC = True
except ImportError:
    HAVE_PETSC = False

你在范例层面上有什么工作，但在这种情况下没有真正的理由通过importlib （你可能不应该使用imp ，因为它在最近的python版本中已被弃用）。