[Partner Nodes] fix(logs-auth): mask authorization headers in logs (#14774)
Signed-off-by: bigcat88 <bigcat88@icloud.com>
This commit is contained in:
parent
7f287b705e
commit
985fb9d6ad
|
|
@ -9,6 +9,7 @@ from typing import Any
|
||||||
import folder_paths
|
import folder_paths
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
_SENSITIVE_HEADERS = {"authorization", "x-api-key"}
|
||||||
|
|
||||||
|
|
||||||
def get_log_directory():
|
def get_log_directory():
|
||||||
|
|
@ -73,6 +74,10 @@ def _format_data_for_logging(data: Any) -> str:
|
||||||
return str(data)
|
return str(data)
|
||||||
|
|
||||||
|
|
||||||
|
def _redact_headers(headers: dict) -> dict:
|
||||||
|
return {k: ("***" if k.lower() in _SENSITIVE_HEADERS else v) for k, v in headers.items()}
|
||||||
|
|
||||||
|
|
||||||
def log_request_response(
|
def log_request_response(
|
||||||
operation_id: str,
|
operation_id: str,
|
||||||
request_method: str,
|
request_method: str,
|
||||||
|
|
@ -101,7 +106,7 @@ def log_request_response(
|
||||||
log_content.append(f"Method: {request_method}")
|
log_content.append(f"Method: {request_method}")
|
||||||
log_content.append(f"URL: {request_url}")
|
log_content.append(f"URL: {request_url}")
|
||||||
if request_headers:
|
if request_headers:
|
||||||
log_content.append(f"Headers:\n{_format_data_for_logging(request_headers)}")
|
log_content.append(f"Headers:\n{_format_data_for_logging(_redact_headers(request_headers))}")
|
||||||
if request_params:
|
if request_params:
|
||||||
log_content.append(f"Params:\n{_format_data_for_logging(request_params)}")
|
log_content.append(f"Params:\n{_format_data_for_logging(request_params)}")
|
||||||
if request_data is not None:
|
if request_data is not None:
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue