Skip to content

Commit

Permalink
linters fix
Browse files Browse the repository at this point in the history
  • Loading branch information
bomzheg committed Feb 11, 2024
1 parent 1700ace commit 844255c
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 11 deletions.
2 changes: 1 addition & 1 deletion shvatka/api/dependencies/auth.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
20 changes: 10 additions & 10 deletions shvatka/api/utils/cookie_auth.py
Original file line number Diff line number Diff line change
@@ -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):
Expand All @@ -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":
Expand All @@ -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

0 comments on commit 844255c

Please sign in to comment.