Edit File: lookup.cpython-311.opt-1.pyc
� a�nh � �� � d dl Z d dlmZmZmZ ddlmZ ddlmZ ddl m Z ddl m Z dd l mZ e� � Zd ed ee d ed ef deded ef f d�Z dd ed ee deegef fd�ZdS )� N)�Any�Callable�List� )�MyImunifyConfig)� is_plesk_service_plan_enabled)�wraps� )� check_feature)�UserArgumentNotFound�name� permissions�func.�user_key�returnc �R � ������ � t j �� � }�|j vrt d��� � �|j � � � j � j u ��� ���� fd��t �� � ��fd�� � }t �� � ��fd�� � }t j �� � r|S |S )z� Wrapper to enable feature management for func :param name: feature name :param func: function/method to wrap :param user_key: parameter name which contains user name :return: new callable object z Expecting argument '%s' for %sc �� �� �r�| vrt d��� � �t j r d S t � � r d S | � ��j � � }t ��|� � d S )Nz3Argument '%s' for '%s' must be specified explicitly)r r �ENABLEDr �get�defaultr )�kwargs�userr r r r �user_key_required� user_params �������^/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/feature_management/lookup.py�checkerz_wrapper.<locals>.checker! s� �� � � ���!7�!7�&�E���� � � � "� � >� �F� (� *� *� � G� �F��z�z�(�J�$6�7�7���d�K��.�.�.�.�.� c �$ �� �di |�� �| i |��S �N� r ��argsr r r s ��r �wrapperz_wrapper.<locals>.wrapper4 s. �� �����&�����t�T�$�V�$�$�$r c � �4 �K � �di |�� �| i |��� d {V ��S r r r! s ��r � async_wrapperz _wrapper.<locals>.async_wrapper9 sD �� � � �����&�����T�4�*�6�*�*�*�*�*�*�*�*�*r )�inspect� signature� parametersr r �emptyr �iscoroutinefunction) r r r r r' r# r% r r r s ```` @@@r �_wrapperr+ s �������� � �!�$�'�'�I��y�+�+�+�"� ,�h�� � � � �%�h�/�J�"�*�j�.>�>��/� /� /� /� /� /� /� /� /� /�& �4�[�[�%� %� %� %� %� �[�%� �4�[�[�+� +� +� +� +� �[�+� �"�4�(�(� � � � �Nr r c � � ��� � ��fd�}|S )a Get decorator to manage function/method with feature management :param name: feature name :param user_key: parameter name which contains user name :param permissions: list of permission values, with which user can access specifig endpoint :return: decorator c � �� t j | � � rvt | di � � � � � D ]Q\ }}|� d� � s7t j |� � r#t ��|�� � }t | ||� � �Rn&t j | � � rt ��| �� � } t � �� � | S )N�__dict__�_) r& �isclass�getattr�items� startswith� isfunctionr+ �setattr�features�add)�obj�m_name�m_objr# r r r s ���r � decoratorzfeature.<locals>.decoratorP s� �� � �?�3� � � =�!(��j�"�!=�!=�!C�!C�!E�!E� 2� 2� ��� �(�(��-�-� 2�'�2D�U�2K�2K� 2� &�t�[�%�� J� J�G��C���1�1�1�� 2� � �� $� $� =��4��c�8�<�<�C�� � �T����� r r )r r r r; s ``` r �featurer< C s0 ���� � � � � � � � � �r )r )r& �typingr r r �contracts.configr �contracts.permissionsr �rpc_tools.lookupr �checkersr � exceptionsr �setr6 �strr+ r< r r r �<module>rE s3 �� ���� &� &� &� &� &� &� &� &� &� &� .� .� .� .� .� .� A� A� A� A� A� A� $� $� $� $� $� $� #� #� #� #� #� #� ,� ,� ,� ,� ,� ,� �3�5�5��3� �3� ��I�3�-5�c�3�h�-?�3�KN�3� �c�3�h��3� 3� 3� 3�n 17�� � �� ��I�� �s�e�S�j��� � � � � r