iUdZddlZddlmZgdZeeeed<gdZeee eefed<deDd d DzZ d ed efd Z ded e fdZ dS)u敏感信息过滤模块。N)Final)z(password|passwd|pwd)\s*=\s*\S+z!(api[_-]?key|apikey)\s*[=:]\s*\S+z,(token|access_token|auth_token)\s*[=:]\s*\S+z (secret|secret_key)\s*[=:]\s*\S+ Bearer\s+\S+!-----BEGIN\s+.*PRIVATE\s+KEY----------BEGIN\s+.*CERTIFICATE-----SENSITIVE_PATTERNS))z!(password|passwd|pwd)\s*=\s*(\S+) \1=[REDACTED])z#(api[_-]?key|apikey)\s*[=:]\s*(\S+)r)z.(token|access_token|auth_token)\s*[=:]\s*(\S+)r)z"(secret|secret_key)\s*[=:]\s*(\S+)r_KEY_VALUE_PATTERNScVg|]&\}}tj|tj|f'Srecompile IGNORECASE).0pattern replacements 8/root/projects/qq-shell/src/butler/security/sanitizer.py rs?Z''5cPg|]#}tj|tjdf$S)z [REDACTED]r )rrs rrrs;Z''6r)rrroutputreturncP|}tD]\}}|||}|S)u 过滤输出中的敏感信息。 Args: output: 原始输出字符串 Returns: 过滤后的安全输出 )_COMPILED_SANITIZERSsub)r sanitizedrrs rsanitize_outputr%s8I 488KK Y77 rtextcPtD]\}}||rdSdS)u 检查文本是否包含敏感信息。 Args: text: 要检查的文本 Returns: 如果包含敏感信息返回 True TF)rsearch)rr_s rcontains_secretsr"5s<+  >>$   44  5r)__doc__r typingrrliststr__annotations__r tuplerrboolr"r rrr*s!!! (((E$s)$555U4c3h01 3  C C     3 4      r