takahe/core/decorators.py

42 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)