관리-도구
편집 파일: utils.cpython-311.opt-1.pyc
� IMg� � � � d dl Z d dlmZ d dlmZmZ d dlmZ d dlm Z d dl m Z d dl m Z dd lmZ e j e� � Zd ed ed ed efd�Zd edee d edee fd�Zd ed efd�Zdee d efd�Zd� ZdS )� N)�chain)�List�Any)�hooks)�FeatureManagementPerms)�instance)�execute_iterable_expression� )�features�user�feature�value�returnc � � K � t j � � � 5 }t j | � � }|� ||� � t j |� � } || |� � }|r t � d||| � � n1t � d||| � � |� � � |cddd� � S # 1 swxY w Y dS )z�Sets a `feature` to `value` for a given `user`. Calls appropriate hook and returns its (bool) result. Logs the result of setting change. If hook fails rollbacks changes to database. z!Applied setting %s=%s for user %sz)Failed to apply setting %s=%s for user %sN) r �db�atomicr �get_perm� set_featurer �get_hook�logger�info�error�rollback)r r r �trx�perm�hook�oks �]/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/feature_management/utils.pyr r s% � � � � �� � � � � ��%�.�t�4�4�� ����%�(�(�(��~�g�&�&�� �T�$�� � �� � � �K�K�3�W�e�T� � � � � �L�L�;���� � � � �L�L�N�N�N��#� � � � � � � � � � � ���� � � � � � s �B C � C�C�users�existing_usersc � � K � g g d�}|D ]p}||vr t � d|� � �"t || |� � � d {V ��r |d � |� � �U|d � |� � �q|S )N)� succeeded�failedzNo such user: %sr" r# )r �warningr �append)r r r r �resultr s r � update_usersr' ( s� � � � � �� ,� ,�F�� *� *�� �~� %� %� �N�N�-�t� 4� 4� 4� � �T�7�E�2�2� 2� 2� 2� 2� 2� 2� *� �;� � &� &�t� ,� ,� ,� ,� �8� � #� #�D� )� )� )� )� �M� c � � K � t j � � }|� | |� � t j | � � } |d |� � S �N)r � get_defaultr r r )r r r r s r �update_defaultr, : sK � � � � !� -� /� /�D����W�e�$�$�$� �>�'� "� "�D� �4��e� � � r( c � � K � t | � � }t j t j � � }t t |� � � � � � }||z }|� t j � � |r;t � d|� � d� }t |t |� � � � ||z }|rt � d|� � |D ]U}t j |� � }t D ]7}|� |� � } t j |� � } | || � � �8�Vt# |� � pt# |� � S )z1Synchronize existing permissions with panel usersz Remove permissions of users %sc � � t j � � � t j � | � � � � S r* )r �delete�wherer �in_)�perms_to_removes r � expressionz sync_users.<locals>.expressionN s7 � �)�0�2�2�8�8�&�+�/�/��@�@�� � r( zAdd permissions to users %s)�setr �selectr r �tuples�remove�DEFAULTr r r �listr r � get_featurer r �bool) r � panel_users� perm_usersr2 r3 � perms_to_addr r r r �callbacks r � sync_usersr@ A s[ � � � ��e�*�*�K�'�.�/E�/J�K�K�J��U�J� -� -� /� /�0�1�1�J� �;�.�O����1�9�:�:�:�� G�� � �4�o�F�F�F� � � � $�J��_�0E�0E�F�F�F� ��+�L�� A�� � �1�<�@�@�@� � "� "��%�.�t�4�4�� � "� "�G��$�$�W�-�-�E� �~�g�.�.�H� �H�T�5� !� !� !� !� "� � � � � 6��o�!6�!6�6r( c � �J K � t t t j t j � � � t j t j k � � � � � � � � }|D ].} | j � � D ] \ }}t |||� � � d{V �� � �/dS )zlSets feature values for all existing users in feature management database to given values in `features`.N) r9 r r r5 r r0 r8 r6 �itemsr )r r r r r s r �reset_featuresrC d s� � � � � � � #� *�+A�+F� G� G� �U�&�+�/E�/M�M�� � �V�X�X� � � �E� � 4� 4�� ,�h�n� .� .� 4� 4�N�G�U� �d�G�U�3�3� 3� 3� 3� 3� 3� 3� 3� 3� 4�4� 4r( )�logging� itertoolsr �typingr r �"defence360agent.feature_managementr �(defence360agent.feature_management.modelr �defence360agent.modelr �defence360agent.utilsr �lookupr � getLogger�__name__r �strr; r r' r, r@ rC � r( r �<module>rP sr �� ���� � � � � � � � � � � � � � � 4� 4� 4� 4� 4� 4� K� K� K� K� K� K� *� *� *� *� *� *� =� =� =� =� =� =� � � � � � � �� �8� $� $���C� �#� �c� �d� � � � �4� �� �c���+.��@D�S� �� � � �$ �#� �c� � � � � 7�D��I� 7�$� 7� 7� 7� 7�F4� 4� 4� 4� 4r(