File: //lib/python3.9/site-packages/iotop/__pycache__/ui.cpython-39.opt-1.pyc
a
��Jc�b � @ sp d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZmZmZmZ d dl mZ d dlmZ d dlmZ d dlmZ g d�Zd d
� Zd d � Zd
d� Z dd� Z dd� Z e � Z G dd� de �Z!G dd� de �Z"G dd� de �Z#G dd� de �Z$G dd � d e �Z%G d d � d e&�Z'd d � Z(d!d"� Z)d#d$� Z*d%e
j+d Z,d&d'� Z-dS )(� )�print_functionN)� find_uids�TaskStatsNetlink� ProcessList�Stats�sysctl_task_delayacct)�
ThreadInfo)�VERSION)�ioprio)�IoprioSetError)�B�K�M�G�T�P�Ec C s` | dkrd}n| dk r"d}| } ndS t t�| d d�d �}d|t| �dd| > t| f S ) Nr � �-z0.00 B� �
z %s%.2f %s� )�int�math�log�float�UNITS)�size�signZexpo� r �,/usr/lib/python3.9/site-packages/iotop/ui.py�
human_size- s r! c C s | j rd|d S t|�S )Nz%.2f Kg �@)� kilobytesr! )�options�bytesr r r � format_size9 s r% c C s t | |ot|�| �d S )Nz/s)r% r )r# r �durationr r r �format_bandwidth> s r' c s� � fdd�}�j r4|j}�fdd�}t�� |j � n|j}�fdd�}||j�}||j�}||j� �}|j|j } t
d| �} || � �}
||||
fS )Nc s dt d| � d � S )Nz%.2f %%g��(\�X@g �cA)�min)Zdelay)r& r r �
delay2percentD s z#format_stats.<locals>.delay2percentc s
t � | �S �N)r% �r r& �r# r r �<lambda>H � z format_stats.<locals>.<lambda>c s t � | |�S r* )r' r+ r, r r r- L s �r ) � accumulated� stats_accum�timeZstats_accum_timestamp� stats_delta�blkio_delay_total�swapin_delay_total�
read_bytes� write_bytes�cancelled_write_bytes�max) r# Zprocessr& r) �statsZdisplay_format�io_delay� swapin_delayr5 Z
written_bytesr6 r )r& r# r � format_statsA s
r< c
C sJ zt td��� �� �W S tyD } zt| � W Y d } ~ dS d } ~ 0 0 d S )Nz/proc/sys/kernel/pid_max� )�len�open�read�strip� Exception�print)�er r r �get_max_pid_widthV s
rE c @ s e Zd ZdS )� CancelInputN��__name__�
__module__� __qualname__r r r r rF d r. rF c @ s e Zd ZdS )�
InvalidIntNrG r r r r rK e r. rK c @ s e Zd ZdS )�
InvalidPidNrG r r r r rL f r. rL c @ s e Zd ZdS )�
InvalidTidNrG r r r r rM g r. rM c @ s e Zd ZdS )�InvalidIoprioDataNrG r r r r rN h r. rN c @ s� e Zd Zdd� dfdd� dfdd� dfdd� dfdd� dfd d� dfd
d� dfd d� dfgZd d
� Zdd� Zdd� Zdd� Zdd� Zd0dd�Z d1dd�Z
dd � Z d d � Z d d � Z
d!d"� Zd2d#d$�Zd3d&d'�Zd(d)� Zd*d+� Zd,d-� Zd.d/� ZdS )4�IOTopUIc C s | j S r* )�pid��p�sr r r r- q r. zIOTopUI.<lambda>Fc C s | � � S r* )Zioprio_sort_keyrQ r r r r- r r. c C s | � � S r* )�get_userrQ r r r r- s r. c C s |j S r* )r5 rQ r r r r- t r. Tc C s |j |j S r* )r6 r7 rQ r r r r- u r. c C s |j S r* )r4 rQ r r r r- v r. c C s |j pt|jp|j �S r* )r3 r r5 r6 rQ r r r r- y s c C s | � � S r* )� get_cmdlinerQ r r r r- { r. c C sr || _ || _d| _tj| j d | _| jjsn|| _| �� z t �
� t � � t � d� W n t j
yl Y n0 d S )N� r r )� process_listr# � sorting_keyrO � sorting_keys�sorting_reverse�batch�win�resize�cursesZuse_default_colorsZ start_color�curs_set�error)�selfr\ rW r# r r r �__init__~ s zIOTopUI.__init__c C s | j �� \| _| _d S r* )r\ Zgetmaxyx�height�width�ra r r r r] � s zIOTopUI.resizec
C sX d}t �� }| jjs.|�tj�� t jt j B � | jj
d u sH|| jj
k �rT| j � � \}}| �
|dk||| j j� | jj
d ur�|d7 }|| jj
kr��qTn |dkr�d}z|�| jjd �}W nD t jy� } z*|jr�|jd tjkr�g }n� W Y d }~n
d }~0 0 |D ]&\}}|t jt jB @ �r t�d� �q | jj�s:| �� |r.| j�� } | �| � q.d S )Nr r g @�@)�select�pollr# r[ �register�sys�stdin�filenoZPOLLINZPOLLPRI�
iterationsrW �refresh_processes�refresh_displayr& �
delay_secondsr` �args�errnoZEINTRZPOLLERRZPOLLHUP�exitr] r\ �getch�
handle_key)
ra rl rg �total�actualZeventsrD �fdZevent�keyr r r �run� s>
��
z IOTopUI.runc C s | j | _ d S r* )rZ re r r r �reverse_sorting� s zIOTopUI.reverse_sortingc C s� | j }| j |7 _ td| j �| _ tttj�d | j �| _ | jsh| j dv rh|dkrZd| _ n|dkrhd| _ || j kr�tj| j d | _d S )Nr r �r= rV � � )rX r8 r( r> rO rY �
has_swapin_iorZ )ra ZdeltaZorig_sorting_keyr r r �adjust_sorting_key� s
zIOTopUI.adjust_sorting_keyNc C s� | j �ddtd�tjB | j� | j �dd|tj� | j �� t� � t�
d� | j � dt |��}t�
d� t�
� |dvr~|S |r�t� �|S )Nr r � )Nr )r\ �hline�ordr^ �A_NORMALrd �addstr�A_BOLD�refreshZechor_ Zgetstrr> ZnoechorF �ra �prompt�defaultZempty_is_cancel�inpr r r �
prompt_str� s
zIOTopUI.prompt_strc C s6 | � |||�}z
t|�W S ty0 t� �Y n0 d S r* )r� r �
ValueErrorrK r� r r r �
prompt_int� s
zIOTopUI.prompt_intc C s< z | � d�W S ty$ t� �Y n ty6 � Y n0 d S )NzPID to ionice: )r� rK rL rF re r r r �
prompt_pid� s
zIOTopUI.prompt_pidc C s< z | � d�W S ty$ t� �Y n ty6 � Y n0 d S )NzTID to ionice: )r� rK rM rF re r r r �
prompt_tid� s
zIOTopUI.prompt_tidc C sb z.|d ur | � d| |d�}n| � dd d�}W n tyF t� �Y n0 |dk sX|dkr^t� �|S )Nz'I/O priority data (0-7, currently %s): FzI/O priority data (0-7): r r} )r� rK rN )ra � ioprio_datar� r r r � prompt_data� s �� zIOTopUI.prompt_datac C sf z|� |�}W n ty$ d}Y n0 t|�d }| j�ddtd�tjB | j� | j� dd|tj
� t|�}t |�D ]D\}}d| } ||u r�tj }
ntj}
| j� d|| |
� |t| �7 }qt| j�
� } | tjtd�fv r�|dkr�|d8 }q2q�| tjtd�fv �r||k �r|d7 }q2q�| tjtd�td �fv �r6|| S | d
tjtjtjtd �td �fv r�t� �q�q2d S )
N���r r r� � %s �l�r�
�
� �q�Q)�indexr� r> r\ r� r� r^ r� rd �insstrr� � enumerate� A_REVERSErs �KEY_LEFT� KEY_RIGHTZ KEY_ENTERZ
KEY_CANCELZ KEY_CLOSEZKEY_EXITrF ) ra r� Z display_listZret_listZselectedZset_len�offset�i�itemZdisplay�attrrx r r r �
prompt_set� s8
�zIOTopUI.prompt_setc C s4 d}g d�}g d�}|d u r d}| � ||||�}|S )NzI/O priority class: )z Real-timez Best-effortZIdle)Zrt�be�idler )r� )ra � ioprio_classr� Zclasses_promptZ classes_retr� r r r � prompt_class s zIOTopUI.prompt_class�Error!c C sN | j �ddtd�tjB | j� | j �ddd| tj� | j �� t �
d� d S )Nr r r� z %s ) r\ r� r� r^ r� rd r� r� r� r1 �sleep)ra r` r r r � prompt_error" s
zIOTopUI.prompt_errorc C s, | j �ddtd�tjB | j� | j �� d S )Nr r r� )r\ r� r� r^ r� rd r� re r r r � prompt_clear( s zIOTopUI.prompt_clearc " s� � fdd�}� fdd�}� fdd�}� fdd�}t d �d
d � t d �d
d � t d�� fdd �t d�� fdd �t d�|t d�|t d�|t d�|t d�|t d�|t d�|t d�|tj� fdd �tj� fdd �tj� fd d �tj� fd d �i}|�|d d � �}|� d S ) Nc s � j jdN _d S �NT)r# r/ r re r r �toggle_accumulated- s z.IOTopUI.handle_key.<locals>.toggle_accumulatedc s � j jdN _d S r� )r# �onlyr re r r �toggle_only_io/ s z*IOTopUI.handle_key.<locals>.toggle_only_ioc s( � j jdN _� j�� � j�� d S r� )r# � processesrW �clearrm r re r r �toggle_processes1 s
z,IOTopUI.handle_key.<locals>.toggle_processesc
sX z�� j jr � �� } � j�| �}n� �� }t|� jj�}|�� }t �
|�\}}� � |�}|dkrdd}n
� � |�}|�
||� � j�� � j�� W n� ty� } z � �d|j � W Y d }~n�d }~0 ty� � �d� Y nv ty� � �d� Y n\ t�y � �d� Y n@ t�y0 � �d� Y n$ t�yJ � �� Y n
0 � �� d S )Nr� r z Error setting I/O priority: %szInvalid process id!zInvalid thread id!zInvalid I/O priority data!zInvalid integer!)r# r� r� rW Z get_processr� r �taskstats_connection�
get_ioprior
Zto_class_and_datar� r� Z
set_ioprior� rm r r� �errrL rM rN rK rF r� )rP Z exec_unit�tidZ ioprio_valuer� r� rD re r r �ionice5 s> ��
$ z"IOTopUI.handle_key.<locals>.ionicer� c S s
t �d�S �Nr �ri rr r r r r r- Z r. z$IOTopUI.handle_key.<locals>.<lambda>r� c S s
t �d�S r� r� r r r r r- \ r. r� c s � � � S r* �rz r re r r r- ^ r. �Rc s � � � S r* r� r re r r r- ` r. �a�A�o�OrR r r� �Ic s
� � d�S )Nr� �r r re r r r- r r. c s
� � d�S )Nr r� r re r r r- t r. c s � � ttj� �S r* �r r> rO rY r re r r r- v r. c s � � ttj��S r* r� r re r r r- x r. c S s d S r* r r r r r r- { r. )r� r^ r� r� ZKEY_HOMEZKEY_END�get)ra rx r� r� r� r� Z key_bindings�actionr re r rt , sN $
�#zIOTopUI.handle_keyc s� �fdd�}�fdd�}t t|�jj�� ��}tj�j d � �jj rPdd� �ndd� �|j
� �fd d��j d
� �jj s�|�j
d d �= t t||��S ) Nc
sR t � j| � jj�}|\}}}}dt }| jf}|d7 }|| �� f7 }|d7 }|| �� d d� f7 }|d7 }|||f7 }� jr�|d7 }|||f7 }n� jj r�|d7 }|d7 }|d 7 }|| }| �
� } � jj �s.� j t |� }
d
|
k r�t | �k �r.n n6|
d d
} |
| d } | d | � d | | d � } || 7 }� jj �sN|d � j � }|S )
Nz%%%ddz %4sz %-8s� z
%11s %11sz %7s %7sr� )z
?unavailable?r� r r �~)
r< r# rW r& �
MAX_PID_WIDTHrP r� rT r~ r[ rU rd r> )
rR r9 r: r; r5 r6 �format�params�lineZcmdlineZremaining_lengthZlen1Zoffset2re r r r� s:
z IOTopUI.get_data.<locals>.formatc s � j j p| �� j j�S r* )r# r� Z did_some_ior/ �rR re r r �
should_format� s
�z'IOTopUI.get_data.<locals>.should_formatr c S s | j S r* )r0 r� r r r r- � r. z"IOTopUI.get_data.<locals>.<lambda>c S s | j S r* )r2 r� r r r r- � r. c s � | �| ��S r* r r� )rx � stats_lambdar r r- � r. )rx �reverser )�list�filterrW r� �valuesrO rY rX r# r/ �sortrZ r[ rc �map)ra r� r� r� r )rx ra r� r �get_data~ s
�
�zIOTopUI.get_datac sl dt �j|d |��d�t �j|d |��d�f dt �j|d |��d�t �j|d |��d�f g�tdtd �d }�jjr�|d7 }n|d 7 }|d
d d d
dddg}tj�_�jr�t � dkr�d�_��
d� �� � }�jj �rdg| }t �
d�� � fdd�|D �}� fdd��D ���jj�r��jjdk�rf�D ]}t|� �q6�jjt|�k�rftd�|�� |D ]} t| �d�� �qjtj�� �nڈj�� t��D ]$\�}�j��d|d �j� � �q��j� t ��dt d�t j B �j� �j}
t!t |��D ]���j�s�dv �r�q�t j } |� } ��j"k�r:| dd � } ��j"k�rf| t j#O } | �j$�r`d�pbd 7 } | d |
� } |
t | �8 }
�j�| | � �q��jdu�r�d }
nd }
t%t |��j&d tt'|
�� �}t!|�D ]j�zL���fd d �}z||� � W n& t(�y ||� �d�� Y n0 W n t j)�y4 Y n0 �q�|
�r^�j�*�j&t �� d|
t j#� �j�+� d S ) Nz,Total DISK READ : %s | Total DISK WRITE : %sr � r z,Actual DISK READ: %s | Actual DISK WRITE: %s� r� �PIDZTIDz PRIOz USERz DISK READz DISK WRITEz SWAPINz IOz COMMANDFz TIMEz %H:%M:%S c s g | ] }� | �qS r r )�.0r� �� current_timer r �
<listcomp>� r. z+IOTopUI.refresh_display.<locals>.<listcomp>c s g | ] }� | �qS r r )r� rS r� r r r� � r. r r zutf-8r{ �>�<zoCONFIG_TASK_DELAY_ACCT and kernel.task_delayacct sysctl not enabled in kernel, cannot determine SWAPIN and IO %c s �j �� t�� d d| � d S )Nr r )r\ r� r> )r� )r� ra �summaryr r �
print_line� s z+IOTopUI.refresh_display.<locals>.print_line),r' r# �rjustr8 r� r� r Zhas_blkio_delay_totalr~ r r r� r1 �strftimer[ �quietrC r �join�encoderi �stdout�flushr\ Zeraser� r� rd r� r> r� r^ r� �rangerX r� rZ r( rc �bool�UnicodeEncodeErrorr` r� r� )ra Z
first_timeru rv r& rP Ztitles�linesrS r� Zremaining_colsr� �titleZ
status_msgZ num_linesr� r )r� r� ra r� r rn � s� ���
�
�
�zIOTopUI.refresh_display)NT)NT)N)r� )rH rI rJ rY rb r] ry rz r r� r� r� r� r� r� r� r� r� rt r� rn r r r r rO n s8
�
�!
R1rO c C sf |j rt�tjtj� n$dd� }t�tj|� t�tj|� t|�}t||�}t| ||�}|� � d S )Nc _ s t �d� d S r� r� )rp �kwargsr r r �
clean_exit s z$run_iotop_window.<locals>.clean_exit)
r[ �signal�SIGPIPE�SIG_DFL�SIGINT�SIGTERMr r rO ry )r\ r# r� r� rW Zuir r r �run_iotop_window s
r� c
C s� z$| j rtd | �W S t�t| �W S W n� ty~ } zB|jtjkrht|tj d� tdtj d� t�
d� n� W Y d }~ndd }~0 tj y� } zBt |�}|�
d�dkr�|�
d�dks�|�
d�dkr�n� W Y d }~n
d }~0 0 d S )N)�filea�
The Linux kernel interfaces that iotop relies on now require root priviliges
or the NET_ADMIN capability. This change occured because a security issue
(CVE-2011-2494) was found that allows leakage of sensitive data across user
boundaries. If you require the ability to run iotop as a non-root user, please
configure sudo to allow you to run iotop as root.
Please do not file bugs on iotop about this.r ZERRr z
nocbreak()zendwin())r[ r� r^ �wrapper�OSErrorrq �EPERMrC ri �stderrrr r` �str�find)r# rD Zstrer r r � run_iotop s* � � �r� c C s� d}z:dd l }dd l}td� |�dt� t� |� |�|�}W nV ty� dd l}dd l }|j
|dd�}td� |� | � |� � |j
�|�}Y n0 |�� |�dd � |�d
� |�d
� t�|� d S ) Nz
iotop.profr zProfiling using cProfilezcontinuation()r )Z
lineeventszProfiling using hotshotr1 Zcalls�2 )�cProfile�pstatsrC Zrunctx�globals�localsr � ImportError�hotshotZ
hotshot.statsZProfileZruncall�closer9 �loadZ
strip_dirsZ
sort_statsZ print_statsZ
print_callees�os�remove)Z continuationZ prof_filer� r r9 r Zprofr r r �_profile6 s(
r aF %s [OPTIONS]
DISK READ and DISK WRITE are the block I/O bandwidth used during the sampling
period. SWAPIN and IO are the percentages of time the thread spent respectively
while swapping in and waiting on I/O more generally. PRIO is the I/O priority at
which the thread is running (set using the ionice command).
Controls: left and right arrows to change the sorting column, r to invert the
sorting order, o to toggle the --only option, p to toggle the --processes
option, a to toggle the --accumulated option, i to change I/O priority, q to
quit, any other key to force a refresh.c s� zt �t jd� W n t jy. td� Y n0 tjtdt d�} | j ddddd d
d � | j d d
dddd� | j ddddddd� | j ddddd d d d � | j d d!dd"d#d$d%d&� | j d'd(d)d*d#d+d,d&� | j d-d.dd/d d0d � | j d1d2dd3d d4d � | j d5d6dd7d d8d � | j d9d:dd;d<d� | j d=d>d?d@dAdBd � | j dCddDd tj
d � | � � \�}|�rr| � dEdF�
|� � t�� �j�p�g �_�j�p��j�p��j�_�fdGdH�� �j�rʇ fdIdJ�}t|� n� � d S )KNr z8unable to set locale, falling back to the default localeziotop )�usage�versionz-oz--only�
store_truer� Fz1only show processes or threads actually doing I/O)r� �destr� �helpz-bz--batchr[ znon-interactive mode)r� r
r z-nz--iterr rl ZNUMz-number of iterations before ending [infinite])�typer
�metavarr z-dz--delayr ro z#delay between iterations [1 second]ZSECr )r r
r r r� z-pz--pid�pids�appendz"processes/threads to monitor [all]r� )r r
r� r r z-uz--userr� Zuserszusers to monitor [all]ZUSERz-Pz --processesr� z$only show processes, not all threadsz-az
--accumulatedr/ z)show accumulated I/O instead of bandwidthz-kz --kilobytesr" z.use kilobytes instead of a human friendly unitz-tz--timer1 z.add a timestamp on each line (implies --batch)z-qz--quiet�countr� r z/suppress some lines of header (implies --batch)z --profile�profilezUnexpected arguments: r� c s t � �S r* )r� r r, r r r- � r. zmain.<locals>.<lambda>c s z
� � W n Y n0 d S r* r r )� main_loopr r �safe_main_loop� s
z main.<locals>.safe_main_loop)�locale� setlocale�LC_ALL�ErrorrC �optparseZ OptionParser�USAGEr Z
add_optionZ
SUPPRESS_HELP�
parse_argsr` r� r r r[ r1 r� r r )�parserrp r r )r r# r �main\ st
� � � ���
�
�
� ��
�
r ).Z
__future__r r^ rq r r r r rf r� ri r1 Z
iotop.datar r r r r r Z
iotop.versionr Ziotopr
Z iotop.ioprior r r! r% r' r< rE r� rB rF rK rL rM rN �objectrO r� r� r �argvr r r r r r �<module> sL
$
�