📁 File Manager Pro
v10.0.2 | PHP: 8.1.34
Server: LiteSpeed
2026-06-28 21:10:59
📂
/
/
opt
/
imunify360
/
venv
/
lib
/
python3.11
/
site-packages
/
defence360agent
/
plugins
/
__pycache__
✏️
Editing: accumulate.cpython-311.opt-1.pyc
� �j � � � d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z m Z d dl m Z m Z mZ d dlmZmZ ee� � Z G d� de e � � ZdS ) � N)� getLogger)� inactivity)� Accumulatable� MessageType� Splittable)� MessageSink� MessageSource�expect)�recurring_check�safe_cancel_taskc �& � � e Zd Zej j ZdZ ee j � dd� � � � Z ee j � dd� � � � Z e e f� fd� Zd� Zd� Zd � Zd � Z eej � � d efd �� � Zd � Z� xZS )� Accumulate�� �%IMUNIFY360_AGGREGATE_MESSAGES_TIMEOUT�< �*IMUNIFY360_AGGREGATE_SHUTDOWN_SEND_TIMEOUT�2 c � �� t � � j di |�� || _ || _ t j t � � | _ d S )N� )�super�__init__�_period�_shutdown_timeout� collections� defaultdict�list�_data)�self�period�shutdown_timeout�kwargs� __class__s ��W/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/accumulate.pyr zAccumulate.__init__" sG �� � �����"�"�6�"�"�"� �� �!1�� � �,�T�2�2�� � � � c � �� K � || _ || _ | j dk rd n<|� t | j � � | j � � � � � � | _ d S )Nr )�_loop�_sinkr � create_taskr �_flush�_task)r �loop�sinks r# � create_sourcezAccumulate.create_source- sf � � � ��� ��� � �|�q� � � �D��!�!�"L�"?�/�$�,�"?�"?�� �"L�"L�"N�"N�O�O� � � � r$ c � � K � || _ d S )N)r& )r r+ s r# � create_sinkzAccumulate.create_sink6 s � � � ��� � � r$ c � �$ K � t j | � � � | j � � � d {V �� d S # t j $ rH t � d| j � � | j � t | j � � � d {V �� Y d S Y d S w xY w)Nz5Timeout (%ss) sending messages to server on shutdown.) �asyncio�wait_for�stopr � TimeoutError�logger�errorr* r �r s r# �shutdownzAccumulate.shutdown9 s� � � � � 3��"�4�9�9�;�;��0F�G�G� G� G� G� G� G� G� G� G� G���#� 3� 3� 3� �L�L�G��&� � � � �z�%�&�t�z�2�2�2�2�2�2�2�2�2�2�2�2� &�%�%� 3���s �28 �AB�Bc � �� K � t � d� � | j �t | j � � � d {V �� t � d� � | � � � � d {V �� d S )Nz Accumulate.stop cancel _taskzAccumulate.stop wait lock)r5 �infor* r r) r7 s r# r3 zAccumulate.stopE s| � � � �� � �2�3�3�3� �:� !�"�4�:�.�.� .� .� .� .� .� .� .�� � �/�0�0�0��k�k�m�m���������r$ �messagec � �D K � t |j t � � r|j n|j f}|� � � r`t j � d� � 5 |D ]"}| j | � |� � �# d d d � � d S # 1 swxY w Y d S d S )N� accumulate) � isinstance� LIST_CLASS�tuple� do_accumulater �track�taskr �append)r r; � list_types� list_types r# �collectzAccumulate.collectM s � � � � �'�,�e�4�4� '�G� � ��$�&� � � � � "� "� :��!�&�&�|�4�4� :� :�!+� :� :�I��J�y�)�0�0��9�9�9�9�:� :� :� :� :� :� :� :� :� :� :� :� :���� :� :� :� :� :� :� :� :s � &B�B�Bc � � K � | j }t j t � � | _ |� � � D ]�\ }}t |t � � r|� |� � n|f}|D ]�}t � d|j � dt |� � � d�� � | j � ||�� � � � � d {V �� �a# t $ r t � d||� � � w xY w��d S )NzPrepare z(<items=z>) for further processing)�itemsz%s, %s)r r r r rI � issubclassr �batchedr5 r: �__name__�lenr' �process_message� TypeErrorr6 )r � copy_datarF �messagesrK �batchs r# r) zAccumulate._flushY sB � � � ��J� � �,�T�2�2�� �#,�?�?�#4�#4� � � �I�x� �i��4�4�!� �!�!�(�+�+�+� �[� � !� � ��� � �-�y� 1� -� -�3�u�:�:� -� -� -�� � �� �*�4�4�Y�Y�U�5K�5K�5K�L�L�L�L�L�L�L�L�L�L�� � � � ��L�L��9�e�<�<�<������ � � s �&*C�(C9)rL � __module__� __qualname__r �ProcessingOrder�POST_PROCESS_MESSAGE�PROCESSING_ORDER�SHUTDOWN_PRIORITY�int�os�environ�get�DEFAULT_AGGREGATE_TIMEOUT�SHUTDOWN_SEND_TIMEOUTr r- r/ r8 r3 r r r rG r) � __classcell__)r" s @r# r r s) �� � � � � �"�2�G�� � � !$�� � ���>��C�C�!� !� � �C� � ���C�R�H�H� � �� )�.� 3� 3� 3� 3� 3� 3� � � �� � � 3� 3� 3� � � � �V�K� %�&�&� :�]� :� :� :� '�&� :�� � � � � � r$ r )r1 r rZ �loggingr �defence360agent.apir �"defence360agent.contracts.messagesr r r �!defence360agent.contracts.pluginsr r r �defence360agent.utilsr r rL r5 r r r$ r# �<module>re s �� ���� � � � � � � � � � � � � � � *� *� *� *� *� *�� � � � � � � � � � � � � � � � � � � � D� C� C� C� C� C� C� C� ��8� � ��Z� Z� Z� Z� Z��m� Z� Z� Z� Z� Zr$
💾 Save Changes
❌ Cancel