관리-도구
편집 파일: commands.cpython-36.pyc
3 �@ag�@ � @ sr d dl Z d dlZd dlZd dlZd dlZd dlZd dljZd dl Z d dl T d dl m Z ej j� ZG dd� d�ZdS )� N)�*)�TunedExceptionc @ s� e Zd Zdbdd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdcdd�Z dddd�Z 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� Zdmd1d2�Zd3d4� Zd5d6� Zdnd7d8�Zd di g d fd9d:�Z dod;d<�Z d=d>� Z d?d@� Z dAdB� Z dpdDdE�Z!dFdG� Z"dHdI� Z#dJdK� Z$dLdM� Z%dqdNdO�Z&dPdQ� Z'dRdS� Z(dTdU� Z)dVdW� Z*dXdY� Z+dZd[� Z,d\d]� Z-d^d_� Z.d`da� Z/dS )r�commandsTc 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 �log�error)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� }ddddddddd�j||�S )N�1�0)�YZYES�TZTRUE�NZNO�FZFALSE)�str�upper�strip�get)r �value�vr r r �get_bool s zcommands.get_boolc C s t jddt|��j� S )Nz\s+� )�re�subr 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 |d�S )N� )r$ )r r Z escape_charr r r �unescape* s zcommands.unescapec C s |d|t |� | S )Nr )�len)r �s1�pos�s2r r r � align_str. s zcommands.align_strc C s2 g }|d k r.x t |j� �D ]}|t|�7 }qW |S )N)�sorted�items�list)r �d�l�ir r r � dict2list4 s zcommands.dict2listc C s( |d kr d S t jddjt|j� �� �S )Nz(%s)z)|()r �compile�joinr/ �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 �<lambda>M s z.commands.multiple_re_replace.<locals>.<lambda>)r( r7 r )r r0 r �r�flagsr )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�| s�t |t�r�|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'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�er 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 �exists�shutil�rmtreerK 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 �src�dstrR 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|||tjd��S )Nr F)r> )rW r( rU r r � MULTILINE)r rP �pattern�replrQ r r r �replace_in_file� s zcommands.replace_in_filec C s4 | j |�}t|�dkrdS | j|| j||tjd��S )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 |tjd�d kr||r�t|�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 �opt�or r r r �add_modify_option_in_file� s .z"commands.add_modify_option_in_filec C s" | j |�}tjt|�jd��j� S )Nzutf-8)rW �hashlibZmd5r �encode� hexdigest)r rP rQ r r r �md5sum� s zcommands.md5sumc C s" | j |�}tjt|�jd��j� 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 } y�t|tt|||ddd�} | j� \} } | j }|r�||kr�d|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|�|k�r(d|k�r(d d j |�|f } |�s(| j | � W Y d d }~X nX |�rJ|| | fS || fS d S )Nr �C�LC_ALLz Executing %s.r& T)�stdout�stderr�env�shell�cwdZ close_fdsZuniversal_newlinesr8 zExecuting '%s' error: %sr rg rg rg )rC �environr_ �updater r �Popen�PIPEZ communicate� returncoder( r5 r rK rL rO �abs)r �argsrx ry rw Z no_errorsZ return_errZretcodeZ _environment�outZerr_msg�proc�errZerr_outrT r r r �execute� s> zcommands.executec C s. t jd|�}|r|jd�S |r*|j� d S |S )Nz.*\[([^\]]+)\].*r8 r )r �match�group�split)r ZoptionsZdosplit�mr r r �get_active_option s zcommands.get_active_optionc C s* t |�}|dkp(| jd| dd�j� 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| �} |r�t| �dkr�d}|j|� d}n|| 7 }q\| dd� j� dkr�d}| }q\| o�| d dks�| d d k�rP| d d � jd �} yPt| �d k�r |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|�dk�r�|j|� x�|D ]�} | jd �} | dd� j� dk�r�|| j | �7 }ndyLt| �d k�r�|ttt | d �t | d �d ��7 }n|jt | d �� W n t k �r g S X �q�W t tt |���} x"|D ]}|| k�r<| j|� �q<W | S ) Nr� Fr& r � Z0xT�^�!r8 �-)�typer/ r r r� r( r� �lower�ranger� r� r� r- �set�remove)r r1 Z strip_chars�rlZllZll2Z negation_listZhexmaskZhvr Zsv�nlZvlZcpu_listr� r r r �cpulist_unpack9 sh ( zcommands.cpulist_unpackc C s� | j |�}|d ks t|�dkr"|S d}|}g }xz|d t|�k r�||d || dkr�||kr�|jt|| �d t|| � � n|jt|| �� |d }|d7 }q0W |d t|�k r�|jt|| �d t|d � � n|jt|d �� |S )Nr r8 r� rg rg )r� r( r� r )r r1 r2 �jr� r r r � cpulist_pack} s" $ $zcommands.cpulist_packc C s. | j |�}| j | jd��}tt|�t|� �S )Nz /sys/devices/system/cpu/online)r� rW r/ r� )r r1 r� Zonliner r r �cpulist_invert� s zcommands.cpulist_invertc s� |d kr d S | j |�}|d kr"d S | j|�}d| � t� �}|d dkrX|d|d 7 }� j|�� dj� fdd�tdt� �d�D ��S )Nz%x� r r� c 3 s | ]}� ||d � V qdS )r� Nr )�.0r2 )r r r � <genexpr>� s z'commands.cpulist2hex.<locals>.<genexpr>)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.<locals>.<genexpr>)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 � <listcomp>� s z(commands.devstr2devs.<locals>.<listcomp>z\s*(?<!\\),\s*c S s g | ]}t |�jd d��qS )z\,r� )r r$ )r� r r r r r� � s )r� r� r r� )r r r1 r r r � devstr2devs� s zcommands.devstr2devsc C s" y |j |d�S |j |�S d S )NF)�wait)r Z terminateZtimer r r r� � s z commands.waitc C s� t |�j� j� }x�dD ]�}|j|�}|dkr.qybt|d |� �}||d � }|dkr^|d9 }n0|dkrp|d 9 }n |dkr�|d 9 }n |dkr�d }|S tk r� d S X qW d S ) N�KB�MB�GBr& r8 i )r� r� r� r& rg i i i @)r r r �rfindr� r� )r r ZunitZunit_ix�val�ur r r �get_size� s& zcommands.get_sizec C s� d}d}y(t tjd��}|j� j� }W d Q R X W nj tk rj } z |jtjkrZtd| ��W Y d d }~X n2 t t fk r� } ztd| ��W Y d d }~X nX yHt tj d��2}|j� j� }|dtj tj gkr�tdtj ��W d Q R X W np tk �r" } z |jtjk�rtd| ��W Y d d }~X n4 t t fk �rT } ztd| ��W Y d d }~X nX |dk�rfd }n |tj k}|dk�r~d }||fS )Nr& r= z!Failed to read active profile: %szInvalid value in file %s.z Failed to read profile mode: %s)rH rq �ACTIVE_PROFILE_FILErV r rL rO �ENOENTr rK �EOFError�PROFILE_MODE_FILE�ACTIVE_PROFILE_AUTO�ACTIVE_PROFILE_MANUAL)r � profile_name�moderP rT �manualr r r �get_active_profile� s4 zcommands.get_active_profilec C s� y2t tjd�� }|d k r&|j|d � W d Q R X W n4 ttfk rf } ztd|j ��W Y d d }~X nX y:t tjd��$}|r�tj ntj }|j|d � W d Q R X W n4 ttfk r� } ztd|j ��W Y d d }~X nX d S )NrB re z!Failed to save active profile: %sz Failed to save profile mode: %s) rH rq r� rI rK rL r �strerrorr� r� r� )r r� r� rP rT r� r r r �save_active_profile� s z commands.save_active_profilec C s� d}y(t tjd��}|j� j� }W d Q R X W nj tk rf } z |jtjkrVtd| ��W Y d d }~X n2 t t fk r� } ztd| ��W Y d d }~X nX |dkr�d }|S )Nr& r= z1Failed to read the active post-loaded profile: %s) rH rq �POST_LOADED_PROFILE_FILErV r rL rO r� r rK r� )r r� rP rT r r r �get_post_loaded_profile s z commands.get_post_loaded_profilec C sl y2t tjd�� }|d k r&|j|d � W d Q R X W n4 ttfk rf } ztd|j ��W Y d d }~X nX d S )NrB re z1Failed to save the active post-loaded profile: %s)rH rq r� rI rK rL r r� )r r� rP rT r r r �save_post_loaded_profile s z!commands.save_post_loaded_profilec C sD yt j||�}W n( tk r8 dd l}|j||�}Y nX |j|�S )Nr )r � maketrans�AttributeError�string� translate)r �textZ source_charsZ dest_charsZtransr� r r r �tr s z commands.trc C s& | p$t dd� |D ��o$t|�tjkS )Nc s s | ]}|t jkV qd S )N)rq ZNAMES_ALLOWED_CHARS)r� �cr r r r� & s z)commands.is_valid_name.<locals>.<genexpr>)�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 �<module> s