U hG@sdZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZmZmZdd lmZmZed Zed d d dd$eedddZed dddedddZeddZeddZeddZeddZeeeeefdd d!Zed"d#ZdS)%z Celery Tasks for Notification Processing This module contains Celery tasks for handling notification delivery, retry logic, and maintenance operations for the notification system. N)OptionalDictAnyList) shared_task)timezone)settings) NotificationNotificationChannelNotificationRule)NotificationServiceTelegramServicez#stream_processor.notification_tasksT<)bind max_retriesdefault_retry_delay)messagechat_idc CszHtdt}|||}tdd|did|pB|jdWStk r}zltd||jj |j krtd|jj d d |j |d n"td d t |ddWYSW5d}~XYnXdS)az Send a message via Telegram with retry logic. This task provides a simple interface for sending Telegram messages with automatic retry handling for failed deliveries. Args: message (str): Message text to send chat_id (str, optional): Target chat ID, uses default if not provided Returns: dict: Delivery result zSending Telegram messagez"Telegram message sent successfullyTresult message_id)successrrz!Failed to send Telegram message: z#Retrying Telegram message (attempt r )excz)Max retries exceeded for Telegram messageFrerrorretries_exhaustedN) loggerinfor send_messagegetdefault_chat_id Exceptionrrequestretriesrretrystr)selfrrtelegram_servicerer,=/var/www/html/StreamProcessor/src/apps/notifications/tasks.pysend_telegram_messages&    r.x)notification_idc Csztd|tjj|d}|rHtd|ddddWSt}||}|rtd|ddt|j |j j |j r|j ndd WSd |}t||r|jj|jkrtd |jjd d |jt|ddt|j || dWSWntjk rBd|d}t|d|dYStk r}z^td||jj|jkrtd|jjd d |j|ddt|ddWYSd}~XYnXdS)aG Process and deliver a notification. This task handles the delivery of individual notifications through their configured channels with retry logic for failed deliveries. Args: notification_id (str): UUID of the notification to process Returns: dict: Processing result zProcessing notification: )idz Notification z already sentT)rZ already_sentz delivered successfullyN)rr1 channel_typesent_atzFailed to deliver notification z(Retrying notification delivery (attempt r rrF)rr1rrzNotification with ID z not found)rrz Notification processing failed: z*Retrying notification processing (attempt r)rr r objectsr"is_sentr _deliver_notificationr(r2channelr3r4 isoformatr can_retryr%r&rr'r$ DoesNotExist)r)r1 notificationnotification_servicer error_msgr+r,r,r-process_notificationEsL        r?c Csztdtjjdtdd}d}|D]}z0d|_|j dgdt t |j |d 7}Wq.tk r}z4td |j d ||d |WYq.W5d }~XYq.Xq.td|dd|dWStk r }ztd|W5d }~XYnXd S)z Process all pending notifications that are ready for delivery. This periodic task finds notifications that are scheduled for delivery and queues them for processing. Returns: dict: Processing results z Processing pending notificationspending)statusZscheduled_at__lte scheduled_atr processingrA update_fieldsr zFailed to queue notification : zQueue failed: NzQueued z notifications for processingT)rZnotifications_queuedz(Pending notification processing failed: )rdebugr r5filterrnoworder_byrAsaver?delayr(r2r$r mark_failedr )pending_notificationsZ queued_countr<r+r,r,r-process_pending_notificationss4   rOc Cstz:tdt}|jdd}td|dd|dWStk rn}ztd|W5d }~XYnXd S) z Retry notifications that failed delivery but are eligible for retry. This periodic task identifies failed notifications that can be retried and queues them for another delivery attempt. Returns: dict: Retry results zRetrying failed notifications) max_age_hourszRetried z failed notificationsT)rZnotifications_retriedz'Failed notification retry task failed: N)rrGr retry_failed_notificationsr r$r)r= retry_countr+r,r,r-rRs   rRc Cs ztdddlm}d}d}t||d}t||d}tjjd|d}| }| tjjd |d }| }| tjjd |d } | } | ||| } t d |d |d| d| d d||| | dWSt k r} zt d| W5d} ~ XYnXdS)z Clean up old notification records to manage database size. This periodic task removes old notification records that are no longer needed for auditing or retry purposes. Returns: dict: Cleanup results zCleaning up old notificationsr timedelta)days completed)rAZ sent_at__ltfailed)rAZcreated_at__lt cancelledz Notification cleanup completed: z sent, z failed, z cancelled (total: rT)rZ sent_deletedZfailed_deletedZcancelled_deleted total_deletedz"Notification cleanup task failed: N)rrGdatetimerUrrIr r5rHcountdeleter r$r) rUZsent_retention_daysZfailed_retention_daysZ sent_cutoffZ failed_cutoffZold_sent sent_countZ old_failedZ failed_countZ old_cancelledcancelled_countr\r+r,r,r-cleanup_old_notificationssL    rbc Cs8ztdtjjdd}d}d}i}|D]}zFd}d}|jdkr|}|drd|dsnd }d }nJzt|d|dd }Wn0t k r} zd }d | }W5d} ~ XYnXnN|jd kr| }|dsd }d}n(|jdkr| }|dsd }d}||j krP||_ |j dgd|r2dnd} td|jd| d||d||j<|rp|d7}n|d7}Wq,t k r} z8td|jd| d t| d||j<|d7}W5d} ~ XYq,Xq,td|d|d d|||d!WSt k r2} ztd"| W5d} ~ XYnXdS)#a Validate notification channel configurations and connectivity. This periodic task checks that notification channels are properly configured and can successfully deliver test messages. Returns: dict: Validation results z Validating notification channelsT is_activerNtelegram bot_tokenrFzMissing bot_token or chat_id)rfrzTelegram validation failed: email recipientszNo recipients configuredwebhookurlzNo webhook URL configuredrdrD activatedZ deactivatedzChannel  z based on validation)validrr zFailed to validate channel rFzChannel validation completed: z valid, z invalid)rZvalid_channelsZinvalid_channelsresultsz Channel validation task failed: )rrGr r5rHr3get_telegram_configr"rr$get_email_configget_webhook_configrdrKr namerr() channelsZ valid_countZ invalid_countZvalidation_resultsr8Z channel_valid error_messageconfigr*r+Z status_changer,r,r-validate_notification_channels%sv              rv) event_typecontextc Cszptd|t}|j||d}|rPtd|d|t|dWStd|d|dd WSWn4tk r}ztd |W5d }~XYnXd S) a Send notifications based on event type and notification rules. This task provides the main interface for triggering notifications from other parts of the system based on events and rules. Args: event_type (str): Type of event that occurred context (Dict[str, Any]): Event context data for templates Returns: dict: Notification results z#Processing notification for event: ) template_typerxz*Notification sent successfully for event: T)rrwZ context_keysz!No notifications sent for event: FZ&no_applicable_rules_or_delivery_failed)rrwreasonz Rule-based notification failed: N) rr r send_notificationlistkeyswarningr$r)rwrxr=rr+r,r,r-send_notification_via_rules*  rc Cs`z"tdddlm}t}||dd}tjj|d }tjj|dd }tjj|d d }tjjd d  }d}|dkr||d }i}t jjd dD]B} tjj| |d} | | jdd  | jd d  d|| j <qt d|ddd d||||t |d|td WStk rZ} ztd| W5d} ~ XYnXdS)z Generate a periodic report about notification system performance. This task creates reports about notification delivery rates, channel performance, and system health. Returns: dict: Report data zGenerating notification reportrrTrP)hours)created_at__gterY)rrArZr@)rAdTrc)r8r)totalsentrZzNotification report: z.1fz% delivery rater ) rZ period_hourstotal_notificationssent_notificationsfailed_notificationsrN delivery_ratechannel_performance timestampz'Notification report generation failed: N)rrGr]rUrrIr r5rHr^r rrr roundr9r$r) rUend_time start_timerrrrNrZ channel_statsr8channel_notificationsr+r,r,r-generate_notification_reports\          r)N)__doc__loggingtypingrrrrceleryr django.utilsr django.confrmodelsr r r servicesr r getLoggerrr(r.r?rOrRrbrvrrr,r,r,r-s0     . J 0  C a.