40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from functools import wraps
|
|
from django.utils.cache import patch_vary_headers
|
|
from django.utils.decorators import available_attrs
|
|
|
|
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, assigned=available_attrs(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, assigned=available_attrs(func))
|
|
def inner_func(*args, **kwargs):
|
|
response = func(*args, **kwargs)
|
|
patch_vary_headers(response, ('Cookie',))
|
|
return response
|
|
return inner_func
|