Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/lib/__pycache__/ustate.cpython-311.pyc
�

L(e�"��z�ddlmZddlmZddlmZmZmZddlZej	d��Z
	ddlZddlmZ
nC#e$r;ZdZe
�dee��z��Gd�d	e��Z
YdZ[ndZ[wwxYw	dd
lmZmZmZddlmZddlmZn#e$r
ed
kr	YnwxYwd�Zd�Zd�ZGd�de��Z dS)�)�absolute_import)�zip)�List�Dict�OptionalN�ustate)�OperationalErrorzCan't import MySQLdb; %sc��eZdZdS)�MySQLOperationalErrorN)�__name__�
__module__�__qualname__���~/builddir/build/BUILDROOT/lve-stats-4.2.1-1.el8.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/lib/ustate.pyrrs�������rr)�	db_access�dblogin_cplogin_pairs�cpusers)�	NoPackage)�run_command�__main__c
���|�d��}|�d��}|���}ddd���fd�|D��}i}d}t|��D�]h\}}|s�	|�dt|��dz
��}t|��t|��ko|d���}	|	s<|�dt|��d	z
��}|�dd
��t|��t|��kr@t�d|dd
|td|dz
��|dz�ii����nctt||����}
|	rqt|
�d����}	|
d=n#t$rYnwxYw|
||<t��||d<||�dd�����|
�dd��|
�dd��|
�dd��|
�dd��	|
||dt|
�d����<��##ttf$r3}t�dt|����Yd}~��bd}~wwxYw|S)a)
    pars /usr/sbin/lveps -c 1 -p -d -n result
    example returned data:
    {504:
        {'CPU': '26%', 'IO': '0', 'MEM': '1', 'EP': '0', 'IOPS': 'N/A', 'PNO': '3', 'TNO': '3', 'TID':
            {4400:
                {'CPU': '26%', 'IO': 'N/A', 'MEM': '1', 'CMD': 'md5sum', 'IOPS': 'N/A'},
            4381:
                {'CPU': '0%', 'IO': 'N/A', 'MEM': '1', 'CMD': 'su', 'IOPS': 'N/A'},
            4382:
                {'CPU': '0%', 'IO': 'N/A', 'MEM': '1', 'CMD': 'bash', 'IOPS': 'N/A'}}},
    500:
        {'CPU': '13%', 'IO': '0', 'MEM': '1', 'EP': '0', 'IOPS': 'N/A', 'PNO': '3', 'TNO': '3', 'TID':
            {4266:
                {'CPU': '0%', 'IO': 'N/A', 'MEM': '1', 'CMD': 'su', 'IOPS': 'N/A'},
            4299:
                {'CPU': '13%', 'IO': 'N/A', 'MEM': '1', 'CMD': 'cat', 'IOPS': 'N/A'},
            4267:
                {'CPU': '0%', 'IO': 'N/A', 'MEM': '1', 'CMD': 'bash', 'IOPS': 'N/A'}}}}

    example of data manipulation:
    getting a list of user id
    >>> lveps_data = _pars_lveps()
    >>> user_id_list = lveps_data.keys()
    a list of processes tid particular user id 504
    >>> user_tid_list = lveps_data[504]['TID'].keys()
    getting CPU load user
    >>> user_cpu = lveps_data[504]['CPU']
    getting CPU load specific process
    >>> lveps_data[504]['TID'][4400]
    �
r�CPU�CMD)�SPEED�COMc�<��g|]}��||����Sr)�get)�.0�col_name�replace_cols  �r�
<listcomp>z_pars_lveps.<locals>.<listcomp>Fs'���U�U�U�H�K�O�O�H�h�7�7�U�U�UrN���zlveps output was incorrect: %s�data�lveps_lines��)�extra�ID�TID�PID�EP�PNO�TNOzCan't parse lveps output: %s)�split�pop�	enumerate�len�isdigit�insert�log�error�max�dictr�int�KeyError�
ValueError�str)
�lveps_outputr(�header_line�columns_name�
lveps_data�user_id�lveps_line_index�
lveps_line�lveps_line_splited�has_id�lveps_dict_line�er"s
            @r�_pars_lvepsrK s)���@�$�$�T�*�*�K��/�/�!�$�$�K��$�$�&�&�L�!�%�0�0�K�U�U�U�U��U�U�U�L��J��G�(1�+�(>�(>�!B�!B�$��*��	��'�-�-�d�C��4E�4E�a�4G�H�H���)�*�*�c�,�.?�.?�?�d�EW�XY�EZ�Eb�Eb�Ed�Ed���	-�!+�!1�!1�$��L�8I�8I�!�8K�!L�!L���%�%�a��,�,�,��!�"�"�c�,�&7�&7�7�7��I�I�6�
����C��;K�a�;O�4P�4P�Rb�eg�Rg�4g�(h�i�Kk�I�
l�
l�
l��E��s�<�1C�D�D�E�E���	B��/�-�-�d�3�3�4�4�G�
�#�E�*�*���
�
�
���
����"1�J�w��)-���J�w���&��w��#�#�E�4�0�0�0�0�����d�+�+�+�����t�,�,�,�����t�,�,�,�����d�+�+�+�
B�N]�
�7�#�E�*�3��/B�/B�5�/I�/I�+J�+J�K�K����)�
B�
B�
B��	�	�8�#�a�&�&�A�A�A�A�A�A�A�A�����
B�����s*�/F3�3
G�?G�1J	�	K
�(K�K
c�<�t��}t|��S�N)�_get_lveps_outputrK�r@s r�	get_lvepsrPos��$�&�&�L��|�$�$�$rc�T�tgd�d���}|�dd��S)N)z/usr/sbin/lvepsz-c�1z-pz-dz-nz-oz;id:10,ep:10,pno:10,pid:15,tno:5,tid:15,cpu:7,mem:15,com:256F)�convert_to_strzutf-8�replace)r�decoderOs rrNrNtsE��� j� j� j�����L����w�	�2�2�2rc�F�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
dd
�Zd	S)�SQLSnapshotc�`�d|_t��|_t��|_dSrM)�_mysql_connr;�_dblogin_cplogin_map�set�	_db_users��selfs r�__init__zSQLSnapshot.__init__s%�����$(�F�F��!�������rc�.�|���|SrM)�connectr]s r�	__enter__zSQLSnapshot.__enter__�s���������rc�.�|���dSrM)�close)r^�exc_type�exc_val�exc_tbs    r�__exit__zSQLSnapshot.__exit__�s���
�
�����rc�>�tstd���t��}|d}|d}|�dd��}	tj|||dd���|_d	S#tj$r!}tt|�����d	}~wwxYw)
zB
        Obtain access data and connect to mysql database
        z/need install mysql server for binding _mysql.so�login�pass�host�	localhostT�utf8mb4)rl�user�passwd�use_unicode�charsetN)	�MySQLdbrrrrarYr	rr?)r^�access�mysql_login�
mysql_pass�
mysql_hostrJs      rrazSQLSnapshot.connect�s����	O��M�N�N�N������W�o���F�^�
��Z�Z���4�4�
�	0�&��J�[�6@�d�7@� B� B� B�D������'�	0�	0�	0�'��A���/�/�/�����	0���s�A,�,B�;B�Bc�F�tt����|_dS)z;
        Refresh <database user>:<system user> map
        N)r;rrZr]s r�_refresh_mapzSQLSnapshot._refresh_map�s!��%)�)>�)@�)@�$A�$A��!�!�!rc�8�|j���dS)z(
        Close Mysql connection
        N)rYrdr]s rrdzSQLSnapshot.close�s��	
���� � � � � rc�0�t��}	|j���}|�d��|���}nC#t
j$r1}t�t|����Yd}~nd}~wwxYw|S)NzSHOW FULL PROCESSLIST)
�tuplerY�cursor�execute�fetchallrsr	r8�warningr?)r^�resultr}rJs    r�_raw_processlistzSQLSnapshot._raw_processlist�s�������	 ��%�,�,�.�.�F��N�N�2�3�3�3��_�_�&�&�F�F���'�	 �	 �	 ��K�K��A���������������	 �����
s�AA�B�"'B�Bc�$�t��}|���D]l}|d}|d}|d}|dpd}|dkr�+|||g}|�|t����}|�|��|||<�m|S)zN
        Group processlist by database user name
        :rtype: dict
        r$�r)�r&�Sleep)r;r�r�list�append)	r^�process_snapshot�	sql_tuple�db_username�sql_cmd�sql_time�	sql_query�
snapshot_line�grouped_by_users	         r�_get_sql_process_listz!SQLSnapshot._get_sql_process_list�s���
 �6�6���.�.�0�0�	<�	<�I�#�A�,�K���l�G� ��|�H�!�!��*��I��'�!�!��$�h�	�:�M�.�2�2�;����G�G�O��"�"�=�1�1�1�,;��[�)�)��rNc���|���}t|�����|jz
}|rp|���t
�dtt|����dd���|D]}|j�	|���t��}|pt��p
t��}t|�����D]a\}}|j
�|��}	|	�@|	|vr<|�|	t����}
|
�|��|
||	<�b|S)z�
        :param cplogin_lst: a list of users to retrieve data;
            None if the data is returned for all users registered in the control panel
        :return: sql queries for each user
        z>New database user(s) %s detected; database users map refreshedr$���)r�r[�keysr\ryr8�debugr?r��addr;r�itemsrZr�extend)r^�cplogin_lstr��new_db_users�new_db_user�sql_snapshot�cplogin_lst_r��sql_snap�cp_username�	sql_snap_s           rrzSQLSnapshot.get�sg�� �5�5�7�7���+�0�0�2�2�3�3�d�n�D���	0��������I�I�5�69�$�|�:L�:L�6M�6M�a�PR�d�6S�
U�
U�
U� ,�
0�
0����"�"�;�/�/�/�/��v�v��"�9�g�i�i�9�4�6�6��%)�*:�*@�*@�*B�*B�%C�%C�	6�	6�!�K���3�7�7��D�D�K��&�;�,�+F�+F�(�,�,�[�$�&�&�A�A�	�� � ��*�*�*�,5��[�)���rrM)rr
rr_rbrhraryrdr�r�rrrrrWrW~s����������
������0�0�0�"B�B�B�!�!�!���� � � �(�����rrW)!�
__future__r�builtinsr�typingrrr�logging�	getLoggerr8rsr	r�ImportErrorrJr�r?�	Exception�clcommon.cpapirrr�clcommon.cpapi.cpapiexceptionsr�clcommon.utilsrrrKrPrN�objectrWrrr�<module>r�s���'�&�&�&�&�&�������'�'�'�'�'�'�'�'�'�'������g���!�!��

��N�N�N�A�A�A�A�A�A�A���
�
�
��G��I�I�(�3�3�q�6�6�1�2�2�2�
�
�
�
�
�	�
�
�
�
�
�
�
�
�����	
����
�H�H�H�H�H�H�H�H�H�H�8�8�8�8�8�8�*�*�*�*�*�*�*���
�
�
��:������
����
L�L�L�^%�%�%�
3�3�3�e�e�e�e�e�&�e�e�e�e�es&�
7�A7�1A2�2A7�;B�B!� B!