Skip to content
Snippets Groups Projects
Commit 5d81ffe5 authored by Mohammad Torkashvand's avatar Mohammad Torkashvand
Browse files

fix linter

parent d8dcff67
No related branches found
No related tags found
No related merge requests found
Pipeline #86201 failed
This commit is part of merge request !198. Comments created here will be created in the context of that merge request.
...@@ -11,7 +11,6 @@ FastAPI to ensure secure API development. ...@@ -11,7 +11,6 @@ FastAPI to ensure secure API development.
import re import re
import ssl import ssl
from collections.abc import AsyncGenerator, Awaitable, Callable, Mapping from collections.abc import AsyncGenerator, Awaitable, Callable, Mapping
from enum import StrEnum
from http import HTTPStatus from http import HTTPStatus
from json import JSONDecodeError from json import JSONDecodeError
from typing import Any, ClassVar, cast from typing import Any, ClassVar, cast
...@@ -188,13 +187,13 @@ class OIDCUser(HTTPBearer): ...@@ -188,13 +187,13 @@ class OIDCUser(HTTPBearer):
resource_server_secret: str resource_server_secret: str
def __init__( def __init__(
self, self,
openid_url: str, openid_url: str,
resource_server_id: str, resource_server_id: str,
resource_server_secret: str, resource_server_secret: str,
*, *,
auto_error: bool = True, auto_error: bool = True,
scheme_name: str | None = None, scheme_name: str | None = None,
): ):
"""Set up OIDCUser with specified OpenID Connect configurations and credentials.""" """Set up OIDCUser with specified OpenID Connect configurations and credentials."""
super().__init__(auto_error=auto_error) super().__init__(auto_error=auto_error)
...@@ -204,7 +203,7 @@ class OIDCUser(HTTPBearer): ...@@ -204,7 +203,7 @@ class OIDCUser(HTTPBearer):
self.scheme_name = scheme_name or self.__class__.__name__ self.scheme_name = scheme_name or self.__class__.__name__
async def __call__( # type: ignore[override] async def __call__( # type: ignore[override]
self, request: Request, token: str | None = None self, request: Request, token: str | None = None
) -> OIDCUserModel | None: ) -> OIDCUserModel | None:
"""Return the OIDC user from OIDC introspect endpoint. """Return the OIDC user from OIDC introspect endpoint.
...@@ -242,7 +241,7 @@ class OIDCUser(HTTPBearer): ...@@ -242,7 +241,7 @@ class OIDCUser(HTTPBearer):
user_info = await self.userinfo(async_request, token) user_info = await self.userinfo(async_request, token)
user_info['client_id'] = intercepted_token.get("client_id") user_info["client_id"] = intercepted_token.get("client_id")
logger.debug("OIDCUserModel object.", intercepted_token=intercepted_token) logger.debug("OIDCUserModel object.", intercepted_token=intercepted_token)
return user_info return user_info
...@@ -367,11 +366,11 @@ def _evaluate_decision(decision: OPAResult, *, auto_error: bool, **context: dict ...@@ -367,11 +366,11 @@ def _evaluate_decision(decision: OPAResult, *, auto_error: bool, **context: dict
def opa_decision( def opa_decision(
opa_url: str, opa_url: str,
oidc_security: OIDCUser, oidc_security: OIDCUser,
*, *,
auto_error: bool = True, auto_error: bool = True,
opa_kwargs: Mapping[str, str] | None = None, opa_kwargs: Mapping[str, str] | None = None,
) -> Callable[[Request, OIDCUserModel, AsyncClient], Awaitable[bool | None]]: ) -> Callable[[Request, OIDCUserModel, AsyncClient], Awaitable[bool | None]]:
"""Create a decision function for Open Policy Agent (OPA) authorization checks. """Create a decision function for Open Policy Agent (OPA) authorization checks.
...@@ -388,9 +387,9 @@ def opa_decision( ...@@ -388,9 +387,9 @@ def opa_decision(
""" """
async def _opa_decision( async def _opa_decision(
request: Request, request: Request,
user_info: OIDCUserModel = Depends(oidc_security), # noqa: B008 user_info: OIDCUserModel = Depends(oidc_security), # noqa: B008
async_request: AsyncClient = Depends(_make_async_client), # noqa: B008 async_request: AsyncClient = Depends(_make_async_client), # noqa: B008
) -> bool | None: ) -> bool | None:
"""Check OIDCUserModel against the OPA policy. """Check OIDCUserModel against the OPA policy.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment