Your IP : 216.73.216.224
U
$�DYOk � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z z e e d�szd dl
m
Z
e
e _
W n eefk
r� e j e _
Y nX d dlmZ d dlmZmZmZmZ d dlmZ d dlmZ d d lmZ d d
l m Z d d d
ddddgZ dd� Z dd� Z dd� Z!dd� Z"dd� Z#e#� Z$G d d � d e%�Z&G d d � d e%�Z'G d d!� d!e%�Z(G d"d#� d#e%�Z)G d$d%� d%e%�Z*G d&d'� d'e+�Z,d(d)� Z-d*d+� Z.d,d-� Z/d.e
j0d Z1d/d0� Z2dS )1� )�print_functionN� monotonic)r )� OrderedDict)� find_uids�TaskStatsNetlink� ProcessList�Stats)�
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)�sizeZsignZexpo� r �*/usr/lib/python3/dist-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_sizeI s r% c C s t | |ot|�| �d S )Nz/s)r% r )r# r �durationr r r �format_bandwidthO 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 �
delay2percentV s z#format_stats.<locals>.delay2percentc s
t � | �S �N)r% �r r& �r# r r �<lambda>Z � z format_stats.<locals>.<lambda>c s t � | |�S r* )r' r+ r, r r r- ^ s �r ) � accumulated� stats_accum�timer Zstats_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_statsS s
r< c
C sL zt td��� �� �W S tk
rF } zt| � W Y �dS d } ~ X Y nX d S )Nz/proc/sys/kernel/pid_max� )�len�open�read�strip� Exception�print)�er r r �get_max_pid_widthi s
rE c @ s e Zd ZdS )� CancelInputN��__name__�
__module__� __qualname__r r r r rF x s rF c @ s e Zd ZdS )�
InvalidIntNrG r r r r rK | s rK c @ s e Zd ZdS )�
InvalidPidNrG r r r r rL � s rL c @ s e Zd ZdS )�
InvalidTidNrG r r r r rM � s rM c @ s e Zd ZdS )�InvalidIoprioDataNrG r r r r rN � s rN c @ s� e Zd Zdd� dfdd� dfdd� dfdd� dfdd� dfd d� dfd
d� dfd d� dfgZd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Z d2dd�Z
d3dd �Z d d � Z d d � Z
d!d"� Zd#d$� Zd4d%d&�Zd5d(d)�Zd*d+� Zd,d-� Zd.d/� Zd0d1� ZdS )6�IOTopUIc C s | j S r* )�pid��p�sr r r r- � r. zIOTopUI.<lambda>Fc C s | � � S r* )Zioprio_sort_keyrQ r r r r- � r. c C s | � � S r* )�get_userrQ r r r r- � r. c C s |j S r* )r5 rQ r r r r- � r. Tc C s |j |j S r* )r6 r7 rQ r r r r- � r. c C s |j S r* )r4 rQ r r r r- � r. c C s |j pt|jp|j �S r* )r3 r r5 r6 rQ r r r r- � s c C s | � � S r* )� get_cmdlinerQ r r r r- � r. c C st || _ || _d| _tj| j d | _| jjsp|| _| �� z t �
� t � � t � d� W n t j
k
rn Y nX 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 sR d}t �� }| jjs.|�tj�� t jt j B � | jj
d ksH|| jj
k �rN| j � � \}}| �
|dk||| j j� | jj
d k r�|d7 }|| jj
kr��qNn |dkr�d}z|�| jjd �}W nB t jk
r� } z"|jr�|jd tjkr�g }n� W 5 d }~X Y nX |D ]"\}}|t jt jB @ r�t�d� q�| jj�s4| �� |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�currentZeventsrD �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 s2 | j }| �|�| _ || j kr.tj| j d | _d S �Nr )rX �get_sorting_keyrO rY rZ )ra �deltaZorig_sorting_keyr r r �adjust_sorting_key� s
zIOTopUI.adjust_sorting_keyc C s0 | j }||7 }td|�}tttj�d |�}|S )Nr r )rX r8 r( r> rO rY )ra r} Znew_sorting_keyr r r r| � s
zIOTopUI.get_sorting_keyNc C s� | j �ddtd�tjB | j� | j �dd|tj� | j �� t� � t�
d� | j � dt |��}t�
d� t�
� |dkr~|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 s8 | � |||�}z
t|�W S tk
r2 t� �Y nX d S r* )r� r �
ValueErrorrK r� r r r �
prompt_int� s
zIOTopUI.prompt_intc C s@ z | � d�W S tk
r& t� �Y n tk
r: � Y nX 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 tk
r& t� �Y n tk
r: � Y nX d S )NzTID to ionice: )r� rK rM rF re r r r �
prompt_tid s
zIOTopUI.prompt_tidc C sd z.|d k r | � d| |d�}n| � dd d�}W n tk
rH t� �Y nX |dk sZ|dkr`t� �|S )Nz'I/O priority data (0-7, currently %s): FzI/O priority data (0-7): r � )r� rK rN )ra � ioprio_datar� r r r � prompt_data
s � � zIOTopUI.prompt_datac C sh z|� |�}W n tk
r& d}Y nX t|�d }| j�ddtd�tjB | j� | j� dd|tj
� t|�}t |�D ]D\}}d| } ||kr�tj }
ntj}
| j� d|| |
� |t| �7 }qv| j�
� } | tjtd�fkr�|dkr�|d8 }q4q�| tjtd�fk�r||k �r|d7 }q4q�| tjtd�td �fk�r8|| S | d
tjtjtjtd �td �fkr�t� �q�q4d S )
N���r r r z %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 s@
� �zIOTopUI.prompt_setc C s8 d}dddg}dddg}|d kr$d}| � ||||�}|S ) NzI/O priority class: z Real-timez Best-effortZIdleZrtZbe�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_errorE 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_clearK 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_accumulatedP s z.IOTopUI.handle_key.<locals>.toggle_accumulatedc s � j jdN _d S r� )r# �onlyr re r r �toggle_only_ioS 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_processesV s
z,IOTopUI.handle_key.<locals>.toggle_processesc
s` z�� j jr � �� } � j�| �}n� �� }t|� jj�}|�� }t �
|�\}}� � |�}|dkrdd}n
� � |�}|�
||� � j�� � j�� W n� tk
r� } z� �d|j � W 5 d }~X Y n� tk
r� � �d� Y n~ tk
r� � �d� Y nb tk
�r � �d� Y nD tk
�r6 � �d� Y n& tk
�rR � �� Y n
X � �� 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 �ionice[ s> ��
z"IOTopUI.handle_key.<locals>.ionicer� c S s
t �d�S �Nr �ri rr r r r r r- � 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. c s
� � d�S r{ r� r re r r r- � r. c s � � ttj� �S r* �r~ r> rO rY r re r r r- � r. c s � � ttj��S r* r� r re r r r- � 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 O sN $
�#zIOTopUI.handle_keyc s� �fdd�}�fdd�}t t|�jj�� ��}tj�j d � �jj rPdd� �ndd� �|j
� �fd d��j d
� t t ||��S ) Nc
s� t � j| � jj�}|\}}}}tjr2d||f }nd}dt }|d | j| �� | � � d d� |||f }| �
� } � jj s؈ j t
|� }
d|
k r�t
| �k r�n n6|
d d } |
| d } | d | � d | | d � } || 7 }� jj s�|d � j � }|S ) Nz%7s %7s z ?unavailable? z%%%ddz %4s %-8s %11s %11s %s� r r �~)r< r# rW r& r �has_blkio_delay_total�
MAX_PID_WIDTHrP r� rT rU r[ rd r> )
rR r9 r: r; r5 r6 Z delay_statsZ
pid_format�lineZcmdlineZremaining_lengthZlen1Zoffset2re r r �format� s2 � 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 �reverse)
�list�filterrW r� �valuesrO rY rX r# r/ �sortrZ �map)ra r� r� r� r )rx ra r� r �get_data� s
�
�zIOTopUI.get_datac s� 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}��� }�jjr�dg| }t�d�� � fdd�|D �}� fdd�|D �}�jj �r^�jj
dk�r<|D ]} t | � �q �jj
t |�k�r<t d�
|�� |D ]}
t |
� �q@tj�� �n��j�� tj�rvd } nd} g } g }
�jj�r��d�}��d�}tddd d d �jj�r�d nd fd!�jj�r�d"nd#fd$�jj�r�d%nd&fd'd(�j�r�d)nd*fd+|| �� fd,|| �� fd-|d �� fd.|d �� fg
�}d}|� � D ]�\}}|�r|d/|d0|g}d|d1k�rndnt j ddg}nd/|d2g}dddg}| �r�|�r�t d�
| | �d�
|� ��j k�r�| �!|� |
�!|� |d7 }n | | |7 < |
| |7 < �qDt |��t t"|���t t"| ��}t | �}�j#� � | | }|d3k �rRg }d�|d4k �rdg }d�|d5k �rvd } d}|d6k �r�g } g }
d}�j#� � | | }t$t |�|�}t%|�D ]$\�} �j�&�d| d �j � � �q�|�r�j�'�dt(d�t j)B �j � d}�j }t*t |��D ]��t j)}|� } ��j+k�rB| dd � } ��j+k�rn|t j,O }| �j�rhd7�pjd87 } | d |� } |t | �8 }| �r��j�&�|| |� |t | �7 }�qt*|�D ]p�zP����fd9d:�} z| |� � W n( t-k
�r | |� �.d;�� Y nX W n t j/k
�r" Y nX �q�t*|�D ]�} �j#| | | } �j�'| dt(d�t j)B �j � d}t*t | | ��D ]B��j�0| || | � t j)|
| � B � |t | | � �7 }�qx�q0| �r�j�0�j#d d| t j,� �j�1� d S )<Nz.Total DISK READ: %s | Total DISK WRITE: %sr � r z.Current DISK READ: %s | Current DISK WRITE: %s� r �PIDZTIDz PRIOz USERz DISK READz DISK WRITEz SWAPINz IOz COMMANDz TIMEz %H:%M:%S c s g | ] }� | �qS r r )�.0r� �� current_timer r �
<listcomp>� s z+IOTopUI.refresh_display.<locals>.<listcomp>c s g | ] }� | �qS r r )r� rS r� r r r� � s r r zNCONFIG_TASK_DELAY_ACCT not enabled in kernel, cannot determine SWAPIN and IO %r� )�keysr )�anyr� )r� �quit)r� r� r� �allZactiverR ZthreadsZprocsr� Z bandwidthZaccum)r� r r� ZascZdesc�left�right�home�endz z: r� �:r= rV r� r� �>�<c s �j �� � � d| � d S r� )r\ r� )r� )r� � len_summary�
len_titlesra r r �
print_lineB s z+IOTopUI.refresh_display.<locals>.print_linezutf-8)2r' r# �rjustr8 r� r� r� r1 �strftimer[ �quietrC r �joinri �stdout�flushr\ Zeraser r� �helpr| r r� r/ rZ rA �itemsr^ Z A_UNDERLINEr> rd �append�boolrc r( r� r� r� r� r� �rangerX r� �UnicodeEncodeError�encoder` r� r� ) ra Z
first_timeru rv r& ZsummaryrP Ztitles�linesrS r� Z
status_msgZ
help_linesZ
help_attrs�prev�nextr� Z help_linerx Z help_itemZ help_attrZlen_status_msgZlen_helpZ max_linesZ num_lines�posZremaining_colsr� �titler� Zlnr� r )r� r� r� r� ra r rn � s ���
�
�
0
( �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� r� rt r� rn r r r r rO � s:
�
�!
!
U'rO 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_windowY s
r c
C s� z$| j rtd | �W S t�t| �W S W nX tk
r| } z:|jtjkrjt|tj d� tdtj d� t�
d� n� W 5 d }~X Y nX d S )N)�filea�
The Linux kernel interfaces that iotop relies on now require root privileges
or the NET_ADMIN capability. This change occurred 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 ) r[ r r^ �wrapper�OSErrorrq ZEPERMrC ri �stderrrr )r# rD r r r � run_iotopg s � r c C s� d}z:dd l }dd l}td� |�dt� t� |� |�|�}W nX tk
r� dd l}dd l }|j
|dd�}td� |� | � |� � |j
�|�}Y nX |�� |�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 �_profile� 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 jk
r0 td� Y nX 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 � | j dEdFdGdHdId � | � � \�}|�r�| � dJdK�
|� � t�� �j�p�g �_�j�p��j�p��j�_�fdLdM�� �j�r� fdNdO�}t|� n� � d S )PNr z8unable to set locale, falling back to the default localeziotop )Zusage�versionz-oz--only�
store_truer� Fz1only show processes or threads actually doing I/O)r� �destr� r� z-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�pidsr� z"processes/threads to monitor [all]r� )r r r� r� r z-uz--user�strZuserszusers 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�profilez --no-helpZ store_falser� Tz suppress listing of shortcutszUnexpected arguments: r c s t � �S r* )r r r, r r r- � r. zmain.<locals>.<lambda>c s$ z
� � W n t k
r Y nX d S r* )rB 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� s�
� � � � � �
�
�
� ��
� �
r( )3Z
__future__r r^ rq r r r$ r rf r� ri r1 �hasattrr r � RuntimeError� collectionsr Z
iotop.datar 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> sZ
L
�