From af8032a531223b6cfec6a0857fa29774de2aced4 Mon Sep 17 00:00:00 2001 From: Ed-XCF Date: Wed, 4 Jan 2023 15:28:14 +0800 Subject: [PATCH] add use cache --- redis_property.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/redis_property.py b/redis_property.py index 1330018..16ffd4f 100644 --- a/redis_property.py +++ b/redis_property.py @@ -6,7 +6,7 @@ import orjson from redis import Redis, RedisError -__all__ = ["redis_property", "cache_ttl", "cache_disable", "no_cache"] +__all__ = ["redis_property", "cache_ttl", "cache_disable", "no_cache", "use_cache"] _redis_cli = None _default_cache_ttl = 24 * 60 * 60 @@ -152,3 +152,15 @@ def wrapper(*args, **kwargs): cache_disable.reset(token) return wrapper + + +def use_cache(func): + @functools.wraps(func) + def wrapper(*args, **kwargs): + token = cache_disable.set(False) + try: + return func(*args, **kwargs) + finally: + cache_disable.reset(token) + + return wrapper