관리-도구
편집 파일: 008_subscription_permissions.cpython-311.pyc
� ��gF � �� � d Z ddlZddlmZmZ ddlmZ ddlmZ ddl m Z m Z m Z ej e� � ZdZ ed� � Z G d� d � � Zd efd �Zedd �� � Zedd�� � ZdS )u This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>. Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see <https://www.imunify360.com/legal/eula> � N)�Model�SqliteDatabase)�FULL)�skip_for_im360)�find_revisium_db�get_all_domains�get_permission_status�ra_cleanup_allowedc �0 � e Zd ZU eed< ed� � � ZdS )�db� FeatureManagementPermissionsc �* � |j d | _ d S )N� feature_management_permissions)�ormr )�cls�migrators �a/opt/imunify360/venv/lib/python3.11/site-packages/imav/migrations/008_subscription_permissions.py�initzdb.init, s � �+3�<� ,�, ��(�(�(� N)�__name__� __module__� __qualname__r �__annotations__� classmethodr � r r r r ) s; � � � � � � �"'�'�'�'�� � � �[� � � r r �returnc �� � i } t � � }n4# t $ r'}t � d|� � | cY d}~S d}~ww xY w|� � � D ]�}|� d� � x}�� t |d t � � }nS# t $ rF}t � dt |d � � t |� � � � Y d}~��d}~ww xY w|| |< ��| S )z1 {"user123": False, "user345": True, ..} z Failed to get Plesk domains: %rN�owner�idz8Failed to get permission status for domain: %s Error: %s�name) r � Exception�logger�warning�values�getr �CLEANUP_ALLOWED_PERM_NAME�str)�user_imav_cleanup_perm� all_domains�e� domain_infor �imav_cleanup_alloweds r �get_imav_permission_mapr- 3 s7 � � ��&�%�'�'� � �� � &� &� &����8�!�<�<�<�%�%�%�%�%�%�%�����&���� #�)�)�+�+� =� =� � �_�_�W�-�-� -�E� 6� � �#8��D�!�#<�$� $� � �� � � � � �N�N�J��K��'�(�(��A��� � � � �H�H�H�H����� ���� )=� �u�%�%� !�!s/ � � A� ?�A�A�6B� C"� <C � C"Fc � � |rd S t � � }|� t � d� � d S t � � }n@# t $ r3}t � dt |� � � � Y d }~d S d }~ww xY w|s t � d� � d S t � | � � |� � � D ]�\ }} t j � |�� � \ } } |r t | _ | � � � �L# t $ r3}t � d|t |� � � � Y d }~��d }~ww xY wd S )Nz/No legacy ImunifyAV database found. Skipping...z6Cannot obtain permissions map for migration, Error: %sz*No users to migrate AV cleanup permissions)�userz<Failed to update feature permissions for user: %s, Error: %s)r r" �infor- r! r# r' r r �itemsr � get_or_creater �av�save) r �database�fake�kwargs�revisium_db_path�imav_permission_mapr* �username�status� feature_perms�_s r �migrater> S s� � � � ���'�)�)��� �� � �E�F�F�F���� 5� 7� 7���� � � � ���� D� ��F�F� � � � ���������� ���� � �� � �@�A�A�A����G�G�H���� /� 5� 5� 7� 7� � ���&� � !� >� L� L� � M� � � �M�1� � %�#'� � � �"�"�$�$�$���� � � � �N�N�N���A��� � � � �H�H�H�H����� ����� s0 �A � A>� (A9�9A>�AD� E�#)E�Ec � � d S )Nr )r r5 r6 r7 s r �rollbackr@ } s � ��Dr )F)�__doc__�logging�peeweer r �,defence360agent.feature_management.constantsr �imav.migration_utils.otherr � imav.migration_utils.revisiumr r r � getLoggerr r" r& �rar �dictr- r> r@ r r r �<module>rJ s@ ��� �* ���� (� (� (� (� (� (� (� (� =� =� =� =� =� =� 5� 5� 5� 5� 5� 5�� � � � � � � � � � �� �8� $� $�� 0� ��^�D���� � � � � � � � � "�� "� "� "� "�@ �&� &� &� � �&�R � � � � � � � � r