From 844255c4d7a95cf162de392dacc0d1c2e129124e Mon Sep 17 00:00:00 2001 From: Yuriy Chebyshev Date: Sun, 11 Feb 2024 10:40:44 +0300 Subject: [PATCH] linters fix --- shvatka/api/dependencies/auth.py | 2 +- shvatka/api/utils/cookie_auth.py | 20 ++++++++++---------- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/shvatka/api/dependencies/auth.py b/shvatka/api/dependencies/auth.py index 36f4f94a..bea2ae87 100644 --- a/shvatka/api/dependencies/auth.py +++ b/shvatka/api/dependencies/auth.py @@ -5,7 +5,7 @@ from datetime import timedelta, datetime from fastapi import Depends, HTTPException, APIRouter -from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm +from fastapi.security import OAuth2PasswordRequestForm from jose import jwt, JWTError from passlib.context import CryptContext from starlette import status diff --git a/shvatka/api/utils/cookie_auth.py b/shvatka/api/utils/cookie_auth.py index a6e95d70..29159794 100644 --- a/shvatka/api/utils/cookie_auth.py +++ b/shvatka/api/utils/cookie_auth.py @@ -1,11 +1,12 @@ -from fastapi.security import OAuth2 -from fastapi.openapi.models import OAuthFlows as OAuthFlowsModel -from fastapi import Request -from fastapi.security.utils import get_authorization_scheme_param +from typing import Dict +from typing import Optional + from fastapi import HTTPException +from fastapi import Request from fastapi import status -from typing import Optional -from typing import Dict +from fastapi.openapi.models import OAuthFlows as OAuthFlowsModel, OAuthFlowPassword +from fastapi.security import OAuth2 +from fastapi.security.utils import get_authorization_scheme_param class OAuth2PasswordBearerWithCookie(OAuth2): @@ -18,11 +19,11 @@ def __init__( ): if not scopes: scopes = {} - flows = OAuthFlowsModel(password={"tokenUrl": token_url, "scopes": scopes}) + flows = OAuthFlowsModel(password=OAuthFlowPassword(tokenUrl=token_url)) super().__init__(flows=flows, scheme_name=scheme_name, auto_error=auto_error) async def __call__(self, request: Request) -> Optional[str]: - authorization: str = request.cookies.get("Authorization") + authorization = request.cookies.get("Authorization", "") scheme, param = get_authorization_scheme_param(authorization) if not authorization or scheme.lower() != "bearer": @@ -32,6 +33,5 @@ async def __call__(self, request: Request) -> Optional[str]: detail="Not authenticated", headers={"WWW-Authenticate": "Bearer"}, ) - else: - return None + return None return param