o  h@s>ddlZddlZddlZddlZddlmZGdddZdS)N)MIMETextc@s:eZdZdefddZddZddZdd Zd d Zd S) AlertingHandleralerting_configcCs||pi|_i|_d|jvr||jd|_dSd|jvr<|jd|jd|jd|jd|jdd|_dSdS)NZsmtp_secret_name smtp_server smtp_port smtp_username smtp_password from_emailrrrrr )rsmtp_credentials!get_smtp_credentials_from_secretsget)selfrrEC:\Users\shubh\Videos\recall\recall_audit_sampler\alerting_handler.py__init__s         zAlertingHandler.__init__c Csz'td}|j|d}t|d}|dt|d|d|d|dd WStyD}ztd t|iWYd}~Sd}~ww) NZsecretsmanager)ZSecretIdZ SecretStringrrrrr r z>[ERROR] Failed to fetch SMTP credentials from SecretsManager: ) boto3clientZget_secret_valuejsonloadsint Exceptionprintstr)r secret_namerresponseZsecreterrrr s   z1AlertingHandler.get_smtp_credentials_from_secretsc Csd|jvr)z|||Wnty(}ztdt|WYd}~nd}~wwd|jvrTz |||WdStyS}ztdt|WYd}~dSd}~wwdS)z/Send both Slack and Email alerts if configured.slack_webhook_urlz&[WARNING] Failed to send Slack alert: N email_listz&[WARNING] Failed to send Email alert: )rsend_slack_alertrrrsend_email_alert)rsubject message_bodyrrrr send_alert)s   zAlertingHandler.send_alertcCsl|jd}dd|d|i}ddi}tj||t|d}|jdkr0td |jd |jt d dS) Nrtext*z* z Content-Typezapplication/json)headersdataz!Slack webhook failed with status z: zUCCESS] Slack alert sent.) rr requestsZpostrdumpsZ status_coderr$r)rr!r"rZ slack_messager&rrrrr7s   z AlertingHandler.send_slack_alertc Cs|js tddS|jd}|jd}|jd}|jd}|jd}|jdg}t|} || d<|| d <d || d <t||} | | ||| ||| Wdn1s`wYtd d |dS) Nz=[INFO] SMTP credentials not configured; skipping email alert.rrrrr rZSubjectZFromz, ZToz[SUCCESS] Email alert sent to: ) r rrr rjoinsmtplibZSMTPZstarttlsZloginZsendmailZ as_string) rr!r"rrrrr Z to_emailsmsgZserverrrrr Ds&      z AlertingHandler.send_email_alertN) __name__ __module__ __qualname__dictrr r#rr rrrrrs  r)rrr,r)Zemail.mime.textrrrrrrs