U
hW @ s d Z ddlZddlZddlZddlZddlmZmZmZm Z ddl
m
Z
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ d d
lmZmZmZmZ G d d d ZG d
d dZG dd dZG dd dZ dS )a
Notification Services
This module contains services for sending notifications through various
channels including Telegram, email, webhooks, and other messaging platforms.
It provides a unified interface for notification delivery with retry logic
and error handling.
N)OptionalDictAnyList)datetime timedelta)settings)timezone) send_mail)render_to_string )NotificationChannelNotificationTemplate NotificationNotificationRulec @ s e Zd ZdZdee ee dddZddddZdeee eee ee
f d d d
Z deeee e ee
f dddZ ee
eeeedddZd eeee ee eedddZdS ) TelegramServicea
Service for sending notifications via Telegram Bot API.
This service handles Telegram bot integration, message formatting,
and delivery with automatic retry logic for failed messages.
Attributes:
logger (Logger): Logger instance for recording operations
bot_token (str): Telegram bot token from configuration
default_chat_id (str): Default chat ID for notifications
api_base_url (str): Base URL for Telegram Bot API
N) bot_tokenchat_idc C sN t d| _|p tjdd| _|p0tjdd| _d| j | _| dS )z
Initialize the Telegram service.
Args:
bot_token (str, optional): Telegram bot token, uses settings if not provided
chat_id (str, optional): Default chat ID, uses settings if not provided
zstream_processor.telegram BOT_TOKEN CHAT_IDz https://api.telegram.org/botN)
logging getLoggerloggerr TELEGRAM_CONFIGgetr default_chat_id api_base_url_validate_config)selfr r r @/var/www/html/StreamProcessor/src/apps/notifications/services.py__init__) s
zTelegramService.__init__)returnc C s< | j s| jd dS t| j dk r,td| jd dS )z
Validate Telegram configuration.
Raises:
ValueError: If required configuration is missing
z:Telegram bot token not configured - notifications disabledN z!Invalid Telegram bot token formatz)Telegram service initialized successfully)r r warninglen
ValueErrorinfor r r r! r > s z TelegramService._validate_configHTMLT)messager
parse_modedisable_web_page_previewr# c
C s | j s | jd dddS |p$| j}|s@| jd dddS | j d}||||d}z| jd | tj||d
d d id
}|j dkr|
} | dr| jd| d d | W S | dd}
| j d|
t
d|
n6| j d|j d|j t
d|j d|j W n^ tjk
rN | j d Y n: tj
k
r } z| j d| W 5 d} ~ X Y nX dS ) a
Send a text message via Telegram.
Args:
message (str): Message text to send
chat_id (str, optional): Target chat ID, uses default if not provided
parse_mode (str): Message parse mode (HTML, Markdown, or None)
disable_web_page_preview (bool): Disable link previews
Returns:
Dict[str, Any]: API response data
Raises:
requests.RequestException: If API request fails
z8Telegram bot token not configured, skipping notificationskippedZbot_token_not_configured)statusreason-No chat ID provided and no default configuredZ
no_chat_idz /sendMessage)r textr, r- z!Sending Telegram message to chat Content-Typeapplication/json)jsontimeoutheaders okz$Telegram message sent successfully: result
message_id description
Unknown errorTelegram API error: z HTTP error : HTTP z Telegram API request timed outz!Failed to send Telegram message: N)r r r( r r% r debugrequestspost status_coder6 r errorRequestExceptionr2 Timeout) r r+ r r, r- target_chat_idurlpayloadresponser; error_msger r r! send_messageN sJ
z TelegramService.send_messager )
photo_pathcaptionr r# c
C s6 |p| j }|std| j d}zt|d}d|i}||d}tj|||dd} | jdkr| }
|
d r| j
d
|
W 5 Q R W S |
d d } t d
| nt d| j d| j W 5 Q R X W n^ t
k
r | j d| Y n8 tk
r0 } z| j d| W 5 d} ~ X Y nX dS )a3
Send a photo with caption via Telegram.
Args:
photo_path (str): Path to the photo file
caption (str): Photo caption text
chat_id (str, optional): Target chat ID
Returns:
Dict[str, Any]: API response data
r1 z
/sendPhotorbZphoto)r rQ < )filesdatar7 r9 r: z Telegram photo sent successfullyr= r> r? rA r@ zPhoto file not found: z Failed to send Telegram photo: N)r r' r openrC rD rE r6 r r r( rG r2 FileNotFoundErrorrF Exception)
r rP rQ r rI rJ Z
photo_filerT rU rL r; rM rN r r r!
send_photo s2
(zTelegramService.send_photo) jingle_name
similarity iframe_path
iframe_urldetection_timer# c C s* d| d|dd| d| d| d
S )a
Format a jingle detection message for Telegram.
Args:
jingle_name (str): Name of detected jingle
similarity (float): Similarity score
iframe_path (str): Path to extracted frame
iframe_url (str): URL to view the frame
detection_time (datetime): When detection occurred
Returns:
str: Formatted Telegram message
u/ 🎵 Jingle Detected 🎵
Name: z
Similarity: z.3fz
Frame Path: z#
Frame URL: View Frame
Detection Time: %Y-%m-%d %H:%M:%Sstrftime)r rZ r[ r\ r] r^ r r r! format_jingle_detection_message s (z/TelegramService.format_jingle_detection_messageF) channel_name
start_timeend_timeduration is_estimatedr# c C s` |rD|rD|rdnd}d| d| d d| d d| d| d S d
| d| d d S d S )
a
Format an ad break message for Telegram.
Args:
channel_name (str): Name of the channel
start_time (datetime): Ad break start time
end_time (datetime, optional): Ad break end time
duration (int, optional): Duration in seconds
is_estimated (bool): Whether duration is estimated
Returns:
str: Formatted Telegram message
Z EstimatedZActualu4 🚨 Ad Break Detected 🚨
Channel: z
Start Time: r_ z
End Time: z
z Duration: z secondsu3 🔊 Ad Break Started 🔊
Channel: z
Status: In Progress...Nr` )r rc rd re rf rg Z
duration_typer r r! format_ad_break_message s . z'TelegramService.format_ad_break_message)NN)Nr* T)r N)NNF)__name__
__module__ __qualname____doc__r strr" r boolr r rO rY floatr rb intrh r r r r! r sN
N
6 " r c @ s> e Zd ZdZdd Zdeeee ee ee edddZ dS ) EmailServicez
Service for sending email notifications.
This service handles email delivery using Django's email framework
with support for HTML templates and attachments.
c C s t d| _dS )z Initialize the email service.zstream_processor.emailNr r r r) r r r! r" s zEmailService.__init__N)subjectr+
recipients html_message
from_emailr# c
C sr z2t |||||dd}| jd| d |dkW S tk
rl } z | jd| W Y dS d}~X Y nX dS )a
Send an email notification.
Args:
subject (str): Email subject line
message (str): Plain text message content
recipients (List[str]): List of recipient email addresses
html_message (str, optional): HTML message content
from_email (str, optional): Sender email address
Returns:
bool: True if email was sent successfully
F)rs r+ rv recipient_listru
fail_silentlyzEmail sent to z recipientsr zFailed to send email: N)r
r r( rX rF )r rs r+ rt ru rv Z
sent_countrN r r r!
send_email s
zEmailService.send_email)NN)
ri rj rk rl r" rm r r rn ry r r r r! rq s rq c @ sF e Zd ZdZdd Zd
eeeef eeeef ee e
ddd Z dS ) WebhookServicez
Service for sending webhook notifications.
This service handles HTTP webhook delivery with configurable
endpoints, headers, and payload formats.
c C s t d| _dS )z Initialize the webhook service.zstream_processor.webhooksNrr r) r r r! r" M s zWebhookService.__init__NPOSTr3 )rJ rU r8 methodr7 r# c
C s zvddi}|r| | tj| ||||d}|jdkrR| jd| W dS | jd|j d|j W d S W n@ t k
r } z"| jd
| d| W Y d S d }~X Y nX d S ) a
Send a webhook notification.
Args:
url (str): Webhook endpoint URL
data (Dict[str, Any]): Data to send in the webhook
headers (Dict[str, str], optional): HTTP headers
method (str): HTTP method (POST, PUT, PATCH)
timeout (int): Request timeout in seconds
Returns:
bool: True if webhook was sent successfully
r4 r5 )r| rJ r6 r8 r7 )r9 z Webhook sent successfully to TzWebhook failed with status r@ FzFailed to send webhook to N)
updaterC requestupperrE r r( rF r2 rX ) r rJ rU r8 r| r7 Zwebhook_headersrL rN r r r! send_webhookQ s&
zWebhookService.send_webhook)Nr{ r3 ) ri rj rk rl r" rm r r r rp rn r r r r r! rz E s
rz c @ s e Zd ZdZdd Zdeeeef ee ee e dddZ
de eeef ee e dd d
Z e
e d d d
ZdeedddZdS )NotificationServicea
Main notification service that coordinates different notification channels.
This service provides a unified interface for sending notifications
through various channels with automatic channel selection, template
rendering, and retry logic.
c C s( t d| _t | _t | _t | _dS )z$Initialize the notification service.z stream_processor.notificationsN) r r r r telegramrq emailrz webhookr) r r r! r" s z NotificationService.__init__N)
template_typecontextrc recipientr# c
C s z~t jj|ddd}| s8| jd| W dS d}|D ]4}| r@||r@| |||r@|d7 }|
q@|dkW S t k
r } z | j d| W Y dS d }~X Y nX d S )
a
Send a notification using the specified template and context.
Args:
template_type (str): Type of notification template to use
context (Dict[str, Any]): Template variables for rendering
channel_name (str, optional): Specific channel to use
recipient (str, optional): Specific recipient override
Returns:
bool: True if notification was sent successfully
T)
event_type is_activez -priorityz,No notification rules found for event type: Fr r z Failed to send notification: N)
r objectsfilterorder_byexistsr r% can_triggercheck_conditions_send_rule_notificationmark_triggeredrX rF ) r r r rc r rulesZ
success_countrulerN r r r! send_notification s(
z%NotificationService.send_notification)r r r r# c
C s z|j |}|j |}|}|sn|jjdkr@|j d}n.|jjdkrn|j dg }|rj|d nd}|s| j d|j
W dS t j j
|j|j ||||d d
}| |W S tk
r } z | j d | W Y dS d} ~ X Y nX dS ) al
Send a notification based on a specific rule.
Args:
rule (NotificationRule): Notification rule to apply
context (Dict[str, Any]): Template rendering context
recipient (str, optional): Recipient override
Returns:
bool: True if notification was sent successfully
r r r rt r Nz!No recipient available for rule: Fpending)channeltemplater rs r+ context_datar/ z"Failed to send rule notification: )r render_subjectrender_messager channel_typeget_telegram_configr get_email_configr rF namer r create_deliver_notificationrX )
r r r r rs r+ Ztarget_recipientrt notificationrN r r r! r s4 z+NotificationService._send_rule_notification)r r# c
C s zN|j }|jdkr`| }| jj|j|jd}|di d}|j|rRt |ndd W dS |jdkr| j
j |j |j|jgd }|r| W dS |
d
W d S n|jd kr"| }| jj|d
|j |j|j|j d|di |ddd}|r| W dS |
d W d S n,| jd|j |
d|j W d S W nR tk
r } z2| jd|j d| |
t | W Y d S d}~X Y nX dS )z
Deliver a notification via its configured channel.
Args:
notification (Notification): Notification to deliver
Returns:
bool: True if delivery was successful
r )r+ r r; r< N) external_idTr )rs r+ rt zEmail delivery failedFr rJ )rs r+ r timestampr8 r| r{ )rJ rU r8 r| zWebhook delivery failedzUnsupported channel type: z Failed to deliver notification r@ )r r r r rO r+ r r mark_sentrm r ry rs mark_failedget_webhook_configr r r scheduled_at isoformatr rF rX id)r r r configr; r< successrN r r r! r s\
z)NotificationService._deliver_notification )
max_age_hoursr# c C sx t t|d }tjjd|d}d}|D ],}| r*d|_| | |r*|d7 }q*|dkrt| j
d| d |S ) z
Retry failed notifications that are eligible for retry.
Args:
max_age_hours (int): Maximum age of notifications to retry
Returns:
int: Number of notifications retried
)hoursfailed)r/ created_at__gter r r zRetried z failed notifications) r nowr r r r can_retryr/ saver r r( )r r cutoff_timefailed_notifications retry_countr r r r! retry_failed_notifications? s
z.NotificationService.retry_failed_notifications)NN)N)r )ri rj rk rl r" rm r r r rn r r r r r rp r r r r r! r s(
2
5Ir ) rl rC r6 timer typingr r r r r r django.confr django.utilsr django.core.mailr
django.template.loaderr modelsr
r r r r rq rz r r r r r! s x3<