mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-06-14 03:09:28 +00:00
19 lines
486 B
Python
19 lines
486 B
Python
""" Custom handler for caching """
|
|
from typing import Any, Callable, Tuple, Union
|
|
|
|
from django.core.cache import cache
|
|
|
|
|
|
def get_or_set(
|
|
cache_key: str,
|
|
function: Callable[..., str],
|
|
*args: Tuple[Any, ...],
|
|
timeout: Union[float, None] = None
|
|
) -> str:
|
|
"""Django's built-in get_or_set isn't cutting it"""
|
|
value = str(cache.get(cache_key))
|
|
if value is None:
|
|
value = function(*args)
|
|
cache.set(cache_key, value, timeout=timeout)
|
|
return value
|