Edit File: icontact_sender.cpython-311.pyc
� a�nhm � � � 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 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)�Core�IContactMessageType)� MessageType)� MessageSink� MessageSource)�TheSink)�IContactThrottle)�cPanel)� HostingPanel)� await_for� create_task_and_log_exceptions�recurring_check�retry_on�Scope)�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 %s�d ��seconds)�logger�warningr )�e�is �\/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/icontact_sender.py�async_log_on_errorr " s@ � � � � �N�N� 3� � � � � � �c������� c � � � e Zd Zej j Zej Z � fd�Z d� Z dd�d�Z de fd�Zd� 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)�self�args�kwargs� __class__s �r r& zIContactSender.__init__0 sG �� ������$�)�&�)�)�)��� � ��� � � @� @� �$�$�$r c � � K � d S �N� )r* �loops r � create_sinkzIContactSender.create_sink7 s � � � � �r N��userc � �| 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 )Nr3 )� message_type�paramsr4 )r6 � timestamp� template_args) r �may_be_notified�_panel�notifyr �GENERIC�refreshr � IContactSent�int�time�_sink�process_message)r* r6 r7 � period_limitr4 r9 � sent_messages r �_send_icontact_messagez%IContactSender._send_icontact_message: s � � � � � � �F� �/� � �� � � � � �F� "�k� 0� 0�,�4��� 1� � � � � � � � � � � ;� � $�\�� =� =� =� =�&�3� )� �d�i�k�k�*�*� +� � � �L� �*�,�,�\�:�:� :� :� :� :� :� :� :� :� :� ;� ;r �sinkc � � K � || _ t � � | _ | j j t j k r t || j � � g| _ d S d S r/ )rB r r; �NAMEr r �generic_notificationsr' )r* r1 rG s r � create_sourcez IContactSender.create_sourceX sT � � � ��� �"�n�n�� � �;� �v�{� *� *�.��$�4�� ��D�K�K�K� +� *r c � �~ K � | j D ]}|� � � �t j | j ddi�� d {V �� d S )N�return_exceptionsT)r' �cancel�asyncio�gather)r* �tasks r �shutdownzIContactSender.shutdownb sU � � � ��K� � �D� �K�K�M�M�M�M��n�d�k�B�T�B�B�B�B�B�B�B�B�B�B�Br � r � T)�on_error� max_tries�silent�log�returnc � � K � g }| j � � � r;| j � � � j t z t j � � k r5t j � � � d {V ��}| j � dd�� � |S )Ni� T)�mode�exist_ok) r) �exists�stat�st_mtimer rA r � notification�touch)r* � notificationss r �get_notificationsz IContactSender.get_notificationsg s� � � � � � ��,�3�3�5�5� J��,�1�1�3�3�<�s�B��i�k�k�� � #,�"8�":�":� :� :� :� :� :� :�M� � (� .� .�E�D� .� I� I� I� � r c � �: K � | � � � � d {V ��x}rzt � dt |� � � � |D ]Q}| � |d |d |d d�|d |� d� � �� � � d {V �� �Pd S d S ) Nz)Sending %s generic icontact notifications�type�notification_subject�notification_body_html)�subject� body_html�notification_period_limit�notification_user)r6 r7 rD r4 )rc r �info�lenrF �get)r* rb r` s r rJ z$IContactSender.generic_notifications� s� � � � �"&�"8�"8�":�":� :� :� :� :� :� :� :�=� � �K�K�;�S� �=O�=O� � � � !.� � � ��1�1�!-�f�!5�#/�0F�#G�%1�2J�%K� � � ".�.I�!J�%�)�)�*=�>�>� 2� � � � � � � � � � � � � � r ) �__name__� __module__� __qualname__r �ProcessingOrder� ICONTACT_SENT�PROCESSING_ORDERr �AV_IM360�SCOPEr& r2 rF r rK rR r r r r r r �listrc r r rJ � __classcell__)r- s @r r" r" , sQ �� � � � � �"�2�@�� �N�E� � � � � � � � � � ;� ;� ;� ;� ;�<�g� � � � �C� C� C� �X����2�&�&�&��� � � � � �X��#��� � � � � �� � � �� �� � � �_�S���� � ��� � � � r r" )(rO �loggingrA �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.hosting_panelr �defence360agent.utilsr r r r r � defence360agent.utils.commonr � getLoggerro r r r" r0 r r �<module>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� D� D� D� D� D� D�� � � � � � � � � � � � � � -� ,� ,� ,� ,� ,� �� �8� $� $��� � �d� d� d� d� d�[�-� d� d� d� d� dr