w " d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
d dlmZ d dlmZmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lm Z d dl m Z m Z m Z m!Z!m"Z" d dl#m$Z$ ej% e& Z'd Z( G d dee Z)dS ) N)Path)IAIDTokenError)APIError) EventsAPI)CoreIContactMessageType) MessageType) MessageSink
MessageSource)TheSink)IContactThrottle)cPanel)Plesk) HostingPanel) await_for create_task_and_log_exceptionsrecurring_checkretry_onScope)DAYc b K t d| | t d d S )Nz[Can't get recommendations for the dashboard due to iaid token error, reason: %s. Attempt %sd seconds)loggerwarningr )eis \/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/icontact_sender.pyasync_log_on_errorr # s@
NN 3 c c e Zd Zej j Zej Z fdZ
d Z dddZ de
fdZd Z ee ed
d d e
eeed d e
defd Z ee d Z xZS )IContactSenderc t j |i | g | _ t t j dz | _ d S )N icontact_generic_notifications)super__init___tasksr r TMPDIR_notification_flag_path)selfargskwargs __class__s r r' zIContactSender.__init__1 sG $)&))) @ @
$$$r! c
K d S N )r+ loops r create_sinkzIContactSender.create_sink8 s r! Nuserc | K |d S t j ||| sd S | j t j || d {V }|rmt j || t j |t t j
| }| j | d {V d S d S )Nr4 ) message_typeparamsr5 )r7 timestamp
template_args)
r
may_be_notified_panelnotifyr GENERICrefreshr IContactSentinttime_sinkprocess_message)r+ r7 r8 period_limitr5 r: sent_messages r _send_icontact_messagez%IContactSender._send_icontact_message; s F /
F "k 0 0,4 1
; $\ = = = =&3 ) dikk** + L
*,,\:: : : : : : : : : : ; ;r! sinkc K || _ t | _ | j j t j t
j fv r t
|| j g| _ d S d S r0 ) rC r r< NAMEr r r generic_notificationsr( )r+ r2 rH s r
create_sourcez IContactSender.create_sourceY s[
"nn ; UZ 8 8 8.$4 DKKK 9 8r! c ~ K | j D ]}| t j | j ddi d {V d S )Nreturn_exceptionsT)r( cancelasynciogather)r+ tasks r shutdownzIContactSender.shutdownc sU K D KKMMMMndkBTBBBBBBBBBBBr!
r T)on_error max_triessilentlogreturnc K g }| j r;| j j t z t j k r5t
j d {V }| j dd |S )Ni T)modeexist_ok) r* existsstatst_mtimer rB r notificationtouch)r+
notificationss r get_notificationsz IContactSender.get_notificationsh s
,3355 J,1133> 2
r! ) __name__
__module__ __qualname__r
ProcessingOrder
ICONTACT_SENTPROCESSING_ORDERr AV_IM360SCOPEr' r3 rG r rL rS r r r r r r listrd r r rK
__classcell__)r. s @r r# r# - sQ "2@ NE
; ; ; ; ;<g C C C
X2&&& X#
_S r! r# )*rP loggingrB pathlibr defence360agent.internals.iaidr defence360agent.api.serverr !defence360agent.api.server.eventsr defence360agent.contracts.configr r "defence360agent.contracts.messagesr !defence360agent.contracts.pluginsr
r "defence360agent.internals.the_sinkr defence360agent.model.icontactr
$defence360agent.subsys.panels.cpanelr #defence360agent.subsys.panels.pleskr +defence360agent.subsys.panels.hosting_panelr defence360agent.utilsr r r r r defence360agent.utils.commonr getLoggerrp r r r# r1 r! r r s 9 9 9 9 9 9 / / / / / / 7 7 7 7 7 7 ; : : : : : 7 6 6 6 6 6 ; ; ; ; ; ; 7 7 7 7 7 7 5 5 5 5 5 5 D D D D D D - , , , , , 8 $ $ d d d d d[- d d d d dr!