U hI@sdZddlmZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZmZmZmZmZmZddlZGd d d e e ZGd d d e e ZGd dde e e ZGddde e ZdS)a Notification Models for Stream Processor Application This module contains Django models for managing notifications, including Telegram messages, email alerts, and other notification channels. These models track notification history, delivery status, and configuration settings. )models)User)timezone)MinValueValidatorMaxValueValidator)TimestampedModel UUIDModel StatusModel)validate_name_formatvalidate_json_configurationvalidate_webhook_configvalidate_email_configvalidate_telegram_configvalidate_template_contentNc@seZdZdZejddegddZdddd d d gZejd ed dZ ej ddddZ ej e ddZejdededgddZejdededgddZejdededgddZejeejdd d!ZGd"d#d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,S)-NotificationChannelaI Model representing a notification delivery channel. Notification channels define how and where notifications should be sent, including configuration for different service types like Telegram, email, webhooks, and other messaging platforms. Attributes: name (CharField): Human-readable name for the channel channel_type (CharField): Type of notification channel is_active (BooleanField): Whether the channel is currently active configuration (JSONField): Channel-specific configuration data rate_limit (PositiveIntegerField): Maximum messages per minute retry_attempts (PositiveIntegerField): Number of retry attempts timeout_seconds (PositiveIntegerField): Timeout for delivery attempts created_by (ForeignKey): User who created this channel dTz0Human-readable name for the notification channel) max_lengthunique validators help_text)telegramz Telegram Bot)emailZEmail)webhookZWebhook)ZslackZSlack)ZdiscordZDiscord)ZsmsZSMSzType of notification channelrchoicesrz+Whether this notification channel is activedefaultdb_indexrz)Channel-specific configuration parametersrriz%Maximum number of messages per minuterrrr z.Number of retry attempts for failed deliveriesi,z(Timeout for delivery attempts in secondsZnotification_channelsz*User who created this notification channel on_delete related_namerc@s6eZdZdZdZdgZejdgdejdgdgZdS)zNotificationChannel.MetazNotification ChannelzNotification Channelsname channel_typefields is_activeN __name__ __module__ __qualname__ verbose_nameverbose_name_pluralorderingrIndexindexesr7r7>/var/www/html/StreamProcessor/src/apps/notifications/models.pyMetais   r9cCs|jd|jdS)z2String representation of the notification channel. ())r)r*selfr7r7r8__str__rszNotificationChannel.__str__cCs,|jdkr(|jdd|jdddSiS)z$Get Telegram-specific configuration.r bot_tokenchat_id)r?rAr* configurationgetr<r7r7r8get_telegram_configvs    z'NotificationChannel.get_telegram_configcCs,|jdkr(|jdg|jdddSiS)z!Get email-specific configuration.r recipientssubject_prefixz[Stream Processor])rFrGrBr<r7r7r8get_email_configs    z$NotificationChannel.get_email_configcCs8|jdkr4|jdd|jdi|jdddSiS)z#Get webhook-specific configuration.rurlr@headersmethodPOST)rIrJrKrBr<r7r7r8get_webhook_configs     z&NotificationChannel.get_webhook_configN)r/r0r1__doc__r CharFieldr r)Z CHANNEL_TYPESr* BooleanFieldr- JSONFielddictrCPositiveIntegerFieldrr rate_limitretry_attemptsZtimeout_seconds ForeignKeyrCASCADE created_byr9r>rErHrMr7r7r7r8rsl     rc@seZdZdZejddddZdddd d d d d gZejdeddZejddddZ ej ddZ ej ddddZ ejeddZejeddZGdddZddZdd Zd!d"Zd#S)$NotificationTemplatea Model for notification message templates. Templates define reusable message formats for different types of notifications, allowing for consistent messaging and easy customization of notification content. Attributes: name (CharField): Name for the template template_type (CharField): Type of notification this template is for subject_template (CharField): Subject line template (for email) message_template (TextField): Main message content template is_active (BooleanField): Whether the template is active variables (JSONField): Available template variables channel_types (JSONField): Compatible channel types rTz#Name for this notification templaterrrZjingle_detectedzJingle DetectedZad_break_startedzAd Break StartedZad_break_endedzAd Break EndedZstream_startedzStream StartedZstream_stoppedzStream StoppedZ stream_errorz Stream Error)Z system_alertz System Alert)Z health_checkz Health Checkr z2Type of notification this template is designed forrz4Subject line template (used for email notifications)rblankrz8Main message content template with variable placeholdersrz'Whether this template is active for userz-List of available variables for this templaterz-List of compatible notification channel typesc@s8eZdZdZdZddgZejdgdejdgdgZdS)zNotificationTemplate.MetazNotification TemplatezNotification Templates template_typer)r+r-Nr.r7r7r7r8r9s   r9cCs|jd|jdS)z3String representation of the notification template.r:r;)r)rer<r7r7r8r>szNotificationTemplate.__str__c CsFz|jjf|WStk r@}zd|WYSd}~XYnXdS)z Render the message template with provided context variables. Args: context (dict): Variables to substitute in the template Returns: str: Rendered message content z!Template error: Missing variable N)message_templateformatKeyErrorr=contexter7r7r8render_messages z#NotificationTemplate.render_messagec CsZ|jsd|jdSz|jjf|WStk rT}zd|WYSd}~XYnXdS)z Render the subject template with provided context variables. Args: context (dict): Variables to substitute in the template Returns: str: Rendered subject line [z] Notificationz Subject error: Missing variable N)subject_templaterergrhrir7r7r8render_subjects z#NotificationTemplate.render_subjectN)r/r0r1rNrrOr)ZTEMPLATE_TYPESrern TextFieldrfrPr-rQlist variablesZ channel_typesr9r>rlror7r7r7r8rYsZ  rYc@s eZdZdZejeejdddZeje ej dddddZ ej dd d Z ej ddd d Zejd dZejedddZejejddZejddddZejdddZejdddZej dddd ZGdddZddZdd Zd!d"Z d#d$Z!d,d&d'Z"d(d)Z#d*d+Z$d%S)- Notificationa Model representing an individual notification instance. This model tracks individual notification messages that have been sent or are pending delivery, including their content, delivery status, and any error information. Attributes: channel (ForeignKey): Notification channel to use for delivery template (ForeignKey): Template used for this notification recipient (CharField): Target recipient identifier subject (CharField): Notification subject line message (TextField): Notification message content context_data (JSONField): Variables used for template rendering scheduled_at (DateTimeField): When the notification should be sent sent_at (DateTimeField): When the notification was actually sent delivery_attempts (PositiveIntegerField): Number of delivery attempts error_message (TextField): Last error message if delivery failed external_id (CharField): External service message ID notificationsz(Notification channel to use for deliveryr&Tz+Template used to generate this notification)r'nullrcr(rraz2Target recipient identifier (email, chat ID, etc.))rrzNotification subject linerbzNotification message contentrdz%Variables used for template rendering)rrcrz%When this notification should be sentrz(When this notification was actually sentrurcrrz Number of delivery attempts madez%Last error message if delivery failed)rcrrz(External service message ID for trackingc@sPeZdZdZdZdgZejddgdejdgdejdgdejdgdgZd S) zNotification.Metars Notificationsz -scheduled_atchannelstatusr+ scheduled_atsent_atNr.r7r7r7r8r9cs   r9cCs"|jjd|jp|jddS)z*String representation of the notification.z - N2)rxr)subjectmessager<r7r7r8r>nszNotification.__str__cCs|jdko|j S)z.Check if the notification is pending delivery.pendingryr{r<r7r7r8 is_pendingrszNotification.is_pendingcCs|jdko|jdk S)z5Check if the notification has been successfully sent. completedNrr<r7r7r8is_sentvszNotification.is_sentcCs |jdkS)z.Check if the notification delivery has failed.failed)ryr<r7r7r8 is_failedzszNotification.is_failedNcCs0d|_t|_|r||_|jdddgddS)z+Mark the notification as successfully sent.rryr{ external_id update_fieldsN)ryrnowr{rsave)r=rr7r7r8 mark_sent~s  zNotification.mark_sentcCs0d|_||_|jd7_|jdddgddS)z3Mark the notification as failed with error message.rr!ry error_messagedelivery_attemptsrN)ryrrr)r=rr7r7r8 mark_failedszNotification.mark_failedcCs|jdko|j|jjkS)z)Check if the notification can be retried.r)ryrrxrUr<r7r7r8 can_retrys  zNotification.can_retry)N)%r/r0r1rNrrVrrWrxrYSET_NULLtemplaterO recipientr}rpr~rQrR context_data DateTimeFieldrrrzr{rSrrrr9r>rrrrrrr7r7r7r8rs s|   rsc @seZdZdZejddddZdddd d d d d dg ZejdeddZej e ej dddZ ej e ej dddZejddddZejeddZejdededgddZejdddd Zejdd!dZGd"d#d#Zd$d%Zd&d'Zd(d)Zd*d+Z d,S)-NotificationRulear Model for automatic notification rules. Rules define when and how notifications should be automatically triggered based on system events, allowing for automated alerting and monitoring without manual intervention. Attributes: name (CharField): Name for the notification rule event_type (CharField): Type of event that triggers this rule channel (ForeignKey): Default channel for notifications template (ForeignKey): Default template for notifications is_active (BooleanField): Whether the rule is currently active conditions (JSONField): Conditions that must be met for triggering throttle_minutes (PositiveIntegerField): Minimum time between notifications last_triggered (DateTimeField): When this rule was last triggered priority (IntegerField): Rule priority for ordering rTzName for this notification rulerZr[r\r]r^r_r`)Zhigh_error_ratezHigh Error Rate)Zsystem_startupzSystem Startup)Zsystem_shutdownzSystem Shutdownr z2Type of event that triggers this notification rulerZrulesz*Default notification channel for this ruler&z+Default notification template for this rulez(Whether this notification rule is activerz5Additional conditions that must be met for triggeringrr%riz4Minimum minutes between notifications from this ruler"z!When this rule was last triggeredrvz=Rule priority for ordering (higher numbers = higher priority)c@sDeZdZdZdZddgZejdgdejdgdejdgdgZd S) zNotificationRule.MetazNotification RulezNotification Rulesz -priorityr) event_typer+r-priorityNr.r7r7r7r8r9s   r9cCs|jd|jdS)z/String representation of the notification rule.r:r;)r)rr<r7r7r8r>szNotificationRule.__str__cCs4|js dS|jsdSt|j}||jdkS)z7Check if the rule can be triggered based on throttling.FT<)r-last_triggeredrr total_secondsthrottle_minutes)r=Ztime_since_lastr7r7r8 can_triggers zNotificationRule.can_triggercCst|_|jdgddS)z0Mark the rule as triggered and update timestamp.rrN)rrrrr<r7r7r8mark_triggered s zNotificationRule.mark_triggeredcCsB|js dS|jD](\}}||kr*dS|||krdSqdS)z Check if event data meets the rule conditions. Args: event_data (dict): Event data to check against conditions Returns: bool: True if conditions are met, False otherwise TF) conditionsitems)r=Z event_datakeyZexpected_valuer7r7r8check_conditionss  z!NotificationRule.check_conditionsN)!r/r0r1rNrrOr)Z EVENT_TYPESrrVrrWrxrYrrPr-rQrRrrSrrrrr IntegerFieldrr9r>rrrr7r7r7r8rsz   r)rN django.dbrdjango.contrib.auth.modelsr django.utilsrdjango.core.validatorsrrZapps.core.modelsrrr apps.core.validatorsr r r r rrjsonrrYrsrr7r7r7r8s    }v