-
-
Notifications
You must be signed in to change notification settings - Fork 363
/
bearer.py
53 lines (43 loc) · 1.77 KB
/
bearer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from fastapi import Response, status
from fastapi.responses import JSONResponse
from fastapi.security import OAuth2PasswordBearer
from pydantic import BaseModel
from fastapi_users.authentication.transport.base import (
Transport,
TransportLogoutNotSupportedError,
)
from fastapi_users.openapi import OpenAPIResponseType
class BearerResponse(BaseModel):
access_token: str
token_type: str
class BearerTransport(Transport):
scheme: OAuth2PasswordBearer
def __init__(self, tokenUrl: str):
self.scheme = OAuth2PasswordBearer(tokenUrl, auto_error=False)
async def get_login_response(self, token: str) -> Response:
bearer_response = BearerResponse(access_token=token, token_type="bearer")
return JSONResponse(bearer_response.dict())
async def get_logout_response(self) -> Response:
raise TransportLogoutNotSupportedError()
@staticmethod
def get_openapi_login_responses_success() -> OpenAPIResponseType:
return {
status.HTTP_200_OK: {
"model": BearerResponse,
"content": {
"application/json": {
"example": {
"access_token": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1"
"c2VyX2lkIjoiOTIyMWZmYzktNjQwZi00MzcyLTg2Z"
"DMtY2U2NDJjYmE1NjAzIiwiYXVkIjoiZmFzdGFwaS"
"11c2VyczphdXRoIiwiZXhwIjoxNTcxNTA0MTkzfQ."
"M10bjOe45I5Ncu_uXvOmVV8QxnL-nZfcH96U90JaocI",
"token_type": "bearer",
}
}
},
},
}
@staticmethod
def get_openapi_logout_responses_success() -> OpenAPIResponseType:
return {}