使用装饰器停用功能

Question

是否可以使用 python 装饰器“停用”功能？举个例子：

cond = False

class C:

    if cond:
        def x(self): print "hi"

    def y(self): print "ho"

是否可以像这样用装饰器重写这段代码？：

class C:

    @cond
    def x(self): print "hi"

    def y(self): print "ho"

背景：在我们的库中，一些依赖项（如 matplotlib）是可选的，只有少数函数需要这些依赖项（用于调试或前端）。这意味着在某些系统上 matplotlib 未安装在其他系统上，但在两者上都应该运行（核心）代码。因此，如果未安装 matplotlib，我想禁用某些功能。有这么优雅的方式吗？

Answer 1

从一个类中删除方法对我来说似乎很奇怪。为什么不使用 try / except

plot_avail = True   
try:
    import matplotlib   
except:
    plot_avail = False

然后在你的函数中

def x(self):
    if not plot_avail: return

所以没有调用结束：类没有属性 X

Answer 2

您可以使用装饰器将函数变成无操作（记录警告）：

def conditional(cond, warning=None):
    def noop_decorator(func):
        return func  # pass through

    def neutered_function(func):
        def neutered(*args, **kw):
            if warning:
                log.warn(warning)
            return
        return neutered

    return noop_decorator if cond else neutered_function

这里conditional是一家装饰工厂。它根据条件返回两个装饰器之一。

一个装饰器只是简单地保持功能不变。另一个装饰器完全替换了被装饰的函数，而是发出警告。

用途：

@conditional('matplotlib' in sys.modules, 'Please install matplotlib')
def foo(self, bar):
    pass

Answer 3

Martijns 答案涉及将函数转换为 noops，我将解释如何将它们从类中实际删除 - 这可能是矫枉过正，我会满足于 Martijns 答案的变体，它会引发某种异常.但无论如何：

您可以使用类装饰器从类中删除受影响的函数。这个需要一个布尔值和一个要删除的属性列表：

def rm_attrs_if(cond, attrs):

    if not cond:
        return lambda c: c #if the condition is false, don't modify the class

    def rm_attrs(cls):
        d = dict(cls.__dict__) #copy class dict
        for attr in attrs:
            del d[attr]        #remove all listed attributes
        return type(cls.__name__, cls.__bases__, d) #create and return new class

    return rm_attrs

像这样使用它：

@rm_attrs_if(something == False, ["f1", "f2"])
class X():
    def f1(): pass
    def f2(): pass
    def f3(): pass