beavyHQ/beavy

View on GitHub
beavy/common/rate_limits.py

Summary

Maintainability
A
0 mins
Test Coverage

def rate_limit(default, *args, **kwargs):
    from beavy.app import limiter, app

    def inner(fn):
        key = "{}.{}".format(fn.__module__, fn.__name__)
        return limiter.limit(app.config.get("RATELIMITS", {}
                                            ).get(key, default),
                             *args, **kwargs)(fn)
    return inner