Format: application/x-python-code
1 o
2
3 h� � @ s> d dl Z d dlZd dlZd dlZd dlmZ G dd� d�ZdS )� N)�MIMETextc @ s: e Zd Zdefdd�Zdd� Zdd� Zdd � Zd
4 d � Zd S )
5 �AlertingHandler�alerting_configc C s| |pi | _ i | _d| j v r| �| j d �| _d S d| j v r<| j �d�| j �d�| j �d�| j �d�| j �d�d�| _d S d S )NZsmtp_secret_name� smtp_server� smtp_port�
6 smtp_username�
7 smtp_password�
8 from_email�r r r r r )r �smtp_credentials�!get_smtp_credentials_from_secrets�get)�selfr � r �EC:\Users\shubh\Videos\recall\recall_audit_sampler\alerting_handler.py�__init__ s
9
10
11
12
13
14
15
16 ��zAlertingHandler.__init__c
17 C s� z't �d�}|j|d�}t�|d �}|d t|d �|d |d |d d �W S tyD } ztd
18 t|�� �� i W Y d }~S d }~ww ) NZsecretsmanager)ZSecretIdZ SecretStringr r r r r r
19 z>[ERROR] Failed to fetch SMTP credentials from SecretsManager: ) �boto3�clientZget_secret_value�json�loads�int� Exception�print�str)r � secret_namer �responseZsecret�er r r r s
20
21 ���z1AlertingHandler.get_smtp_credentials_from_secretsc
22 C s� d| j v r)z| �||� W n ty( } ztdt|�� �� W Y d}~nd}~ww d| j v rTz | �||� W dS tyS } ztdt|�� �� W Y d}~dS d}~ww dS )z/Send both Slack and Email alerts if configured.�slack_webhook_urlz&[WARNING] Failed to send Slack alert: N�
23 email_listz&[WARNING] Failed to send Email alert: )r �send_slack_alertr r r �send_email_alert)r �subject� message_bodyr r r r �
24 send_alert) s
25 ��
26 ���zAlertingHandler.send_alertc C sl | j �d�}dd|� d|� �i}ddi}tj||t�|�d�}|jdkr0td |j� d
27 |j� ���t d � d S ) Nr �text�*z*
28 z Content-Typezapplication/json)�headers�data�� z!Slack webhook failed with status z: z[SUCCESS] Slack alert sent.)
29 r r
30 �requestsZpostr �dumpsZ status_coder r$ r )r r! r" r Z
31 slack_messager& r r r r r 7 s �
32 z AlertingHandler.send_slack_alertc C s� | j s td� d S | j d }| j d }| j d }| j d }| j d }| j�dg �}t|�} || d< || d < d
33 �|�| d < t�||��}
34 |
35 �� |
36 � ||� |
37 �
38 ||| � � � W d � n1 s`w Y td d
39 �|�� �� d S )
40 Nz=[INFO] SMTP credentials not configured; skipping email alert.r r r r r r ZSubjectZFromz, ZToz [SUCCESS] Email alert sent to: ) r r r r
41 r �join�smtplibZSMTPZstarttlsZloginZsendmailZ as_string) r r! r" r r r r r Z to_emails�msgZserverr r r r D s&
42
43
44
45
46 �z AlertingHandler.send_email_alertN) �__name__�
47 __module__� __qualname__�dictr r r# r r r r r r r s
48 r )r r r, r) Zemail.mime.textr r r r r r �<module> s