3
`g@ @ sr d dl Z d dlZd dlZd dlZd dlZd dlZd dljZd dl Z d dl
T d dl m Z ej
j ZG dd dZdS ) N)*)TunedExceptionc @ s e Zd ZdbddZdd Zdd Zdd Zd
d Zd d
ZdcddZ ddddZ
dd Z dd Z dd Z
dedd Zdfd d Zdgd d!Zdhd#d$Zdid%d&Zdjd'd(Zdkd)d*Zdld+d,Zd-d. Zd/d0 Zdmd1d2Zd3d4 Zd5d6 Zdnd7d8Zd di g d fd9d:Z dod;d<Z d=d> Z d?d@ Z dAdB Z dpdDdEZ!dFdG Z"dHdI Z#dJdK Z$dLdM Z%dqdNdOZ&dPdQ Z'dRdS Z(dTdU Z)dVdW Z*dXdY Z+dZd[ Z,d\d] Z-d^d_ Z.d`da Z/dS )rcommandsTc C s
|| _ d S )N)_logging)selfZlogging r /usr/lib/python3.6/commands.py__init__ s zcommands.__init__c C s | j rtj| d S )N)r logerror)r msgr r r _error s zcommands._errorc C s | j rtj| d S )N)r r
debug)r r r r r _debug s zcommands._debugc C s. t |j j }dddddddddj||S )N10)YZYESTZTRUENZNOFZFALSE)strupperstripget)r valuevr r r get_bool s zcommands.get_boolc C s t jddt|j S )Nz\s+ )resubr r )r sr r r remove_ws s zcommands.remove_wsc C s t jdd|S )Nz^"(.*)"$z\1)r r )r r r r r unquote" s zcommands.unquote\c C s |j |d||f S )Nz%s%s)replace)r r Z what_escapeZ escape_byr r r escape& s zcommands.escapec C s |j |dS )N )r$ )r r Z escape_charr r r unescape* s zcommands.unescapec C s |d|t | | S )Nr )len)r s1poss2r r r align_str. s zcommands.align_strc C s2 g }|d k r.x t |j D ]}|t|7 }qW |S )N)sorteditemslist)r dlir r r dict2list4 s
zcommands.dict2listc C s( |d kr d S t jddjt|j S )Nz(%s)z)|()r compilejoinr/ keys)r r0 r r r re_lookup_compile< s zcommands.re_lookup_compileNr c sV d kr|d kr.|S nt dks*|d kr.|S |d kr@| j }|j fdd||S )Nr c s t j | jd S )N )r/ values lastindex)mo)r0 r r M s z.commands.multiple_re_replace..)r( r7 r )r r0 r rflagsr )r0 r multiple_re_replaceD s
z commands.multiple_re_replacec C sR t |dks|d krd S |d kr*| j|}|j|}|rNt|j |jd S d S )Nr r8 )r( r7 searchr/ r9 r: )r r0 r r= r; r r r re_lookupS s
zcommands.re_lookupFc C s | j d||f |r2tjj|}tjj|r2d}y6|rBtj| t|d}|jt| |j d}W n` t
t fk
r } z@d}t |t
r| st |tr|j|kr| jd||f W Y dd}~X nX |S )a Write data to a file.
Parameters:
f -- filename where to write
data -- data to write
makedir -- if True and the path doesn't exist, it will be created
no_error -- if True errors are silenced, it can be also list of ignored errnos
Return:
bool -- True on success
z Writing to file: '%s' < '%s'FwTz Writing to file '%s' error: '%s'N)r ospathdirnameisdirmakedirsopenwriter closeOSErrorIOError
isinstanceboolr/ errnor
) r fdataZmakedirno_errorr0 fdZrcer r r
write_to_file] s$
$zcommands.write_to_filer& c C sv |}y t |d}|j }|j W n< ttfk
r^ } z |sN| jd||f W Y d d }~X nX | jd||f |S )Nr= z"Error when reading file '%s': '%s'z Read data from file: '%s' > '%s')rH readrJ rK rL r
r )r rP Zerr_retrR Z old_valuerT r r r read_file| s
$zcommands.read_filec C sj | j d| tjj|rfytj|| W n: tk
rd } z |sVtjd|t |f dS d }~X nX dS )NzRemoving tree: '%s'z cannot remove tree '%s': '%s'FT)
r rC rD existsshutilrmtreerK r
r r )r rP rR r r r r rZ s zcommands.rmtreec C sh | j d| tjj|rdytj| W n: tk
rb } z |sTtjd|t|f dS d }~X nX dS )NzRemoving file: '%s'z cannot remove file '%s': '%s'FT) r rC rD rX unlinkrK r
r r )r rP rR r r r r r[ s zcommands.unlinkc C sd | j d||f ytj|| W n< tk
r^ } z |sPtjd||t|f dS d }~X nX dS )NzRenaming file '%s' to '%s'z%cannot rename file '%s' to '%s': '%s'FT)r rC renamerK r
r r )r srcdstrR r r r r r\ s zcommands.renamec C s` y"t jd||f tj|| dS tk
rZ } z |sLt jd|||f dS d }~X nX d S )Nzcopying file '%s' to '%s'Tz!cannot copy file '%s' to '%s': %sF)r
r rY copyrL r )r r] r^ rR rT r r r r_ s z
commands.copyc C s6 | j |}t|dkrdS | j|tj|||tjdS )Nr F)r> )rW r( rU r r MULTILINE)r rP patternreplrQ r r r replace_in_file s
zcommands.replace_in_filec C s4 | j |}t|dkrdS | j|| j||tjdS )Nr F)r> )rW r( rU r? r r` )r rP r0 rQ r r r multiple_replace_in_file s
z!commands.multiple_replace_in_filec C s | j |}x|D ]}t|}t|| }tjd| d |tjdd kr||rt|dkrj|d dkrj|d7 }|d||f 7 }qtjd| d d
| j| d |tjd}qW | j||S )Nz\bz\s*=.*$)r> r r8
z%s="%s"
z\b(z\s*=).*$z\1"z\1") rW r r r@ r` r( r r% rU )r rP r0 addrQ optor r r r add_modify_option_in_file s
.z"commands.add_modify_option_in_filec C s" | j |}tjt|jdj S )Nzutf-8)rW hashlibZmd5r encode hexdigest)r rP rQ r r r md5sum s
zcommands.md5sumc C s" | j |}tjt|jdj S )Nzutf-8)rW rl Zsha256r rm rn )r rP rQ r r r sha256sum s
zcommands.sha256sumc C s | j tj|j S )N)rW constsZMACHINE_ID_FILEr )r rR r r r get_machine_id s zcommands.get_machine_idc C sV d}t jj }d|d< |j| | jdt| d} d }
yt|tt|||ddd} | j \} } | j }|r||krd|kr| d d }
t
|
dkr| d d }
d d
j ||
f }
|s| j |
W nx t
tfk
r8 } zV|jd k r|j nd
}t||kr(d|kr(d d
j ||f }
|s(| j |
W Y d d }~X nX |rJ|| |
fS || fS d S )Nr CLC_ALLz
Executing %s.r& T)stdoutstderrenvshellcwdZ close_fdsZuniversal_newlinesr8 zExecuting '%s' error: %sr rg rg rg )rC environr_ updater r PopenPIPEZ communicate
returncoder( r5 r
rK rL rO abs)r argsrx ry rw Z no_errorsZ
return_errZretcodeZ _environmentoutZerr_msgprocerrZerr_outrT r r r execute s>
zcommands.executec C s. t jd|}|r|jdS |r*|j d S |S )Nz.*\[([^\]]+)\].*r8 r )r matchgroupsplit)r ZoptionsZdosplitmr r r get_active_option s
zcommands.get_active_optionc C s* t |}|dkp(| jd| ddj dkS )Nr z$/sys/devices/system/cpu/cpu%s/onlineT)rR r )r rW r )r cpuZscpur r r
is_cpu_online s zcommands.is_cpu_onlinec C s\ |d kr d S t |jdd}yt|d}W n& tk
rP tjdt | g S X | j|S )N,r& z invalid hexadecimal mask '%s')r r$ int
ValueErrorr
r bitmask2cpulist)r maskr r r r hex2cpulist s zcommands.hex2cpulistc C s< d}g }x.|dkr6|d@ r$|j | |dL }|d7 }q
W |S )Nr r8 )append)r r r cpusr r r r & s
zcommands.bitmask2cpulist'"c C s^ g }|d kr|S |}t |tk rD|d k r6t|j|}t|jd}g }g }d}d}x|D ]} t| }
|rt|
dkrd}|j| d}n||
7 }q\|
dd j dkrd}|
}q\|
o|
d dks|
d d krP|
d
d jd } yPt| d
kr |ttt | d t | d
d
7 }n|jt |
d
d W n t
k
rL g S X q\t|
dkr\|j|
q\W t|dkr|j| x|D ]} | jd } | dd j dkr|| j | 7 }ndyLt| d
kr|ttt | d t | d
d
7 }n|jt | d W n t
k
r g S X qW t tt
|}
x"|D ]}||
kr<|
j| q s z'commands.cpulist2hex..)r cpulist2bitmaskr( zfillr5 r )r r1 Zulr Zlsr )r r cpulist2hex s
zcommands.cpulist2hexc C s$ d}x|D ]}|t d|O }q
W |S )Nr r )pow)r r1 r r r r r r s
zcommands.cpulist2bitmaskc s dj fdd|D S )Nr c 3 s | ]} t | V qd S )N)r )r r )prefixr r r s z*commands.cpulist2string..)r5 )r r1 r r )r r cpulist2string s zcommands.cpulist2stringc C sb |dd j dkrH|dd }|dd j dkrHdd | j|D S tjd|}dd |D S )Nr r zcpulist:c S s g | ]}d t | qS )r )r )r r r r r
s z(commands.devstr2devs..z\s*(?.)allr( rq ZNAMES_MAX_LENGTH)r namer r r
is_valid_name% s zcommands.is_valid_name)T)r# r# )r# )Nr )N)FF)r& F)F)F)F)F)T)T)T)r )r& )0__name__
__module__ __qualname__r r
r r r! r" r% r' r, r3 r7 r? rA rU rW rZ r[ r\ r_ rc rd rk ro rp rr 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 sZ
'
D
r )rO rl Z
tuned.logsZtunedr_ rC rY Z tuned.constsrq r
subprocessZtuned.exceptionsr Zlogsr r
r r r r r s