File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 99import folder_paths
1010
1111logger = logging .getLogger (__name__ )
12+ _SENSITIVE_HEADERS = {"authorization" , "x-api-key" }
1213
1314
1415def get_log_directory ():
@@ -73,6 +74,10 @@ def _format_data_for_logging(data: Any) -> str:
7374 return str (data )
7475
7576
77+ def _redact_headers (headers : dict ) -> dict :
78+ return {k : ("***" if k .lower () in _SENSITIVE_HEADERS else v ) for k , v in headers .items ()}
79+
80+
7681def log_request_response (
7782 operation_id : str ,
7883 request_method : str ,
@@ -101,7 +106,7 @@ def log_request_response(
101106 log_content .append (f"Method: { request_method } " )
102107 log_content .append (f"URL: { request_url } " )
103108 if request_headers :
104- log_content .append (f"Headers:\n { _format_data_for_logging (request_headers )} " )
109+ log_content .append (f"Headers:\n { _format_data_for_logging (_redact_headers ( request_headers ) )} " )
105110 if request_params :
106111 log_content .append (f"Params:\n { _format_data_for_logging (request_params )} " )
107112 if request_data is not None :
You can’t perform that action at this time.
0 commit comments