def rate_limited()

in server/app/lib/middleware.py [0:0]

• 13 lines of code
• 3 McCabe index (conditional complexity)

def rate_limited(func):
    """Decorator for calls that are rate-limited for anonymous users.
    Once the number of requests per day has been exceeded, this decorator
    will return a 429 HTTP response to the client instead.
    """

    @functools.wraps(func)
    async def session_wrapper(*args):
        ip = quart.request.headers.get("X-Forwarded-For", quart.request.remote_addr).split(",")[-1].strip()
        usage = config.rate_limits.get(ip, 0) + 1
        if config.server.rate_limit_per_ip and usage > config.server.rate_limit_per_ip:
            return quart.Response(
                status=429, response="Your request has been rate-limited. Please check back tomorrow!"
            )
        config.rate_limits[ip] = usage
        print(ip, usage)
        return await func(*args)

    return session_wrapper