@i| dZddlZddlmZmZddlmZddlmZddl m Z  ddl Z ejeZGddZGd d eZGd d eZGd deZGddZy#e $rdZ YQwxYw)z) Notification system for cronjob events. N)ListOptional)settings) send_mail) CronExecutionc2eZdZdZdedefdZddedefdZy) NotificationBackendz%Base class for notification backends. execution error_messagect)zSend failure notification.)NotImplementedError)selfr r s V/home/cursorai/projects/django-cronjob-utils/src/django_cronjob_utils/notifications.pynotify_failurez"NotificationBackend.notify_failures!!messagecy)z%Send success notification (optional).N)rr rs rnotify_successz"NotificationBackend.notify_successs rN)__name__ __module__ __qualname____doc__rstrrrrrrr r s+/" "c"   rr c,eZdZdZdefdZdedefdZy) EmailBackendzEmail notification backend.configc|jdg|_|jdtj|_|jdd|_y)N recipients from_emailsubject_templatezCronjob Failed: {task_name})getr!rDEFAULT_FROM_EMAILr"r#rrs r__init__zEmailBackend.__init__#sA **\26 **\83N3NO & +=?\ ]rr r cL|jstjdy|jj |j |j }d|j d|j d|jd|jd|jxsd d |d |jxsd d } t|||j|jd tjd|j y#t$r"}tjd|Yd}~yd}~wwxYw)z$Send failure notification via email.z4Email backend configured but no recipients specifiedN) task_name task_codez! Cronjob Execution Failed Task:  (z) Execution Date: z Started: z Error Code: N/Az Error Message: z Message: z4 Please check the execution logs for more details. F)subjectrr"recipient_list fail_silentlyz(Email notification sent for failed task z#Failed to send email notification: )r!loggerwarningr#formatr)r*execution_datestarted error_coderrr"info Exceptionerror)rr r r-res rrzEmailBackend.notify_failure(s9 NNQ R ''..))))/  2i1123))*+      ! ! *U +, e" D ??##   KKB9CVCVBWX Y D LL>qcB C C Ds1AC88 D#DD#N rrrrdictr'rrrrrrrr s)%^t^ &D &Dc&Drrc,eZdZdZdefdZdedefdZy) SlackBackendzSlack notification backend.rc|jd|_|jdd|_|jdd|_|js t dy)N webhook_urlchannelz #cronjobsusernameCronbotzSlack webhook_url is required)r$r?r@rA ValueErrorr&s rr'zSlackBackend.__init__TsQ!::m4zz)[9  :y9 <= = rr r c zttjdy|j|jd|j dd|j d|j ddd d t|jdd d |jjd dd d |jxsddd d|dddd gdgd} tj|j|d}|jtjd|j y#t $r"}tjd|Yd}~yd}~wwxYw)z$Send failure notification via Slack.Nz[requests library is required for Slack notifications. Install it with: pip install requestsu❌ Cronjob Failed: dangerTaskr+)T)titlevalueshortDateStarted%Y-%m-%d %H:%M:%Sz Error Coder,z Error MessageiF)colorfields)r@rAtext attachments jsontimeoutz(Slack notification sent for failed task z#Failed to send Slack notification: )requestsr0r8r@rAr)r*rr3r4strftimer5postr?raise_for_statusr6r7)rr r payloadresponser9s rrzSlackBackend.notify_failure\sE   LLv w || *9+>+>*?@&"(i6I6I5J"YM`M`Laab3cnrs"(3y7O7O3P[_`"+i6G6G6P6PQd6eptu".9M9M9VQVaef"1M$3qcB C C Ds:AD D:D55D:Nr:rrrr=r=Qs'%>t>D DcDrr=c,eZdZdZdefdZdedefdZy)TelegramBackendzTelegram notification backend.rc|jd|_|jd|_|js td|js tdd|jd|_y)N bot_tokenchat_idzTelegram bot_token is requiredzTelegram chat_id is requiredzhttps://api.telegram.org/botz /sendMessage)r$r_r`rCapi_urlr&s rr'zTelegramBackend.__init__s]K0zz), ~~=> >||;< <5dnn5E\R rr r cttjdyd|jd|jd|j d|j jdd|jxsd d |dd  }|j|d d } tj|j|d}|jtjd|jy#t$r"}tjd|Yd}~yd}~wwxYw)z'Send failure notification via Telegram.Nz^requests library is required for Telegram notifications. Install it with: pip install requestsu(❌ Cronjob Failed Task: r+z) Date: z Started: rMz Error Code: r,z Error: iHTML)r`rP parse_moderRrSz+Telegram notification sent for failed task z&Failed to send Telegram notification: )rVr0r8r)r*r3r4rWr5r`rXrarYr6r7)rr r rrZr[r9s rrzTelegramBackend.notify_failures   LLy z %//093F3F2GH%4456(0099:MNOP""+"6"6"?%!@A+ET23  5 ||    G}}T\\LH  % % ' KKEiFYFYEZ[ \ G LLA!E F F Gs AC D (DD Nr:rrrr]r]|s)( St SG GcGrr]c>eZdZdZdZdZdedefdZd dedefdZ y ) NotificationManagerz)Manages notifications for cronjob events.c2g|_|jy)N)backends_load_backends)rs rr'zNotificationManager.__init__s35  rcttdijdi}|jdg}d|vr7 |jdi}|jj t |d|vr7 |jdi}|jj t|d |vr8 |jd i}|jj t|yy#t $r"}tjd|Yd}~d}~wwxYw#t $r"}tjd|Yd}~d}~wwxYw#t $r"}tjd |Yd}~yd}~wwxYw) z&Load configured notification backends. CRONJOB_UTILS NOTIFICATIONS on_failureemailz$Failed to initialize Email backend: Nslackz$Failed to initialize Slack backend: telegramz'Failed to initialize Telegram backend: ) getattrrr$rhappendrr7r0r8r=r])rrrm email_configr9 slack_configtelegram_configs rriz"NotificationManager._load_backendssT?B7;;ORPZZ b1 j  I%zz'26  $$\,%?@ j  I%zz'26  $$\,%?@  # L"(**Z"< $$__%EF $ I CA3GHH I I CA3GHH I L FqcJKK LsG6C'46D/6E' D0D  D ED;;E E. E))E.r r c|jD]} |j||y#t$r9}tj d|j j d|Yd}~Vd}~wwxYw)zNotify stakeholders of failure.Notification backend failed: N)rhrr7r0r8 __class__r)rr r backendr9s rrz"NotificationManager.notify_failuresg}}G _&&y-@% _ 4W5F5F5O5O4PPYZ[Y\]^^ _s& A(/A##A(rcttdijdi}|jdg}|sy|jD]L}|jj j jdd|vs: |j||Ny#t$r9}tjd|jj d|Yd}~d}~wwxYw) z/Notify stakeholders of success (if configured).rkrl on_successNrzrrwrx) rqrr$rhryrlowerreplacerr7r0r8)rr rrr|rzr9s rrz"NotificationManager.notify_successs?B7;;ORPZZ b1  }}G  ))//199)RHJVc**9g>%!cLL#89J9J9S9S8TT]^_]`!abbcs>B C/CCNr) rrrrr'rirrrrrrrrfrfs<3L6_ _c_ c c crrf)rloggingtypingrr django.confrdjango.core.mailrdjango_cronjob_utils.modelsrrV ImportError getLoggerrr0r rr=r]rfrrrrs! &5   8 $   .D&.Db(D&(DV(G)(GV7c7csHsA..A87A8