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(), 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") authorization = conn.headers.get("Authorization")
scheme, credentials = get_authorization_scheme_param(authorization) scheme, credentials = get_authorization_scheme_param(authorization)
if not (authorization and scheme and credentials): if not (authorization and scheme and credentials):
@ -300,7 +302,9 @@ class HTTPBearer(HTTPBase):
self.scheme_name = scheme_name or self.__class__.__name__ self.scheme_name = scheme_name or self.__class__.__name__
self.auto_error = auto_error 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") authorization = conn.headers.get("Authorization")
scheme, credentials = get_authorization_scheme_param(authorization) scheme, credentials = get_authorization_scheme_param(authorization)
if not (authorization and scheme and credentials): if not (authorization and scheme and credentials):
@ -401,7 +405,9 @@ class HTTPDigest(HTTPBase):
self.scheme_name = scheme_name or self.__class__.__name__ self.scheme_name = scheme_name or self.__class__.__name__
self.auto_error = auto_error 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") authorization = conn.headers.get("Authorization")
scheme, credentials = get_authorization_scheme_param(authorization) scheme, credentials = get_authorization_scheme_param(authorization)
if not (authorization and scheme and credentials): 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 import FastAPI, Security, WebSocket
from fastapi.security.http import HTTPAuthorizationCredentials, HTTPBase from fastapi.security.http import HTTPAuthorizationCredentials, HTTPBase
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
@ -12,7 +10,7 @@ security = HTTPBase(scheme="Other", auto_error=False)
@app.get("/users/me") @app.get("/users/me")
def read_current_user( def read_current_user(
credentials: Optional[HTTPAuthorizationCredentials] = Security(security), credentials: HTTPAuthorizationCredentials | None = Security(security),
): ):
if credentials is None: if credentials is None:
return {"msg": "Create an account first"} return {"msg": "Create an account first"}
@ -22,7 +20,7 @@ def read_current_user(
@app.websocket("/users/timeline") @app.websocket("/users/timeline")
async def read_user_timeline( async def read_user_timeline(
websocket: WebSocket, websocket: WebSocket,
credentials: Optional[HTTPAuthorizationCredentials] = Security(security), credentials: HTTPAuthorizationCredentials | None = Security(security),
): ):
await websocket.accept() await websocket.accept()
await websocket.send_json( await websocket.send_json(

Loading…
Cancel
Save