관리-도구
편집 파일: myimunify_id.cpython-311.pyc
� IMg� � �. � d dl Z d dlZd dlmZ d dlmZmZmZ d dlm Z d dl m Z d dl m Z mZ d dlmZ d dlmZ d Z G d � d e� � Zd ed edee fd�Zdee fd�Zdeeef fd�Zd edefd�Zdededefd�Z dedefd�Z d edefd�Z dS )� N)�Path)�Dict�List�Optional)�logger)�instance)� MyImunify�update_users_protection)� HostingPanel)� safe_fileopsz .myimunify_idc � � e Zd ZdZdS )�MyImunifyIdErrorz5Exception representing issues related to MyImunify idN)�__name__� __module__� __qualname__�__doc__� � �[/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/contracts/myimunify_id.pyr r s � � � � � �?�?�?�?r r �user� protection�returnc � � K � t j |�� � \ }}|� � � t | |g|� � � d{V �� t j d||� � t |� � � d{V ��}n# t $ r Y dS w xY w|S )z5Save subscription type to the DB and generate id file�r Nz(Applied setting MyImunify=%s for user %s)r � get_or_create�saver r �info�_get_or_generate_idr )�sinkr r � myimunify�_� myimunify_ids r �add_myimunify_userr# s� � � � � �*��5�5�5�L�I�q� �N�N���� !�$��� � ;� ;�;�;�;�;�;�;�;� �K�:�J��M�M�M�� 0�� 6� 6�6�6�6�6�6�6� � �� � � � ��t�t����� �s � A4 �4 B�Bc � �N K � g } t � � � � � � d{V ��}t � � � d{V ��}t j � � � 5 t |� � � � � D ]�\ }}t j |�� � \ }}| � |� |i � � � dd� � |||j |� |i � � � dd� � d�� � �� ddd� � n # 1 swxY w Y | S )zP Get a list of MyImunify users, their subscription types and unique ids Nr �email� �locale)r% �usernamer" r r' ) r �get_user_details�_myimunify_user_to_idr �db� transaction�sorted�itemsr r �append�getr )�users� user_details�myimunify_user_to_idr � myimunify_uid�recordr! s r �get_myimunify_usersr6 & s~ � � � � �E�%� � �8�8�:�:�:�:�:�:�:�:�L�!6�!8�!8�8�8�8�8�8�8�� �� � � "� "� � �#)�*>�*D�*D�*F�*F�#G�#G� � � �D�-�!�/�T�:�:�:�I�F�A� �L�L� )� -� -�d�B� 7� 7� ;� ;�G�R� H� H� $�$1�"(�"3� *� .� .�t�R� 8� 8� <� <�X�r� J� J� � � � � � � � � � � � � � � � � � ���� � � � � �Ls � B0D�D �!D c � �& K � i } t � � � � � � d{V ��D ]e} t |� � � d{V ��| |< � # t $ r Y �(t j $ r-}t j d|t |� � � � Y d}~�^d}~ww xY w| S )z+Get a list of users and their MyImunify idsNz+Unable to generate id for user=%s, error=%s) r � get_usersr r r �UnsafeFileOperationr �error�str)� user_to_idr �es r r* r* = s� � � � � �J�"�n�n�.�.�0�0�0�0�0�0�0�0� � �� �%8��%>�%>� >� >� >� >� >� >�J�t� � �� � � � � �H��/� � � � �L�=�t�S��V�V� � � � �H�H�H�H����� ���� �s �A� B�B�!#B � Bc � �� K � t | � � � d{V ��} t |� � S # t t f$ r1 t j � � j }t ||� � � d{V ��cY S w xY w)zS Read MyImunify id if exists or generate a new one and write into the file N)�_get_myimunify_id_file�_read_id�FileNotFoundErrorr �uuid�uuid1�hex� _write_id)r �id_filer" s r r r O s� � � � � +�4�0�0�0�0�0�0�0�0�G�6��� � � �� � /� 0� 6� 6� 6��z�|�|�'� � �|�W�5�5�5�5�5�5�5�5�5�5�5�6���s �( �?A*�)A*r" rF c � �� K � d| � d�} t j t |� � |� � � d{V �� n6# t t f$ r"}t j d|� � t |�d}~ww xY w| S )zWrite MyImunify id to filezE# DO NOT EDIT # This file contains MyImunify id unique to this user � Nz1Unable to write myimunify_id in user home dir: %s)r � write_textr; �OSError�PermissionErrorr r: r )r" rF �textr= s r rE rE [ s� � � � � � � � � � � &��%�c�'�l�l�D�9�9�9�9�9�9�9�9�9�9�� �_� %� &� &� &�� �H�!�L�L�L� �A�%�����&���� �s �(3 �A&� A!�!A&c �. � | � d� � 5 }t |� � � � � D ]?}|r;|� d� � s&|� � � x}r|c cddd� � S �@ ddd� � n # 1 swxY w Y t �)zRead MyImunify id from file�r�#N)�open�reversed� readlines� startswith�stripr )rF �f�liner" s r r@ r@ k s� � � � � �c� � � (�a� �Q�[�[�]�]�+�+� (� (�D�� (�D�O�O�C� 0� 0� (�#'�:�:�<�<�/�<� (�'�'�'� (� (� (� (� (� (� (� (�� (�(� (� (� (� (� (� (� (� (� (� (���� (� (� (� (� �s �AB�7B�B � B c � � K � t j | � � }t |j � � t z }|� � � s�|j � � � s t j d| � � t � t j t |� � � � � d{V �� nf# t t f$ r"}t j d|� � t |�d}~ww xY wn/# t $ r"}t j d| � � t |�d}~ww xY w|S )z<Get a file with MyImunify id and create it if does not existzNo such user homedir: %sNz/Unable to put myimunify_id in user home dir: %szNo such user: %s)�pwd�getpwnamr �pw_dir�MYIMUNIFY_ID_FILE_NAME�exists�parentr r: r r �touchr; rK rJ �KeyError)r �user_pwdrF r= s r r? r? w s2 � � � �.��<��%�%�� �x��'�'�*@�@���~�~� � � .��>�(�(�*�*� '�� � 7��>�>�>�&�&� .�"�(��W� � �6�6�6�6�6�6�6�6�6�6��#�W�-� .� .� .�� �E�q�� � � '�A�-����� .���� .�� � &� &� &�� �'��.�.�.� �A�%�����&���� �Ns/ �C �>'B& �&C�7 C�C� D �' D�D ) rX rB �pathlibr �typingr r r �%defence360agent.contracts.permissionsr �defence360agent.modelr � defence360agent.myimunify.modelr r �+defence360agent.subsys.panels.hosting_panelr �defence360agent.utilsr r[ � Exceptionr r; �boolr# r6 r* r rE r@ r? r r r �<module>rj s� �� � � � � � � � � � � � � � � '� '� '� '� '� '� '� '� '� '� 8� 8� 8� 8� 8� 8� *� *� *� *� *� *� N� N� N� N� N� N� N� N� D� D� D� D� D� D� .� .� .� .� .� .�(� �@� @� @� @� @�y� @� @� @����!%�� �c�]�� � � �&�4��:� � � � �.�T�#�s�(�^� � � � �$ 6�C� 6�C� 6� 6� 6� 6� �#� �� �� � � � � �d� �s� � � � ��s� �t� � � � � � r