装饰一个python类，使大多数方法在条件下引发异常

Question

我遇到过这样一种情况，如果某个条件是False，类的大多数方法在调用时都需要引发异常，除了一个。可以进入大多数方法并编写if not condition，这样如果条件不成立，每个方法都会引发异常，但我认为可以通过类顶部的单个装饰器以某种方式做到这一点.

This question 类似，但它涉及单独装饰每个方法，如果我要这样做，我不妨将if 语句放入每个方法中。

这里有一些代码和 cmets 来帮助传达它：

CONDITION = True  # change to False to test

def CheckMethods():
    if CONDITION:
        # run all the methods as usual if they are called
        pass 
    else:
        # raise an exception for any method which is called except 'cow'
        # if 'cow' method is called, run it as usual
        pass


@CheckMethods
class AnimalCalls:
    def dog(self):
        print("woof")
    def cat(self):
        print("miaow")
    def cow(self):
        print("moo")
    def sheep(self)
        print("baa") 

a = AnimalCalls()
a.dog()
a.cat()
a.cow()
a.sheep()

有人知道怎么做吗？以前从未装饰过一个类或尝试过像这样检查它的方法。

Answer 1

实现代理就这么简单

class Proxy:
    def __init__(self, inst):
        self.__inst = inst

    def __getattr__(self, name):
        return getattr(self.__inst, name)

您可以使用obj = Proxy(SomeClass()) 而不是obj = SomeClass()。所有对obj.attribute 的访问都被Proxy.__getattr__ 拦截。这是您可以添加更多逻辑的方法，例如：

class MethodChecker:
    def __init__(self, inst, check):
        self.__inst = inst
        self.__check = check

    def __getattr__(self, name):
        self.__check()
        return getattr(self.__inst, name)

Answer 2

代理将是我的选择，但这里是根据要求提供的装饰器。

我添加了一个测试来排除以下划线开头的任何方法。您可能想要包含 _internal 方法，但请注意不要混淆任何特殊的 __dunder__ 方法。

# cond = lambda attr: True  # full access
cond = lambda attr: attr == 'cow'

def methodcheck(cls):
    def cond_getattribute(self, name):
        if name.startswith('_') or cond(name):
            return saved_gettattribute(self, name)
        raise AttributeError("access forbidden")
    saved_gettattribute = cls.__getattribute__
    cls.__getattribute__ = cond_getattribute
    return cls 

@methodcheck
class AnimalCalls:
    def dog(self):
        print("woof")
    def cat(self):
        print("miaow")
    def cow(self):
        print("moo")
    def sheep(self):
        print("baa"

Answer 3

大量改编找到here的代码：

condition = False

def CheckMethods(Cls):
    class NewCls(object):
        def __init__(self,*args,**kwargs):
            self.oInstance = Cls(*args,**kwargs)
        def __getattribute__(self,s):
            try:    
                x = super(NewCls,self).__getattribute__(s)
            except AttributeError:      
                pass
            else:
                return x
            x = self.oInstance.__getattribute__(s)
            if condition:
                return x
            else:
                if s == 'cow':
                    return x
                else:
                    raise ValueError('Condition not true')
    return NewCls

@CheckMethods
class AnimalCalls(object):
    def dog(self):
        print("woof")
    def cat(self):
        print("miaow")
    def cow(self):
        print("moo")
    def sheep(self):
        print("baa") 

oF = AnimalCalls()

结果：

contition = False; of.moo() -> 'moo'
contition = True; of.moo() -> 'moo'
condition = False; of.dog() -> 'ValueError: Condition not true'
condition = True; of.dog() -> 'woof'