wip

Signed-off-by: Grant Ramsay <[email protected]>

fastapi_redis_cache/cache.py

@@ -54,6 +54,7 @@ async def inner_wrapper(
             request = func_kwargs.pop("request", None)
             response = func_kwargs.pop("response", None)
             create_response_directly = not response
+
             if create_response_directly:
                 response = Response()
                 # below fix by @jaepetto on the original repo.
@@ -149,6 +150,8 @@ async def inner_wrapper(
             **kwargs: Any,  # noqa: ANN401
         ) -> Any:  # noqa: ANN401
             """Invalidate all cached responses with the same tag."""
+            _headers = kwargs.get("request", None).headers
+
             redis_cache = FastApiRedisCache()
             orig_response = await get_api_response_async(func, *args, **kwargs)
 

tests/live_test.py

@@ -99,7 +99,7 @@ def cache_invalid_type(request: Request, response: Response) -> logging.Logger:
 
 @app.get("/cache_with_args/{user}")
 @cache_one_hour(tag="user_tag")
-def cache_with_args(user: int) -> dict[str, Union[bool, str]]:
+def cache_with_args(user: int, request: Request) -> dict[str, Union[bool, str]]:
     """Have a varying cache key based on the user argument."""
     return {
         "success": True,
@@ -109,7 +109,9 @@ def cache_with_args(user: int) -> dict[str, Union[bool, str]]:
 
 @app.put("/cache_with_args/{user}")
 @expires(tag="user_tag")
-def put_cache_with_args(user: int) -> dict[str, Union[bool, str]]:
+def put_cache_with_args(
+    user: int, request: Request
+) -> dict[str, Union[bool, str]]:
     """Put request to change data for a specific user."""
     return {
         "success": True,