관리-도구
편집 파일: model.cpython-311.pyc
� IMgF � �b � d dl mZmZmZ d dlmZmZmZmZm Z d dl m Z m Z G d� de � � Z dS )� )� CharField�Check� TextField)�AV� AV_REPORT�FULL�NA� PROACTIVE)�Model�instancec �\ � e Zd ZdZdZ G d� d� � Z ed�� � Z ed e d� e e � � � � ge � � � Z ed e d � e ee � � � � ge� � � Zed ed d fd �� � Zed� � � Zd� Zded efd�Zdedefd�Zd� ZdS )�FeatureManagementPermszrPermissions state for Feature Management. Each record/instance is a set of permissions of a single user. � c � � e Zd Zej ZdZdS )�FeatureManagementPerms.Meta� feature_management_permissionsN)�__name__� __module__� __qualname__r �db�database�db_table� � �]/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/feature_management/model.py�Metar s � � � � � ��;��3���r r T)�uniqueFzproactive in ('{}','{}'))�null� constraints�defaultzav in ('{}','{}','{}')�user�returnc � � | � � � }|�|S t |j t |j i}| � ||�� � \ }}|S )z� Get feature permissions by user name :param user: user name :return: :class:`FeatureManagementPerms` object for user N)r! �defaults)� get_defaultr �avr � proactive� get_or_create)�clsr! r r$ �perm�_s r �get_permz FeatureManagementPerms.get_perm- sX � � �/�/�#�#�� �<��N� �� � �w�(� �� �#�#���#�A�A���a�� r c �8 � | � | j �� � S )zGet default permissions)r! )�get�DEFAULT)r) s r r% z"FeatureManagementPerms.get_defaultB s � � �w�w�C�K�w�(�(�(r c �" � | j | j k S )z&Check if current permission is default)r! r/ ��selfs r � is_defaultz!FeatureManagementPerms.is_defaultG s � ��y�D�L�(�(r �keyc �" � t | |� � S )z Get permission by feature name)�getattr)r2 r4 s r � get_featurez"FeatureManagementPerms.get_featureK s � ��t�S�!�!�!r �valuec �P � t | ||� � | � � � dS )zSet permissionN)�setattr�save)r2 r4 r8 s r � set_featurez"FeatureManagementPerms.set_featureO s% � ���c�5�!�!�!� � � � � � � � r c �6 � t | j t | j iS )N)r r& r r' r1 s r �as_dictz FeatureManagementPerms.as_dictT s � � ��� �t�~� � r N)r r r �__doc__r/ r r r! r r �formatr r r' r r& � classmethod�strr, r% r3 r7 r< r> r r r r r s� � � � � � �� � �G�4� 4� 4� 4� 4� 4� 4� 4� �9�D� !� !� !�D� � � ��U�5�<�<�R��F�F�G�G�H��� � �I� �� � �E�*�1�1�"�i��F�F� G� G� � � � � �B� ��C� �$<� � � � �[��( �)� )� �[�)�)� )� )�"�s� "�s� "� "� "� "��s� �3� � � � � � � � � r r N)�peeweer r r �,defence360agent.feature_management.constantsr r r r r �defence360agent.modelr r r r r r �<module>rF s� �� .� .� .� .� .� .� .� .� .� .�� � � � � � � � � � � � � � 2� 1� 1� 1� 1� 1� 1� 1�K � K � K � K � K �U� K � K � K � K � K r