Public/fastapi-cache
1
0
Fork 0
mirror of https://github.com/long2ice/fastapi-cache.git synced 2026-03-25 04:57:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: make request and response optional

Browse Source
This commit is contained in:
long2ice
2022-09-10 20:06:37 +08:00
parent a4b3386bf0
commit cb9259807e
8 changed files with 288 additions and 422 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
fastapi_cache/decorator.py
View File

@@ -3,6 +3,8 @@ from functools import wraps
from typing import Callable, Optional, Type
from fastapi.concurrency import run_in_threadpool
from starlette.requests import Request
from starlette.responses import Response
from fastapi_cache import FastAPICache
from fastapi_cache.coder import Coder
@@ -25,6 +27,36 @@ def cache(
"""
def wrapper(func):
signature = inspect.signature(func)
request_param = next(
(param for param in signature.parameters.values() if param.annotation is Request),
None,
)
response_param = next(
(param for param in signature.parameters.values() if param.annotation is Response),
None,
)
parameters = [*signature.parameters.values()]
if not request_param:
parameters.append(
inspect.Parameter(
name="request",
annotation=Request,
kind=inspect.Parameter.KEYWORD_ONLY,
),
)
if not response_param:
parameters.append(
inspect.Parameter(
name="response",
annotation=Response,
kind=inspect.Parameter.KEYWORD_ONLY,
),
)
if parameters:
signature = signature.replace(parameters=parameters)
func.__signature__ = signature
@wraps(func)
async def inner(*args, **kwargs):
nonlocal coder
@@ -66,7 +98,10 @@ def cache(
return response
response.headers["ETag"] = etag
return coder.decode(ret)
if not request_param:
kwargs.pop("request")
if not response_param:
kwargs.pop("response")
if inspect.iscoroutinefunction(func):
ret = await func(*args, **kwargs)
else: