feat: add Nailong Fund advisor capabilities
This commit is contained in:
@@ -1,14 +1,17 @@
|
||||
from functools import lru_cache
|
||||
|
||||
from fastapi import Depends, HTTPException, Request, status
|
||||
from fastapi import Depends, Request
|
||||
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
|
||||
|
||||
from app.core.config import get_settings
|
||||
from app.core.contracts import RequestContext
|
||||
from app.core.errors import UnauthorizedAgentError
|
||||
from app.core.security import JwtAuthenticator
|
||||
from app.service.identity_service import IdentityService
|
||||
from app.service.risk_questionnaire_service import RiskQuestionnaireService
|
||||
|
||||
_bearer = HTTPBearer(auto_error=False)
|
||||
TRACE_ID_HEADER = "X-Trace-ID"
|
||||
|
||||
|
||||
@lru_cache(maxsize=1)
|
||||
@@ -20,16 +23,18 @@ async def build_request_context(
|
||||
request: Request,
|
||||
credentials: HTTPAuthorizationCredentials | None = Depends(_bearer), # noqa: B008
|
||||
) -> RequestContext:
|
||||
request.state.trace_id = request.headers.get(TRACE_ID_HEADER)
|
||||
if credentials is None or credentials.scheme.lower() != "bearer":
|
||||
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Unauthorized")
|
||||
raise UnauthorizedAgentError("token is missing, invalid, or revoked")
|
||||
try:
|
||||
context = _authenticator().authenticate(credentials.credentials)
|
||||
context = await IdentityService().resolve(context)
|
||||
except Exception as exc:
|
||||
from app.core.errors import UnauthorizedAgentError
|
||||
|
||||
if isinstance(exc, UnauthorizedAgentError):
|
||||
raise HTTPException(status_code=401, detail="Unauthorized") from exc
|
||||
raise
|
||||
raise UnauthorizedAgentError("token is missing, invalid, or revoked") from exc
|
||||
request.state.request_context = context
|
||||
onboarding_path = request.url.path.startswith("/api/v1/onboarding/")
|
||||
if not onboarding_path and await RiskQuestionnaireService().is_required(context):
|
||||
from app.core.errors import OnboardingRequiredError
|
||||
|
||||
raise OnboardingRequiredError("请先完成开户风险测评问卷")
|
||||
return context
|
||||
|
||||
Reference in New Issue
Block a user