from flask import g, request

from app import app


def after_this_request(f):
    """
    Decorator to execute methods after the request is handled, to
    modify the response before sending back to the client. This could
    be used to set cookies.
    """

    if not hasattr(g, 'after_request_callbacks'):
        g.after_request_callbacks = []

    g.after_request_callbacks.append(f)
    return f


@app.after_request
def call_after_request_callbacks(response):
    for callback in getattr(g, 'after_request_callbacks', ()):
        callback(response)

    return response