Edit File: plugins.cpython-311.opt-1.pyc
� �gh6 � �x � d dl Z d dlZd dlmZ d dlmZmZ ej e� � Z G d� dej � � Z dS )� N)�utils)�messages�pluginsc � � � e Zd ZdZdZ ej ej j � � d� � � Z � fd�Z � fd�Z d� Z d� Zd� Zd � Zd � Z� xZS ) �IDSAwareMessageSinkz^A message sink that accepts messages taking into account the current IDS (strategy). Nc � � K � | j �|j | j k r�| � � � s� | � � � � d{V �� t � d| � � dS # t j $ r � t $ r2 t � d| � � t j � � Y dS w xY wdS | � � � r� | � � � � d{V �� t � d| � � dS # t t j f$ r2 t � d| � � t j � � Y dS w xY wdS )z7De/activate the plugin on strategy change if necessary.NzPlugin %r is activatedz Failed to activate %r plugin�Plugin %r is deactivated� Failed to deactivate %r plugin) �STRATEGY�strategy� is_active�activate�logger�info�asyncio�CancelledError� Exception� exceptionr �fail_agent_service� deactivate)�self�messages �L/opt/imunify360/venv/lib/python3.11/site-packages/im360/contracts/plugins.py�on_strategy_changez&IDSAwareMessageSink.on_strategy_change s� � � � � �=� �G�$4�� �$E�$E��>�>�#�#� @�@� �-�-�/�/�)�)�)�)�)�)�)� �K�K� 8�$�?�?�?�?�?�� �-� � � �� � /� /� /��$�$�%C�T�J�J�J��,�.�.�.�.�.�.�/���� @� @� �^�^� � � >� >��o�o�'�'�'�'�'�'�'�'�'� � � � 6��=�=�=�=�=�� �w� 5�6� +� +� +�� � �!A�4�H�H�H��(�*�*�*�*�*�*� +���� >� >s% �A& �&AB1�0B1� D �AE � E c � � �K � | � � � s t |t j j � � r't � � � |� � � d {V ��S d S )N)r � isinstancer � MessageType�StrategyChange�super�process_message)r r � __class__s �r r z#IDSAwareMessageSink.process_message- sk �� � � � �>�>� � � :�z� �X�)�8� � � :� ���0�0��9�9�9�9�9�9�9�9� 9� :� :� c � �X �K � | � � � rn | � � � � d {V �� t � d| � � n7# t t j f$ r t � d| � � Y nw xY wt � � � � � � d {V ��S )Nr r ) r r r r r r r r r �shutdown)r r! s �r r$ z IDSAwareMessageSink.shutdown3 s� �� � � � �>�>� � � >� >��o�o�'�'�'�'�'�'�'�'�'� � � � 6��=�=�=�=�� �w� 5�6� I� I� I�� � �!A�4�H�H�H�H�H� I���� �W�W�%�%�'�'�'�'�'�'�'�'�'s �A �1B�Bc � �2 K � | � � � dS )zrThis method will be called for plugins with specified strategy when this strategy becomes active. N)�_mark_as_active�r s r r z IDSAwareMessageSink.activate> s � � � � � � � � � � � r" c � �2 K � | � � � dS )ztThis method will be called for plugins with specified strategy when this strategy becomes inactive. N)�_mark_as_inactiver' s r r z IDSAwareMessageSink.deactivateD s � � � � � � � � � � � r" c �$ � t | dd� � S )z Indicates if plugin is active.� _is_activeF)�getattrr' s r r z IDSAwareMessageSink.is_activeJ s � ��t�\�5�1�1�1r" c � � d| _ d S )NT�r+ r' s r r& z#IDSAwareMessageSink._mark_as_activeN s � � ����r" c � � d| _ d S )NFr. r' s r r) z%IDSAwareMessageSink._mark_as_inactiveQ s � � ����r" )�__name__� __module__� __qualname__�__doc__r r �expectr r r r r r$ r r r r& r) � __classcell__)r! s @r r r s� �� � � � � �� � �H��W�^�H�(�7�8�8�>� >� 9�8�>�4:� :� :� :� :� (� (� (� (� (� � � � !� !� !� 2� 2� 2� � � � � � � � � � r" r ) r �logging�defence360agentr �defence360agent.contractsr r � getLoggerr0 r � MessageSinkr � r" r �<module>r< s� �� ���� ���� !� !� !� !� !� !� 7� 7� 7� 7� 7� 7� 7� 7� �� �8� $� $��H � H � H � H � H �'�-� H � H � H � H � H r"