Add enable param to init

This commit is contained in:
long2ice
2021-10-28 15:52:21 +08:00
parent 4faa5b7101
commit 9928f4cda0
3 changed files with 11 additions and 1 deletions

View File

@@ -11,6 +11,7 @@ class FastAPICache:
_init = False
_coder = None
_key_builder = None
_enable = True
@classmethod
def init(
@@ -20,6 +21,7 @@ class FastAPICache:
expire: int = None,
coder: Coder = JsonCoder,
key_builder: Callable = default_key_builder,
enable: bool = True,
):
if cls._init:
return
@@ -29,6 +31,7 @@ class FastAPICache:
cls._expire = expire
cls._coder = coder
cls._key_builder = key_builder
cls._enable = enable
@classmethod
def get_backend(cls):
@@ -51,6 +54,10 @@ class FastAPICache:
def get_key_builder(cls):
return cls._key_builder
@classmethod
def get_enable(cls):
return cls._enable
@classmethod
async def clear(cls, namespace: str = None, key: str = None):
namespace = cls._prefix + ":" + namespace if namespace else None

View File

@@ -29,7 +29,9 @@ def cache(
copy_kwargs = kwargs.copy()
request = copy_kwargs.pop("request", None)
response = copy_kwargs.pop("response", None)
if request and request.headers.get("Cache-Control") == "no-store":
if (
request and request.headers.get("Cache-Control") == "no-store"
) or not FastAPICache.get_enable():
return await func(*args, **kwargs)
coder = coder or FastAPICache.get_coder()