Add default config when init

update JsonEncoder
This commit is contained in:
long2ice
2020-10-16 16:55:33 +08:00
parent c0259bc699
commit dc2ac9cc90
7 changed files with 153 additions and 65 deletions

View File

@@ -1,17 +1,34 @@
from typing import Callable
from fastapi_cache.coder import Coder, JsonCoder
from fastapi_cache.key_builder import default_key_builder
class FastAPICache:
_backend = None
_prefix = None
_expire = None
_init = False
_coder = None
_key_builder = None
@classmethod
def init(cls, backend, prefix: str = "", expire: int = None):
def init(
cls,
backend,
prefix: str = "",
expire: int = None,
coder: Coder = JsonCoder,
key_builder: Callable = default_key_builder,
):
if cls._init:
return
cls._init = True
cls._backend = backend
cls._prefix = prefix
cls._expire = expire
cls._coder = coder
cls._key_builder = key_builder
@classmethod
def get_backend(cls):
@@ -25,3 +42,11 @@ class FastAPICache:
@classmethod
def get_expire(cls):
return cls._expire
@classmethod
def get_coder(cls):
return cls._coder
@classmethod
def get_key_builder(cls):
return cls._key_builder