OwlCyberSecurity - MANAGER

Edit File: vary.py

from functools import wraps

from django.utils.cache import patch_vary_headers

def vary_on_headers(*headers):
    """
    A view decorator that adds the specified headers to the Vary header of the
    response. Usage:

@vary_on_headers('Cookie', 'Accept-language')
       def index(request):
           ...

Note that the header names are not case-sensitive.
    """

def decorator(func):
        @wraps(func)
        def inner_func(*args, **kwargs):
            response = func(*args, **kwargs)
            patch_vary_headers(response, headers)
            return response

return inner_func

return decorator

def vary_on_cookie(func):
    """
    A view decorator that adds "Cookie" to the Vary header of a response. This
    indicates that a page's contents depends on cookies. Usage:

@vary_on_cookie
        def index(request):
            ...
    """

@wraps(func)
    def inner_func(*args, **kwargs):
        response = func(*args, **kwargs)
        patch_vary_headers(response, ("Cookie",))
        return response

return inner_func