File: //usr/lib/python3/dist-packages/iotop/__pycache__/ui.cpython-310.pyc
o
$�DYOk � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z ze e d�s=d dl
m
Z
e
e _
W n eefyM e j e _
Y nw d dlmZ d dlmZmZmZmZ d dlmZ d dlmZ d d lmZ d d
l m Z g d �Z d d
� Z dd� Z dd� Z!dd� Z"dd� Z#e#� Z$G dd� de%�Z&G dd� de%�Z'G dd� de%�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)d*� Z2dS )+� )�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 rd}| } 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�sign�expo� r! �*/usr/lib/python3/dist-packages/iotop/ui.py�
human_size; s �r# c C s | j r d|d S t|�S )Nz%.2f Kg �@)� kilobytesr# )�options�bytesr! r! r" � format_sizeI s r' c C s t | |o t|�| �d S )Nz/s)r' r )r% r �durationr! r! r" �format_bandwidthO s r) c s� � fdd�}�j r|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)�delay)r( r! r" �
delay2percentV � 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/ r0 r! r" r1 ^ s �r ) � accumulated� stats_accum�timer �stats_accum_timestamp� stats_delta�blkio_delay_total�swapin_delay_total�
read_bytes� write_bytes�cancelled_write_bytes�max) r% �processr( r, �stats�display_format�io_delay� swapin_delayr: �
written_bytesr; r! )r( r% r" � format_statsS s
rD c
C sF z t td��� �� �W S ty" } z t| � W Y d } ~ dS d } ~ ww )Nz/proc/sys/kernel/pid_max� )�len�open�read�strip� Exception�print)�er! r! r" �get_max_pid_widthi s ��rM c @ � e Zd ZdS )� CancelInputN��__name__�
__module__� __qualname__r! r! r! r" rO x � rO c @ rN )�
InvalidIntNrP r! r! r! r" rU | rT rU c @ rN )�
InvalidPidNrP r! r! r! r" rV � rT rV c @ rN )�
InvalidTidNrP r! r! r! r" rW � rT rW c @ rN )�InvalidIoprioDataNrP r! r! r! r" rX � rT rX 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
d2dd �Z d d � Z d d � Z
d!d"� Zd#d$� Zd3d%d&�Zd4d(d)�Zd*d+� Zd,d-� Zd.d/� Zd0d1� ZdS )5�IOTopUIc C � | j S r. )�pid��p�sr! r! r" r1 � � zIOTopUI.<lambda>Fc C � | � � S r. )�ioprio_sort_keyr\ r! r! r" r1 � � c C r` r. )�get_userr\ r! r! r" r1 � rb c C � |j S r. )r: r\ r! r! r" r1 � r_ Tc C s |j |j S r. )r; r< r\ r! r! r" r1 � s c C rd r. )r9 r\ r! r! r" r1 � r_ c C s |j p t|jp |j �S r. )r8 r r: r; r\ r! r! r" r1 � s c C r` r. )� get_cmdliner\ r! r! r" r1 � rb c C sv || _ || _d| _tj| j d | _| jjs9|| _| �� zt �
� t � � t � d� W d S t j
y8 Y d S w d S )N� r r )� process_listr% � sorting_keyrY � sorting_keys�sorting_reverse�batch�win�resize�curses�use_default_colors� start_color�curs_set�error)�selfrl rg r% r! r! r" �__init__� s ��zIOTopUI.__init__c C s | j �� \| _| _d S r. )rl �getmaxyx�height�width�rs r! r! r" rm � r- zIOTopUI.resizec
C sj d}t �� }| jjs|�tj�� t jt j B � | jj
d u s#|| jj
k r�| j � � \}}| �
|dk||| j j� | jj
d urI|d7 }|| jj
krHd S n|dkrOd}z |�| jjd �}W n" t jy| } z|jrq|jd tjkrqg }n� W Y d }~nd }~ww |D ]\}}|t jt jB @ r�t�d� q| jjs�| �� |r�| j�� } | �| � | jj
d u s#|| jj
k s#d S d S )Nr r g @�@)�select�pollr% rk �register�sys�stdin�fileno�POLLIN�POLLPRI�
iterationsrg �refresh_processes�refresh_displayr( �
delay_secondsrr �args�errno�EINTR�POLLERR�POLLHUP�exitrm rl �getch�
handle_key)
rs r� rz �total�current�eventsrL �fd�event�keyr! r! r" �run� sJ � � ���
�
�z IOTopUI.runc C s | j | _ d S r. )rj rx r! r! r" �reverse_sorting� � zIOTopUI.reverse_sortingc C s6 | j }| �|�| _ || j krtj| j d | _d S d S �Nr )rh �get_sorting_keyrY ri rj )rs �delta�orig_sorting_keyr! r! r" �adjust_sorting_key� s
�zIOTopUI.adjust_sorting_keyc C s0 | j }||7 }td|�}tttj�d |�}|S )Nr r )rh r= r* rF rY ri )rs r� �new_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�
� |dvr?|S |rDt� �|S )Nr r � )Nr )rl �hline�ordrn �A_NORMALrw �addstr�A_BOLD�refresh�echorq �getstrrF �noechorO �rs �prompt�default�empty_is_cancel�inpr! r! r" �
prompt_str� s
zIOTopUI.prompt_strc C s. | � |||�}zt|�W S ty t� �w r. )r� r �
ValueErrorrU r� r! r! r" �
prompt_int� s
�zIOTopUI.prompt_intc C �0 z| � d�W S ty t� � ty � w )NzPID to ionice: )r� rU rV rO rx r! r! r" �
prompt_pid� � �zIOTopUI.prompt_pidc C r� )NzTID to ionice: )r� rU rW rO rx r! r! r" �
prompt_tid r� zIOTopUI.prompt_tidc C s^ z|d ur| � d| |d�}n| � dd d�}W n
ty! t� �w |dk s*|dkr-t� �|S )Nz'I/O priority data (0-7, currently %s): FzI/O priority data (0-7): r � )r� rU rX )rs � ioprio_datar� r! r! r" � prompt_data
s ��� �zIOTopUI.prompt_datac C s\ z|� |�}W n ty d}Y nw t|�d } | j�ddtd�tjB | j� | j� dd|tj
� t|�}t |�D ]"\}}d| } ||u rKtj }
ntj}
| j� d|| |
� |t| �7 }q; | j�
� } | tjtd�fv rv|dkrv|d8 }n7| tjtd�fv r�||k r�|d7 }n%| tjtd �td
�fv r�|| S | d tjtjtjtd �td
�fv r�t� �q_q)N���r Tr r� z %s �l�r�
�
� �q�Q)�indexr� rF rl r� r� rn r� rw �insstrr� � enumerate� A_REVERSEr� �KEY_LEFT� KEY_RIGHT� KEY_ENTER�
KEY_CANCEL� KEY_CLOSE�KEY_EXITrO ) rs r� � display_list�ret_list�selected�set_len�offset�i�item�display�attrr� r! r! r" �
prompt_set sD �
���zIOTopUI.prompt_setc C s4 d}g d�}g d�}|d u rd}| � ||||�}|S )NzI/O priority class: )z Real-timez Best-effort�Idle)�rt�be�idler )r� )rs � ioprio_classr� �classes_prompt� 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 ) rl r� r� rn r� rw r� r� r� r5 �sleep)rs rr r! r! r" � prompt_errorE s
zIOTopUI.prompt_errorc C s, | j �ddtd�tjB | j� | j �� d S )Nr r r� )rl r� r� rn r� rw r� rx r! r! r" � prompt_clearK s zIOTopUI.prompt_clearc s � fdd�}� fdd�}� fdd�}� fdd�}i 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 ��}|�|d d � �}|� d S ) Nc � � j jdN _d S �NT)r% r3 r! rx r! r" �toggle_accumulatedP � z.IOTopUI.handle_key.<locals>.toggle_accumulatedc r� r� )r% �onlyr! rx r! r" �toggle_only_ioS r� z*IOTopUI.handle_key.<locals>.toggle_only_ioc s( � j jdN _� j�� � j�� d S r� )r% � processesrg �clearr� r! rx r! r" �toggle_processesV s
z,IOTopUI.handle_key.<locals>.toggle_processesc
s^ zH� j jr� �� } � j�| �}n � �� }t|� jj�}|�� }t �
|�\}}� � |�}|dkr2d}n� � |�}|�
||� � j�� � j�� W n` tyc } z� �d|j � W Y d }~d S d }~w tyq � �d� Y d S ty � �d� Y d S ty� � �d� Y d S ty� � �d� Y d S ty� � �� Y d S w � �� 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� rg � get_processr� r �taskstats_connection�
get_ioprior �to_class_and_datar� r� �
set_ioprior� r� r r� �errrV rW rX rU rO r� )r[ � exec_unit�tid� ioprio_valuer� r� rL rx r! r" �ionice[ sB ��
� � z"IOTopUI.handle_key.<locals>.ionicer� c S �
t �d�S �Nr �r| r� r! r! r! r" r1 � r2 z$IOTopUI.handle_key.<locals>.<lambda>r� c S r� r� r� r! r! r! r" r1 � r2 r� c � � � � S r. �r� r! rx r! r" r1 � rb �Rc r� r. r� r! rx r! r" r1 � rb �a�A�o�Or] r r� �Ic �
� � d�S )Nr� �r� r! rx r! r" r1 � r2 c r� r� r� r! rx r! r" r1 � r2 c s � � ttj� �S r. �r� rF rY ri r! rx r! r" r1 � s c s � � ttj��S r. r� r! rx r! r" r1 � s c S s d S r. r! r! r! r! r" r1 � s )r� rn r� r� �KEY_HOME�KEY_END�get)rs r� r� r� r� r� � key_bindings�actionr! rx r" r� O sn #��
�
� � �
������
�
�
�
�#
zIOTopUI.handle_keyc s� �fdd�}�fdd�}t t|�jj�� ��}tj�j d � �jj r(dd� �ndd� �|j
� �fd d��j d
� t t ||��S ) Nc
s� t � j| � jj�}|\}}}}tjrd||f }nd}dt }|d | j| �� | � � d d� |||f }| �
� } � jj sl� j t
|� }
d|
k rOt
| �k rln n|
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 �~)rD r% rg r( r �has_blkio_delay_total�
MAX_PID_WIDTHr[ r� rc re rk rw rF )
r] r? rA rB r: r; � delay_stats�
pid_format�line�cmdline�remaining_length�len1�offset2rx r! r" �format� s* � z IOTopUI.get_data.<locals>.formatc s � j j p | �� j j�S r. )r% r� � did_some_ior3 �r] rx r! r" �
should_format� s
�z'IOTopUI.get_data.<locals>.should_formatr c S rZ r. )r4 r r! r! r" r1 � r_ z"IOTopUI.get_data.<locals>.<lambda>c S rZ r. )r7 r r! r! r" r1 � r_ c s � | �| ��S r. r! r )r� � stats_lambdar! r" r1 � s )r� �reverse)
�list�filterrg r� �valuesrY ri rh r% r3 �sortrj �map)rs r r r� r! )r� rs 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 }�jjrF|d7 }n|d 7 }|d
d d d
dddg}��� }�jjrxdg| }t�d�� � fdd�|D �}� fdd�|D �}�jj r��jj
dkr�|D ]} t | � q��jj
t |�kr�t d�
|�� |D ]}
t |
� q�tj�� d S �j�� tjr�d } nd} g } g }
�jj�rw��d�}��d�}tddd d d �jjr�d nd fd!�jjr�d"nd#fd$�jjr�d%nd&fd'd(�jr�d)nd*fd+|| �� fd,|| �� fd-|d �� fd.|d �� fg
�}d}|� � D ]^\}}|�r4d/|d0|g}d|d1k�r-dnt j ddg}n d/|d2g}g d3�}| �rU|�rUt d�
| | �d�
|� ��j k�re| �!|� |
�!|� |d7 }�q| | |7 < |
| |7 < �qt |��t t"|���t t"| ��}t | �}�j#� � | | }|d4k �r�g }d�|d5k �r�g }d�|d6k �r�d } d}|d7k �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 ]J�t j)}|� } ��j+k�r| dd � } ��j+k�r-|t j,O }| �j�r*d8�p+d97 } | d |� } |t | �8 }| �rE�j�&�|| |� |t | �7 }�qt*|�D ]6�z(����fd:d;�} z| |� � W n t-�yx | |� �.d<�� Y nw W �qQ t j/�y� Y �qQw t*|�D ]G} �j#| | | } �j�'| dt(d�t j)B �j � d}t*t | | ��D ]!��j�0| || | � t j)|
| � B � |t | | � �7 }�q��q�| �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� �PID�TIDz PRIOz USERz DISK READz DISK WRITEz SWAPINz IOz COMMANDz TIMEz %H:%M:%S c � g | ]}� | �qS r! r! )�.0r� �� current_timer! r" �
<listcomp>� � z+IOTopUI.refresh_display.<locals>.<listcomp>c r r! r! )r r^ r r! r" r" � r# r r zNCONFIG_TASK_DELAY_ACCT not enabled in kernel, cannot determine SWAPIN and IO %r� )�keysr )�anyr� )r� �quit)r� r� r� �all�activer] �threads�procsr� � bandwidth�accum)r r r� �asc�desc�left�right�home�endz z: r% �:)r r r rE rf r� r �>�<c s �j �� � � d| � d S r� )rl r� )r )r� � len_summary�
len_titlesrs r! r" �
print_lineB s z+IOTopUI.refresh_display.<locals>.print_linezutf-8)2r) r% �rjustr= r r� r r5 �strftimerk �quietrK r �joinr| �stdout�flushrl �eraser r �helpr� r r� r3 rj rI �itemsrn � A_UNDERLINErF rw �append�boolrv r* r� r� r� r� r� �rangerh r� �UnicodeEncodeError�encoderr r� r� ) rs �
first_timer� r� r( �summaryr[ �titles�linesr^ r� �
status_msg�
help_lines�
help_attrs�prev�nextr@ � help_liner� � help_item� help_attr�len_status_msg�len_help� max_lines� num_lines�pos�remaining_colsr� �titler8 �lnr r! )r! r� r6 r7 rs r" r� � s ���
�
�
0
��� (��zIOTopUI.refresh_displayr� r. )r� )rQ rR rS ri rt rm r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r! r! r! r" rY � s<
�
�!
!
U 'rY c C sf |j r t�tjtj� ndd� }t�tj|� t�tj|� t|�}t||�}t| ||�}|� � d S )Nc _ s t �d� d S r� r� )r� �kwargsr! r! r" �
clean_exit] r� z$run_iotop_window.<locals>.clean_exit)
rk �signal�SIGPIPE�SIG_DFL�SIGINT�SIGTERMr r rY r� )rl r% r] r� rg �uir! r! r" �run_iotop_windowY s
rd c
C s~ z| j r
td | �W S t�t| �W S ty> } z"|jtjkr2t|tj d� tdtj d� t�
d� n� W Y d }~d S d }~ww )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 ) rk rd rn �wrapper�OSErrorr� �EPERMrK r| �stderrr� )r% rL r! r! r" � run_iotopg s � ���rj c C s� d}z dd l }dd l}td� |�dt� t� |� |�|�}W n+ tyJ dd l}dd l }|j
|dd�}td� |� | � |� � |j
�|�}Y nw |�� |�dd � |�d
� |�d
� t�|� d S ) Nz
iotop.profr zProfiling using cProfilezcontinuation()r )�
lineeventszProfiling using hotshotr5 �calls�2 )�cProfile�pstatsrK �runctx�globals�localsr � ImportError�hotshot�
hotshot.stats�Profile�runcall�closer? �load�
strip_dirs�
sort_stats� print_stats�
print_callees�os�remove)� continuation� prof_filern ro r? rt �profr! 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� z t �t jd� W n t jy td� Y nw 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�� �jp�g �_�jpԈjpԈj�_�fdLdM�� �jr� fdNdO�}t|� d S � � d S )PNr 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� r@ z-bz--batchrk znon-interactive mode)r r� r@ z-nz--iterr r� �NUMz-number of iterations before ending [infinite])�typer� �metavarr@ z-dz--delayr r� z#delay between iterations [1 second]�SECr )r� r� r@ r� r� z-pz--pid�pidsrC z"processes/threads to monitor [all]r )r� r� r r@ r� z-uz--user�str�userszusers to monitor [all]�USERz-Pz --processesr� z$only show processes, not all threadsz-az
--accumulatedr3 z)show accumulated I/O instead of bandwidthz-kz --kilobytesr$ z.use kilobytes instead of a human friendly unitz-tz--timer5 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-help� store_falser@ Tz suppress listing of shortcutszUnexpected arguments: r� c s t � �S r. )rj r! r0 r! r" r1 � rb zmain.<locals>.<lambda>c s" z� � W d S t y Y d S w r. )rJ r! )� main_loopr! r" �safe_main_loop� s
�z main.<locals>.safe_main_loop)�locale� setlocale�LC_ALL�ErrorrK �optparse� OptionParser�USAGEr
�
add_option�
SUPPRESS_HELP�
parse_argsrr r< r r� rk r5 r; r� r� )�parserr� r� r! )r� r% r" �main� s~ �
� � � �� �
�
�
� ��
� �
r� )3�
__future__r rn r� r� r r� r~ ry r^ r| r5 �hasattrr rs � RuntimeError� collectionsr �
iotop.datar r r r r �
iotop.versionr
�iotopr � iotop.ioprior r r# r' r) rD rM r rJ rO rU rV rW rX �objectrY rd rj r� �argvr� r� r! r! r! r" �<module> s`
� � L
�