Current File : //opt/alt/python37/lib/python3.7/site-packages/lvestats/lib/__pycache__/ustate.cpython-37.pyc
B
��1b�" �
@ s d dl mZ d dlmZ d dlmZmZmZ d dlZe� d�Z
yd dl Z d dl m Z
W nH ek
r� Z z*dZ e
�dee� � G dd � d e�Z
W ddZ[X Y nX y0d d
lmZmZmZ d d lmZ d d lmZ W n ek
r� ed
kr�Y nX dd� Z dd� Z dd� Z G dd� de �Z dS )� )�absolute_import)�zip)�List�Dict�OptionalNZustate)�OperationalErrorzCan't import MySQLdb; %sc @ s e Zd ZdS )�MySQLOperationalErrorN)�__name__�
__module__� __qualname__� r r �D/opt/alt/python37/lib/python3.7/site-packages/lvestats/lib/ustate.pyr s r )� db_access�dblogin_cplogin_pairs�cpusers)� NoPackage)� run_command�__main__c s� | � d�}|�d�}|� � }ddd�� � fdd�|D �}i }d}�x�t|�D �]�\}}|s\qL|� dt|�d �}t|�t|�ko�|d �� } | s�|� dt|�d
�}|�dd � t|�t|�kr�tjd |d
d|td|d �|d � iid� P t t
||��}
| �r\t |
�d��}y
|
d= W n t k
�r2 Y nX |
||< t � || d< || �dd� qL|
�dd� |
�dd� |
�dd� |
�dd� y |
|| d t |
�d��< W qL t
t fk
�r� } zt�dt| �� W dd} ~ X Y qLX qLW |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 ZCPUZCMD)ZSPEEDZCOMc s g | ]}� � ||��qS r )�get)�.0Zcol_name)� replace_colr r
�
<listcomp>F s z _pars_lveps.<locals>.<listcomp>N� � � z lveps output was incorrect: %s�data� lveps_lines� � )�extraZIDZTIDZPIDZEPZPNOZTNOz Can't parse lveps output: %s)�split�pop� enumerate�len�isdigit�insert�log�error�max�dictr �int�KeyError�
ValueError�str) � lveps_outputr Z header_lineZ columns_nameZ
lveps_dataZuser_idZlveps_line_indexZ
lveps_lineZlveps_line_splitedZhas_idZlveps_dict_line�er )r r
� _pars_lveps sL
&
&r1 c C s t � } t| �S )N)�_get_lveps_outputr1 )r/ r r r
� get_lvepso s r3 c C s( t ddddddddgd d
�} | �d d �S )
Nz/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)Zconvert_to_strzutf-8�replace)r �decode)r/ r r r
r2 t s
r2 c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
ddd�Z dS )� SQLSnapshotc C s d | _ t� | _t� | _d S )N)� _mysql_connr* �_dblogin_cplogin_map�set� _db_users)�selfr r r
�__init__ s zSQLSnapshot.__init__c C s | � � | S )N)�connect)r< r r r
� __enter__� s zSQLSnapshot.__enter__c C s | � � d S )N)�close)r< �exc_typeZexc_val�exc_tbr r r
�__exit__� s zSQLSnapshot.__exit__c
C s~ t s td��t� }|d }|d }|�dd�}yt j|||ddd�| _W n0 t jk
rx } ztt|���W d d }~X Y nX d S )
zB
Obtain access data and connect to mysql database
z/need install mysql server for binding _mysql.soZlogin�pass�hostZ localhostTZutf8mb4)rE �userZpasswdZ use_unicode�charsetN) �MySQLdbr r r r> r8 r r r. )r< �accessZ mysql_loginZ
mysql_passZ
mysql_hostr0 r r r
r> � s zSQLSnapshot.connectc C s t t� �| _dS )z;
Refresh <database user>:<system user> map
N)r* r r9 )r< r r r
� _refresh_map� s zSQLSnapshot._refresh_mapc C s | j �� dS )z(
Close Mysql connection
N)r8 r@ )r< r r r
r@ � s zSQLSnapshot.closec
C s^ t � }y | j�� }|�d� |�� }W n2 tjk
rX } zt�t |�� W d d }~X Y nX |S )NzSHOW FULL PROCESSLIST)
�tupler8 �cursor�executeZfetchallrH r r' Zwarningr. )r< �resultrL r0 r r r
�_raw_processlist� s
z SQLSnapshot._raw_processlistc C st t � }xh| �� D ]\}|d }|d }|d }|d p6d}|dkrBq|||g}|�|t� �}|�|� |||< qW |S )zN
Group processlist by database user name
:rtype: dict
r � r � r ZSleep)r* rO r �list�append) r< �process_snapshotZ sql_tuple� db_usernameZsql_cmdZsql_timeZ sql_queryZ
snapshot_lineZgrouped_by_userr r r
�_get_sql_process_list� s
z!SQLSnapshot._get_sql_process_listNc C s� | � � }t|�� �| j }|r\| �� t�dtt|��dd� � x|D ]}| j� |� qHW t
� }|ppt � ppt� }xTt|� � �D ]D\}}| j
�|�} | dk r�| |kr�|�| t� �}
|
�|� |
|| < q�W |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 ���N)rV r: �keysr; rJ r' �debugr. rR �addr* r �itemsr9 r �extend) r< Z cplogin_lstrT Z new_db_usersZ new_db_userZ sql_snapshotZ cplogin_lst_rU Zsql_snapZ cp_usernameZ sql_snap_r r r
r � s"
zSQLSnapshot.get)N) r r
r r= r? rC r> rJ r@ rO rV r r r r r
r7 ~ s
r7 )!Z
__future__r �builtinsr �typingr r r ZloggingZ getLoggerr' rH r r � ImportErrorr0 rY r. � ExceptionZclcommon.cpapir r r Z clcommon.cpapi.cpapiexceptionsr Zclcommon.utilsr r r1 r3 r2 �objectr7 r r r r
�<module> s,
" O
Mini Shell