Edit File: whm.cpython-311.pyc
� a�nhG � �� � d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m Z ej e � � Z dZg d�Z G d� de � � Z G d � d e� � Zdd �Zd � Zd� Zd� Zd� ZdS )� N)�quote)� check_run� CheckRunError)�PanelExceptionz/usr/sbin/whmapi1)zno certificatezno key with the idzcannot read license filezinvalid license filezlicense file expiredc � � e Zd ZdZdS )�WHMAPIExceptionz5Got broken output or other problem during WHMAPI callN��__name__� __module__� __qualname__�__doc__� � �]/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/subsys/panels/cpanel/whm.pyr r s � � � � � �?�?��Dr r c � � e Zd ZdZdS )�WHMAPILicenseErrorz$Raises when cannot Read License FileNr r r r r r s � � � � � �.�.��Dr r Fc � � K � |rdgng }|� t d| g� � d� |� � � D � � } t ||z � � � d {V ��� � � }nS# t $ rF}|j t j k r)t � |� � t |� � �|�d }~ww xY w t j |� � }n-# t j $ r}t d|�d|� �� � |�d }~ww xY w |d d r|d S t d � |d d |d d � � � � �# t $ r_}d |� � � v r!t � d � � t" �t d� ||� � � � �d }~ww xY w)N�sudo� --output=jsonc �Z � g | ](\ }}d � |t |� � � � ��)S )z{}={})�formatr )�.0�k�vs r � <listcomp>zwhmapi1.<locals>.<listcomp>' s0 � � E� E� E�d�a��g�n�n�Q��a���)�)� E� E� Er z Broken output from whmapi1: z , reason: �metadata�result�dataz whmapi {} command failed: {}�command�reason)� statusmsgzCannot Read License Filez Cannot Read CPanel License Filez/Broken output from whmapi1 (KeyError: {}): {!r})�extend� WHMAPI1_CMD�itemsr �decoder � returncode�signal�SIGTERM�logger�warningr �json�loads�JSONDecodeErrorr �KeyErrorr )�functionr �kwargs�cmd�params� raw_output�e�outputs r �whmapi1r6 $ s � � � �� "�6�(�(��C��J�J� �_�h�7�8�8�8� E� E�f�l�l�n�n� E� E� E�F�� %�c�F�l� 3� 3�3�3�3�3�3�3�;�;�=�=� � �� � � � � �<�F�N�?� *� *� �N�N�1� � � �!�!�$�$� $��G����� ���� ���J�'�'���� � � � � � � F�:� F� F�1� F� F� � �� ��������� � �*� �h� '� ��&�>� !�!�.�5�5��:�&�y�1�6�*�3E�h�3O�� �� � �� � � � � 4�� � ��� F� F� �N�N�<� =� =� =�$� $�!�A�H�H��v�� �� � ����� ���sP �*A1 �1 C�;AB<�<C�C �D�)C?�?D�E � ;E � G�#AF=�=Gc � � t g| �d�} t � d|� � t j |dt j �� � }n*# t j $ r}t d|z � � |�d }~ww xY w|r�t j |j � � � � � }g }t |� � D ]l\ }} |} |D ] } | | } � |� | � � �+# t $ r5}|dk rt d� � |�|� d � � Y d }~�ed }~ww xY wnd S t |� � dk r|d S |S ) Nr zsubprocess.run(%r)T)�check�stdoutzFailed to run whmapi1: %sr z Could not parse whmapi1 output� )r# r) �debug� subprocess�run�PIPE�CalledProcessErrorr r+ r, r9 r% � enumerate�appendr. �len) �args� path_listr1 �resr4 �decoded_outputr �i� element_path�item�keys r � run_whmapirK L s� � � � /�� /�� /�C�F�� � �)�3�/�/�/��n�S��Z�_�E�E�E���� � (� F� F� F� � 9�A� =�>�>�A�E�����F���� � � ��C�J�$5�$5�$7�$7�8�8���� (�� 3� 3� (� (�O�A�|� (�%��'� %� %�C� ��9�D�D�� � �d�#�#�#�#��� (� (� (���6�6� *�8�� � � � �M�M�$�'�'�'�'�'�'�'�'����� (���� (�" �� �6�{�{�a����a�y��� s/ �<A � A2�A-�-A2�9$C � D �(+D�D c �& � t | ddg� � S )Nr r �rK )rC s r �run_whmapi_resultrN s s � � �d�Z�� 2� 3� 3�3r c �: � t | ddgddg� � \ }}||fS )Nr r r rM )rC r r s r � run_whmapi_result_and_reasonrP w s2 � � � �z�8�$�z�8�&<�� �N�F�F� �6�>�r c � � � � fd�}|S )Nc � �� �K � d } �| i |��� d {V ��}nZ# t $ r1}t � t |� � � � Y d }~n$d }~w t � d� � Y nxY w|S )NzSomething went wrong)r r) �error�str� exception)rC r0 �rv�sww�funcs �r �wrapperz catch_exception.<locals>.wrapper� s� �� � � � �� 5��t�T�,�V�,�,�,�,�,�,�,�,�B�B�� � #� #� #� �L�L��S��� "� "� "� "� "� "� "� "����� 5� � � � 3� 4� 4� 4� 4� 4����� s � � A-�'A � A-r )rX rY s ` r �catch_exceptionrZ s# �� � � � � � � �Nr )F)r+ �loggingr< r' � urllib.parser �defence360agent.utilsr r �"defence360agent.subsys.panels.baser � getLoggerr r) r# �WHMAPI_CERT_ERROR_LISTr r r6 rK rN rP rZ r r r �<module>ra sC �� � � � � ���� � � � � � � � � � � � � � � :� :� :� :� :� :� :� :� =� =� =� =� =� =� �� �8� $� $�� "� �� � � � � � � � �n� � � � � � � � � � � � � %� %� %� %�P$� $� $�N4� 4� 4�� � �� � � � r