Edit File: middleware.cpython-311.pyc
� �ghD � �6 � d dl mZ d dlmZ d dlmZ d� Zd� ZdS )� ��wraps)�Strategy)�IPListc �< � � t � � � � fd�� � }|S )Nc � � �K � �| i |��� d {V ��}t |t � � s J d|z � � �t j � � |d< |S )Nz Result should be a dictionary %s�strategy)� isinstance�dictr �get)�args�kwargs�result�fs ��P/opt/imunify360/venv/lib/python3.11/site-packages/im360/simple_rpc/middleware.py�wrapperz add_strategy.<locals>.wrapper sp �� � � ��q�$�)�&�)�)�)�)�)�)�)�)���&�$�'�'� � � .�� 7� � �'� &�\�^�^��z��� � r �r r s ` r � add_strategyr s3 �� � �1�X�X�� � � � �X�� �Nr c �< � � t � � � � fd�� � }|S )z� Replaces 'GRAY_SPLASHSCREEN' value for *listname* field for queries like `graylist ip list` for compatibility with UI c � � �K � �| i |��� d {V ��}t |� d� � t � � s|S t j � � � }|� dg � � D ]�}t |t � � rz|� d� � re|d � � � |k rG|d � � � r t j n t j � � � |d< ��|S )N�items�listname) r r �listr �GRAY_SPLASHSCREEN�lowerr �isupper�GRAY)r r r �gray_splashscreen�itemr s �r r z4replace_gray_splashscreen_with_gray.<locals>.wrapper s �� � � ��q�$�)�&�)�)�)�)�)�)�)�)���&�*�*�W�-�-�t�4�4� ��M� "� 4� :� :� <� <���J�J�w��+�+� � �D��4��&�&� ��H�H�Z�(�(� � ��$�*�*�,�,�0A�A�A� �J�'�/�/�1�1�-�F�K�K� ��*�*�,�,� �Z� �� � r r r s ` r �#replace_gray_splashscreen_with_grayr! s5 �� � �1�X�X�� � � � �X��& �Nr N)� functoolsr �im360.internals.strategyr �im360.model.firewallr r r! � r r �<module>r& sc �� � � � � � � -� -� -� -� -� -� '� '� '� '� '� '� � � � � � � � r