Python 从装饰器导入模块

Question

我正在 Python 3 中开发一个应用程序，我所做的是非常规的。

cx_Oracle 是一个很难设置的模块，对于我的应用程序来说是一个可选的依赖项。我想做的是将模块的导入包装在装饰器中，仅放置在使用它的函数之上。这将避免在我的模块顶部导入并允许它不被设置。

class Loader():
    def load_cx_oracle(func):
        def inner(*args, **kwargs):

            # Additional code before import.

            import cx_Oracle

            return func(*args, **kwargs)
        return inner

    @load_cx_oracle
    def function_using_cx_oracle(self):
        conn = cx_Oracle.connect()

但是，当我尝试上述操作时，我得到 NameError: name 'cx_Oracle' is not defined

Answer 1

如果您想要一个可选的导入，请在模块的开头使用 try-except：

try:
    import cx_Oracle
except ImportError:
    cx_Oracle = None

Answer 2

import 仅绑定您使用它的名称空间中的模块名称。如果您在一个函数内执行 import cx_Oracle，名称 cx_Oracle 将仅在该函数内可用。

但是，您可以使用 global 进行全局分配。更改您的装饰器以使用：

global cx_Oracle
import cx_Oracle

这种方法是否真的正确还有待商榷。根据您的代码的使用方式，如果用户想让 cx_Oracle 可用，使用 try-wrapped 导入，或者加载由一些外部设置（例如配置文件）定义。

Answer 3

接受的答案有几个问题。其中最重要的是每次调用该函数时它都会运行导入逻辑。第二个是装饰器必须在它使用的同一个模块中定义，否则装饰器和被装饰器将具有不同的全局变量。您可以通过函数的 __globals__ 属性直接访问函数的全局变量。在执行导入逻辑之前，代码示例首先检查模块是否存在于函数的全局变量中。该示例还使用 functools.wraps 装饰器在使用 help(func).

等内容时保留文档字符串、函数名称和参数名称

from functools import wraps

def load_operator(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        if "operator" not in func.__globals__:
            # setup logic -- only executed once
            import operator
            func.__globals__["operator"] = operator
        return func(*args, **kwargs)
    return wrapper

class A:
    @load_operator
    def add(self, x, y):
        return operator.add(x, y)

    def subtract(self, x, y):
        return operator.subtract(x, y)

a = A()
try:
    a.subtract(1, 2)
except NameError:
    print("operator not yet loaded")
print(a.add(1, 2))
print(A.add)

Python 从装饰器导入模块

Python Import Module from Decorator

python

python-3.x

python-import

python-decorators