3 À<×eù?ã@srddlZddlZddlZddlZddlZddlZddljZddl Z ddl TddlmZej jƒZGdd„dƒZdS)éN)Ú*)ÚTunedExceptionc@s˜eZdZd`dd„Zdd„Zdd„Zdd „Zd d„Zdd „Zdadd„Z dbdd„Z dd„Zdd„Zdd„Z dcdd„Zdddd„Zded d!„Zdfd#d$„Zdgd%d&„Zdhd'd(„Zdid)d*„Zdjd+d,„Zd-d.„Zd/d0„Zdkd1d2„Zd3d4„Zd5d6„Zdld7d8„Zddigdfd9d:„Zdmd;d<„Zd=d>„Zd?d@„ZdAdB„Z dndDdE„Z!dFdG„Z"dHdI„Z#dJdK„Z$dLdM„Z%dodNdO„Z&dPdQ„Z'dRdS„Z(dTdU„Z)dVdW„Z*dXdY„Z+dZd[„Z,d\d]„Z-d^d_„Z.dS)pÚcommandsTcCs ||_dS)N)Ú_logging)ÚselfZlogging©rú/usr/lib/python3.6/commands.pyÚ__init__szcommands.__init__cCs|jrtj|ƒdS)N)rÚlogÚerror)rÚmsgrrrÚ_errorszcommands._errorcCs|jrtj|ƒdS)N)rr Údebug)rrrrrÚ_debugszcommands._debugc Cs.t|ƒjƒjƒ}dddddddddœj||ƒS)NÚ1Ú0)ÚYZYESÚTZTRUEÚNZNOÚFZFALSE)ÚstrÚupperÚstripÚget)rÚvalueÚvrrrÚget_boolszcommands.get_boolcCstjddt|ƒƒjƒS)Nz\s+ú )ÚreÚsubrr)rÚsrrrÚ remove_wsszcommands.remove_wscCstjdd|ƒS)Nz^"(.*)"$z\1)rr)rrrrrÚunquote"szcommands.unquoteú\cCs|j|d||fƒS)Nz%s%s)Úreplace)rr Zwhat_escapeZ escape_byrrrÚescape&szcommands.escapecCs|j|dƒS)NÚ)r$)rr Zescape_charrrrÚunescape*szcommands.unescapecCs|d|t|ƒ|S)Nr)Úlen)rÚs1ÚposÚs2rrrÚ align_str.szcommands.align_strcCs2g}|dk r.x t|jƒƒD]}|t|ƒ7}qW|S)N)ÚsortedÚitemsÚlist)rÚdÚlÚirrrÚ dict2list4s zcommands.dict2listcCs(|dkrdStjddjt|jƒƒƒƒS)Nz(%s)z)|()rÚcompileÚjoinr/Úkeys)rr0rrrÚre_lookup_compile<szcommands.re_lookup_compileNrcsVˆdkr|dkr.|Sntˆƒdks*|dkr.|S|dkr@|jˆƒ}|j‡fdd„||ƒS)Nrcstˆjƒƒ|jdS)Né)r/ÚvaluesÚ lastindex)Úmo)r0rrÚMsz.commands.multiple_re_replace..)r(r7r)rr0r ÚrÚflagsr)r0rÚmultiple_re_replaceDs zcommands.multiple_re_replacecCsRt|ƒdks|dkrdS|dkr*|j|ƒ}|j|ƒ}|rNt|jƒƒ|jdSdS)Nrr8)r(r7Úsearchr/r9r:)rr0r r=r;rrrÚ re_lookupSs zcommands.re_lookupFc CsÎ|jd||fƒ|r2tjj|ƒ}tjj|ƒr2d}y6|rBtj|ƒt|dƒ}|jt|ƒƒ|j ƒd}Wn`t tfk rÈ}z@d}t|t ƒr’|s¦t|tƒr¸|j|kr¸|jd||fƒWYdd}~XnX|S)aWrite 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 zWriting to file: '%s' < '%s'FÚwTz Writing to file '%s' error: '%s'N)rÚosÚpathÚdirnameÚisdirÚmakedirsÚopenÚwriterÚcloseÚOSErrorÚIOErrorÚ isinstanceÚboolr/Úerrnor ) rÚfÚdataZmakedirÚno_errorr0ÚfdZrcÚerrrÚ write_to_file]s$ $zcommands.write_to_filer&cCsv|}yt|dƒ}|jƒ}|jƒWn<ttfk r^}z|sN|jd||fƒWYdd}~XnX|jd||fƒ|S)Nr=z"Error when reading file '%s': '%s'z Read data from file: '%s' > '%s')rHÚreadrJrKrLr r)rrPZerr_retrRZ old_valuerTrrrÚ read_file|s $zcommands.read_filecCsj|jd|ƒtjj|ƒrfytj||ƒWn:tk rd}z|sVtjd|t |ƒfƒdSd}~XnXdS)NzRemoving tree: '%s'zcannot remove tree '%s': '%s'FT) rrCrDÚexistsÚshutilÚrmtreerKr rr)rrPrRrrrrrZˆszcommands.rmtreecCsh|jd|ƒtjj|ƒrdytj|ƒWn:tk rb}z|sTtjd|t|ƒfƒdSd}~XnXdS)NzRemoving file: '%s'zcannot remove file '%s': '%s'FT) rrCrDrXÚunlinkrKr rr)rrPrRrrrrr[“szcommands.unlinkcCsd|jd||fƒytj||ƒWn<tk r^}z |sPtjd||t|ƒfƒdSd}~XnXdS)NzRenaming file '%s' to '%s'z%cannot rename file '%s' to '%s': '%s'FT)rrCÚrenamerKr rr)rÚsrcÚdstrRrrrrr\žszcommands.renamecCs`y"tjd||fƒtj||ƒdStk rZ}z|sLtjd|||fƒdSd}~XnXdS)Nzcopying file '%s' to '%s'Tz!cannot copy file '%s' to '%s': %sF)r rrYÚcopyrLr)rr]r^rRrTrrrr_¨sz commands.copycCs6|j|ƒ}t|ƒdkrdS|j|tj|||tjdƒS)NrF)r>)rWr(rUrrÚ MULTILINE)rrPÚpatternÚreplrQrrrÚreplace_in_file²s zcommands.replace_in_filecCs4|j|ƒ}t|ƒdkrdS|j||j||tjdƒS)NrF)r>)rWr(rUr?rr`)rrPr0rQrrrÚmultiple_replace_in_fileºs z!commands.multiple_replace_in_filecCs¶|j|ƒ}xž|D]–}t|ƒ}t||ƒ}tjd|d|tjddkr||r¦t|ƒdkrj|ddkrj|d7}|d||f7}qtjd|d d |j|ƒd|tjd}qW|j||ƒS)Nz\bz\s*=.*$)r>rr8Ú z%s="%s" z\b(z\s*=).*$z\1ú"éÿÿÿÿz\1") rWrrr@r`r(rr%rU)rrPr0ÚaddrQÚoptÚorrrrÚadd_modify_option_in_fileÄs .z"commands.add_modify_option_in_filecCs"|j|ƒ}tjt|ƒjdƒƒjƒS)Nzutf-8)rWÚhashlibZmd5rÚencodeÚ hexdigest)rrPrQrrrÚmd5sumÔs zcommands.md5sumcCs"|j|ƒ}tjt|ƒjdƒƒjƒS)Nzutf-8)rWrlZsha256rrmrn)rrPrQrrrÚ sha256sumÙs zcommands.sha256sumcCs|jtj|ƒjƒS)N)rWÚconstsZMACHINE_ID_FILEr)rrRrrrÚget_machine_idÞszcommands.get_machine_idcCsVd}tjjƒ}d|d<|j|ƒ|jdt|ƒƒd} d} y†t|tt|||ddd}|jƒ\} }|j }|r¾||kr¾d|kr¾|dd…} t | ƒdkrž| dd…} d d j|ƒ| f} |s¾|j| ƒWnxt tfk r8}zV|jdk rê|jnd }t|ƒ|kr(d|kr(d d j|ƒ|f} |s(|j| ƒWYdd}~XnX|rJ|| | fS|| fSdS)NrÚCÚLC_ALLz Executing %s.r&T)ÚstdoutÚstderrÚenvÚshellÚcwdZ close_fdsZuniversal_newlinesr8zExecuting '%s' error: %srrgrgrg)rCÚenvironr_ÚupdaterrÚPopenÚPIPEZcommunicateÚ returncoder(r5r rKrLrOÚabs)rÚargsrxryrwZ no_errorsZ return_errZretcodeZ_environmentÚoutZerr_msgÚprocÚerrZerr_outrTrrrÚexecuteäs> zcommands.executecCs.tjd|ƒ}|r|jdƒS|r*|jƒdS|S)Nz.*\[([^\]]+)\].*r8r)rÚmatchÚgroupÚsplit)rZoptionsZdosplitÚmrrrÚget_active_options zcommands.get_active_optioncCs*t|ƒ}|dkp(|jd|ddjƒdkS)Nrz$/sys/devices/system/cpu/cpu%s/onlineT)rRr)rrWr)rÚcpuZscpurrrÚ is_cpu_onlineszcommands.is_cpu_onlinecCs\|dkrdSt|ƒjddƒ}yt|dƒ}Wn&tk rPtjdt|ƒƒgSX|j|ƒS)Nú,r&ézinvalid hexadecimal mask '%s')rr$ÚintÚ ValueErrorr rÚbitmask2cpulist)rÚmaskrˆrrrÚhex2cpulistszcommands.hex2cpulistcCs<d}g}x.|dkr6|d@r$|j|ƒ|dL}|d7}q W|S)Nrr8)Úappend)rr‘rŠÚcpusrrrr&s zcommands.bitmask2cpulistú'"cCs^g}|dkr|S|}t|ƒtk rD|dk r6t|ƒj|ƒ}t|ƒjdƒ}g}g}d}d}x|D]} t| ƒ} |r˜t| ƒdkrŽd}|j|ƒd}n|| 7}q\| dd…jƒdkr¶d}| }q\| oÐ| ddksÔ| dd krP| d d…jdƒ}yPt|ƒd kr|ttt |dƒt |d ƒd ƒƒ7}n|jt | d d…ƒƒWnt k rLgSXq\t| ƒdkr\|j| ƒq\Wt|ƒdkr‚|j|ƒx¢|D]š} | jdƒ}| dd…jƒdkr¼||j| ƒ7}ndyLt|ƒd krô|ttt |dƒt |d ƒd ƒƒ7}n|jt |dƒƒWnt k rgSXqˆWttt |ƒƒƒ} x"|D]}|| kr<| j|ƒq¥sz'commands.cpulist2hex..)r¡Úcpulist2bitmaskr(Úzfillr5rœ)rr1ZulrˆZlsr)r rÚcpulist2hex™s zcommands.cpulist2hexcCs$d}x|D]}|td|ƒO}q W|S)Nrr–)Úpow)rr1rˆrrrrr¨§s zcommands.cpulist2bitmaskcsdj‡fdd„|DƒƒS)NrŒc3s|]}ˆt|ƒVqdS)N)r)r¦r)Úprefixrrr§®sz*commands.cpulist2string..)r5)rr1r¬r)r¬rÚcpulist2stringszcommands.cpulist2stringcCsb|dd…jƒdkrH|dd…}|dd…jƒdkrHdd„|j|ƒDƒStjd|ƒ}dd„|DƒS)Nrr¥zcpulist:cSsg|]}dt|ƒ‘qS)rŠ)r)r¦rrrrú ¸sz(commands.devstr2devs..z\s*(?s