mirror of
https://github.com/jointakahe/takahe.git
synced 2024-11-25 16:51:00 +00:00
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from functools import partial, wraps
|
|
|
|
from django.views.decorators.cache import cache_page as dj_cache_page
|
|
|
|
from core.models import Config
|
|
|
|
|
|
def cache_page(
|
|
timeout: int | str = "cache_timeout_page_default",
|
|
*,
|
|
per_identity: bool = False,
|
|
key_prefix: str = "",
|
|
):
|
|
"""
|
|
Decorator for views that caches the page result.
|
|
timeout can either be the number of seconds or the name of a SystemOptions
|
|
value.
|
|
"""
|
|
_timeout = timeout
|
|
|
|
def decorator(function):
|
|
@wraps(function)
|
|
def inner(request, *args, **kwargs):
|
|
prefix = key_prefix
|
|
if per_identity:
|
|
identity_id = request.identity.pk if request.identity else "0"
|
|
prefix = f"{key_prefix or ''}:ident{identity_id}"
|
|
if isinstance(_timeout, str):
|
|
timeout = getattr(Config.system, _timeout)
|
|
else:
|
|
timeout = _timeout
|
|
return dj_cache_page(timeout=timeout, key_prefix=prefix)(function)(
|
|
request, *args, **kwargs
|
|
)
|
|
|
|
return inner
|
|
|
|
return decorator
|
|
|
|
|
|
per_identity_cache_page = partial(cache_page, per_identity=True)
|