如何通过自省确定定义方法的类

Question

我正在使用redis 商店在flask 中构建一个限速装饰器，它将识别不同端点上的不同限制。 （我意识到那里有许多限制速率的装饰器，但我的用例非常不同，所以我自己滚动是有意义的。）

基本上我遇到的问题是确保我存储在 redis 中的密钥是特定于类的。我在烧瓶中使用蓝图模式，基本上是这样的：

class SomeEndpoint(MethodView):
    def get(self):
        # Respond to get request
    def post(self):
        # Respond to post request

这里的问题是我希望能够在不添加任何额外命名约定的情况下对这些类的post 方法进行速率限制。在我看来，最好的方法是这样的：

class SomeEndpoint(MethodView):

    @RateLimit  # Access SomeEndpoint class name
    def post(self):
        # Some response

但在装饰器中，只有post 函数在作用域内。给定post 函数，我将如何回到SomeEndpoint 类？这是装饰器的基本布局。这可能会令人困惑，所以这里有一个更具体的装饰器示例。

class RateLimit(object):
"""
The base decorator for app-specific rate-limiting.
"""
def __call__(self, f):
    def endpoint(*args, **kwargs):
        print class_backtrack(f)  # Should print SomeEnpoint
        return f(*args, **kwargs)
    return endpoint

基本上是在寻找class_backtrack 函数的样子。我查看了inspect 模块，但没有发现任何似乎可以实现此目的的东西。

Answer 1

你可以装饰整个类而不仅仅是方法：

def wrap(Class, method):
    def wrapper(self, *args, **kwargs):
        print Class
        return method(self, *args, **kwargs)
    return method.__class__(wrapper, None, Class)

def rate_limit(*methods):
    def decorator(Class):
        for method_name in methods:
            method = getattr(Class, method_name)
            setattr(Class, method_name, wrap(Class, method))
        return Class
    return decorator

@rate_limit('post')
class SomeEndpoint(object):

    def post(self):
        pass

class Subclass(SomeEndpoint):
    pass

a = Subclass()
a.post()
# prints <class 'SomeEndpoint'>