| Current File : //lib/python3.6/site-packages/iotop/__pycache__/ioprio.cpython-36.pyc |
3
B��Qw � @ s� d dl Z d dlZd dlZd dlZdBdCdDdEdFdGdHdIdJdKdLg ZdMdNdOdPdQdRdSdTdUdVdWg Zd"d#� ZG d$d%� d%e�Zee�Z ee�Z
ye j dd&d'�Z W n e
k
� r� e j d�Z Y nX e jZdd(d)d*gZd+Zd,Zd+e> d+ Zd-d.� Zd/d0� Zd1d2� Ze jZed3�\ZZZZZ Z e j Z d Z d4d5� Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$e%d>k�r�d dl&Z&e'e&j(�d?k�rle)e&j(d+ �Z*nej+� Z*e,d@e*� e,dAe!e*�� dS )X� N�alpha�*� �arm*�; �i*86�" �ia64*� �parisc*� �powerpc*� �s390*� �sparc*�� �sh*�! �x86_64*�32bit�64bit� � �: � � � � �� � � c C sL t j� d }tj� d }x.| D ]&\}}}tj||�r tj||�r |S q W d S )N� r )�os�uname�platformZ architecture�fnmatch)Z syscall_listZarch�bitsZcandidate_archZcandidate_bitsZ
syscall_nr� r( � /usr/lib/python3.6/ioprio.py�find_ioprio_syscall_number9 s r* c @ s e Zd Zdd� ZdS )�IoprioSetErrorc C s0 yt j|�| _W n tk
r* || _Y nX d S )N)r# �strerror�err� TypeError)�selfr- r( r( r) �__init__C s zIoprioSetError.__init__N)�__name__�
__module__� __qualname__r0 r( r( r( r) r+ B s r+ T)� use_errnoZrtZbe�idle� �
c C s: yt j| �} W n tk
r, t jd �} Y nX | t> |B S )N)�PRIORITY_CLASSES�index�
ValueError�IOPRIO_CLASS_SHIFT)� ioprio_class� ioprio_datar( r( r) � ioprio_valueY s
r> c C s t | t? S )N)r8 r; )�ioprior( r( r) r< ` s r< c C s | t @ S )N)�IOPRIO_PRIO_MASK)r? r( r( r) r= c s r= � c C sJ t | �}tt| �}|d d }|ttfkr2d| S |tkr>dS d| S d S )N� � zrt/%dr5 zbe/%d)�sched_getscheduler� getpriority� PRIO_PROCESS�
SCHED_FIFO�SCHED_RR�
SCHED_IDLE)�pidZ scheduler�niceZ ioprio_nicer( r( r) �get_ioprio_from_schedl s
rL c C sT t d kr dS tt t| �}|dk r$dS t|�}|s8t| �S |dkrD|S d|t|�f S )Nz?sysr z?errr5 z%s/%d)�__NR_ioprio_get�syscall�IOPRIO_WHO_PROCESSr< rL r= )rJ r? Z
prio_classr( r( r) �getx s rP c C sf t d krtd��t||�}tt | ||dd�}|dk rby tj� }W n tk
rX d}Y nX t|��d S )NzNo ioprio_set syscall foundT)r4 r z<Unknown error (errno support not available before Python2.6))�__NR_ioprio_setr+ r> rN �ctypesZ get_errno�AttributeError)ZwhichZwhor< r= Z
ioprio_val�retr- r( r( r) �
set_ioprio� s
rU c C st | d dkrt | d � S d| krT| jd�r2d}n| jd�r@d}t| jd�d �}n| dkrdd}d}d|t > | S ) Nr �?r6 �/zrt/zbe/r5 � )�ord�
startswith�int�splitr; )�keyZshiftZprior( r( r) �sort_key� s
r^ c C s6 d| kr&| j d�}|d t|d �fS | dkr2dS dS )NrW r r6 r5 )r5 r )NN)r\ r[ )Z
ioprio_strr\ r( r( r) �to_class_and_data� s
r_ �__main__rX zpid:zioprio:)r r r )r r r )r r r )r r r
)r r r )r
r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r
r r )r r r )r r r )r r r )r r r )r r r! )-rR r& r# r% ZIOPRIO_GET_ARCH_SYSCALLZIOPRIO_SET_ARCH_SYSCALLr* � Exceptionr+ rM rQ ZCDLLZ
ctypes_handler. rN r8 rO r; r@ r> r< r= rD �range� SCHED_OTHERrG rH � SCHED_BATCHZ SCHED_ISOrI rE rF rL rP rU r^ r_ r1 �sys�len�argvr[ rJ �getpid�printr( r( r( r) �<module> sr