From a10a186c889aac00fac581bbd5f79d8a8ada3d86 Mon Sep 17 00:00:00 2001 From: Mix <32300164+mnixry@users.noreply.github.com> Date: Sat, 26 Aug 2023 15:17:30 +0800 Subject: [PATCH] Fix linting --- fastapi/security/http.py | 4 +++- fastapi/security/utils.py | 6 +++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/fastapi/security/http.py b/fastapi/security/http.py index 764dfac07..7c12f47ac 100644 --- a/fastapi/security/http.py +++ b/fastapi/security/http.py @@ -66,7 +66,9 @@ class HTTPBasic(HTTPBase): self.auto_error = auto_error @handle_exc_for_ws - async def __call__(self, request: HTTPConnection) -> Optional[HTTPBasicCredentials]: + async def __call__( # type: ignore + self, request: HTTPConnection + ) -> Optional[HTTPBasicCredentials]: authorization = request.headers.get("Authorization") scheme, param = get_authorization_scheme_param(authorization) if self.realm: diff --git a/fastapi/security/utils.py b/fastapi/security/utils.py index 2a0849303..c40d526a7 100644 --- a/fastapi/security/utils.py +++ b/fastapi/security/utils.py @@ -17,15 +17,15 @@ def get_authorization_scheme_param( _SecurityDepFunc = TypeVar( - "_SecurityDepFunc", bound=Callable[[Any, HTTPConnection], Awaitable] + "_SecurityDepFunc", bound=Callable[[Any, HTTPConnection], Awaitable[Any]] ) def handle_exc_for_ws(func: _SecurityDepFunc) -> _SecurityDepFunc: @wraps(func) - async def wrapper(self, request: HTTPConnection, *args, **kwargs): + async def wrapper(self: Any, request: HTTPConnection) -> Any: try: - return await func(self, request, *args, **kwargs) + return await func(self, request) except HTTPException as e: if not isinstance(request, WebSocket): raise e