관리-도구
편집 파일: lookup.cpython-311.opt-1.pyc
� IMgE � �� � d dl Z d dlmZmZmZ ddlmZ ddlmZ ddl m Z ddl m Z e � � Zd ed ee d ed ef d eded ef f d�Z dd ed ee deegef fd�ZdS )� N)�Any�Callable�List� )�MyImunifyConfig)�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 | � ��j � � }t ��|� � d S )Nz3Argument '%s' for '%s' must be specified explicitly)r r �ENABLED�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 ss �� � � ���!7�!7�&�E���� � � � "� � >� �F��z�z�(�J�$6�7�7���d�K��.�.�.�.�.� c �$ �� �di |�� �| i |��S �N� r ��argsr r r s ��r �wrapperz_wrapper.<locals>.wrapper/ s. �� �����&�����t�T�$�V�$�$�$r c � �4 �K � �di |�� �| i |��� d {V ��S r r r s ��r � async_wrapperz _wrapper.<locals>.async_wrapper4 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>.decoratorK 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; > s0 ���� � � � � � � � � �r )r )r% �typingr r r �contracts.configr �rpc_tools.lookupr �checkersr � exceptionsr �setr5 �strr* r; r r r �<module>rC s! �� ���� &� &� &� &� &� &� &� &� &� &� .� .� .� .� .� .� $� $� $� $� $� $� #� #� #� #� #� #� ,� ,� ,� ,� ,� ,� �3�5�5��/� �/� ��I�/�-5�c�3�h�-?�/�KN�/� �c�3�h��/� /� /� /�f 17�� � �� ��I�� �s�e�S�j��� � � � � r