Browse Source

🎨 Auto format

pull/10147/head
pre-commit-ci-lite[bot] 5 months ago
committed by GitHub
parent
commit
20c4b7a077
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
  1. 12
      fastapi/security/http.py
  2. 6
      tests/test_security_http_base_optional.py

12
fastapi/security/http.py

@ -91,7 +91,9 @@ class HTTPBase(SecurityBase):
headers=self.make_authenticate_headers(),
)
async def __call__(self, conn: HTTPConnection) -> HTTPAuthorizationCredentials | None:
async def __call__(
self, conn: HTTPConnection
) -> HTTPAuthorizationCredentials | None:
authorization = conn.headers.get("Authorization")
scheme, credentials = get_authorization_scheme_param(authorization)
if not (authorization and scheme and credentials):
@ -300,7 +302,9 @@ class HTTPBearer(HTTPBase):
self.scheme_name = scheme_name or self.__class__.__name__
self.auto_error = auto_error
async def __call__(self, conn: HTTPConnection) -> HTTPAuthorizationCredentials | None:
async def __call__(
self, conn: HTTPConnection
) -> HTTPAuthorizationCredentials | None:
authorization = conn.headers.get("Authorization")
scheme, credentials = get_authorization_scheme_param(authorization)
if not (authorization and scheme and credentials):
@ -401,7 +405,9 @@ class HTTPDigest(HTTPBase):
self.scheme_name = scheme_name or self.__class__.__name__
self.auto_error = auto_error
async def __call__(self, conn: HTTPConnection) -> HTTPAuthorizationCredentials | None:
async def __call__(
self, conn: HTTPConnection
) -> HTTPAuthorizationCredentials | None:
authorization = conn.headers.get("Authorization")
scheme, credentials = get_authorization_scheme_param(authorization)
if not (authorization and scheme and credentials):

6
tests/test_security_http_base_optional.py

@ -1,5 +1,3 @@
from typing import Optional
from fastapi import FastAPI, Security, WebSocket
from fastapi.security.http import HTTPAuthorizationCredentials, HTTPBase
from fastapi.testclient import TestClient
@ -12,7 +10,7 @@ security = HTTPBase(scheme="Other", auto_error=False)
@app.get("/users/me")
def read_current_user(
credentials: Optional[HTTPAuthorizationCredentials] = Security(security),
credentials: HTTPAuthorizationCredentials | None = Security(security),
):
if credentials is None:
return {"msg": "Create an account first"}
@ -22,7 +20,7 @@ def read_current_user(
@app.websocket("/users/timeline")
async def read_user_timeline(
websocket: WebSocket,
credentials: Optional[HTTPAuthorizationCredentials] = Security(security),
credentials: HTTPAuthorizationCredentials | None = Security(security),
):
await websocket.accept()
await websocket.send_json(

Loading…
Cancel
Save