Your IP : 216.73.216.213
B
�d{T8 � @ s d Z ddlmZ dZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZmZmZmZmZ ddlmZ ddlmZmZ ddl m Z m Z dd l m Z dd
l!m"Z" ydd l#m$Z$m%Z% W n. e&k
�r G d d
� d
e'�Z%dd� Z$Y nX G dd� de(�Z)G dd� de(�Z*d9dd�Z+ej,fdd�Z-dd� Z.dd� Z/d e fd d �Z0e fd d �Z1d!d"� Z2G d#d$� d$�Z3e"d%�e3�Z3d:d(d)�Z4d*d+� Z5G d,d-� d-e(�Z6G d.d/� d/e(�Z7e(� Z8G d0d1� d1e(�Z9G d2d3� d3e(�Z:e:� Z;d4d5� Z<d6e
j=e
j> fd7d8�Z?dS );zTshell/term utilities, useful to write some python scripts instead of shell
scripts.
� )�print_functionzrestructuredtext enN)�exists�isdir�islink�basename�join)� string_types)�range�input)�
STD_BLACKLIST�_handle_blacklist)� str_to_bytes)�
deprecated)�ProcInfo�
NoSuchProcessc @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r �J/opt/alt/python37/lib/python3.7/site-packages/logilab/common/shellutils.pyr 3 s r c C s
t � �d S )N)r )�pidr r r r 5 s r c @ s e Zd Zdd� Zdd� ZdS )�tempdirc C s t �� | _| jS )N)�tempfile�mkdtemp�path)�selfr r r � __enter__; s
ztempdir.__enter__c C s t �| j� |d kS )N)�shutil�rmtreer )r �exctype�value� tracebackr r r �__exit__? s ztempdir.__exit__N)r r r r r" r r r r r 9 s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�pushdc C s
|| _ d S )N)� directory)r r$ r r r �__init__F s zpushd.__init__c C s t �� | _t �| j� | jS )N)�os�getcwd�cwd�chdirr$ )r r r r r I s
zpushd.__enter__c C s t �| j� d S )N)r& r) r( )r r r r! r r r r" N s zpushd.__exit__N)r r r r% r r" r r r r r# E s r# c C s� |dkrd}n6y t |�}W n( tk
rB ddl}|�|�j}Y nX |dkrRd}n6y t |�}W n( tk
r� ddl}|�|�j}Y nX t� | ||� dS )z�Same as `os.chown` function but accepting user login or group name as
argument. If login or group is omitted, it's left unchanged.
Note: you must own the file to chown it (or be root). Otherwise OSError is raised.
N���r )
�int�
ValueError�pwd�getpwnam�pw_uid�grp�getgrnam�gr_gidr& �chown)r Zlogin�group�uidr- �gidr0 r r r r3 R s r3 c
C s� t � | �}t|�dkrFt|�s"t�x�|D ]}||t|t|��� q(W n�y |d } W n tk
rr td| ��Y nX t|�r�t|�r�t|t| ��}y|| |� W n4 tk
r� } ztd| ||f ��W dd}~X Y nX dS )z+A shell-like mv, supporting wildcards.
� r zNo file matching %sz Unable to move %r to %r (%s)N) �glob�lenr �AssertionErrorr r �
IndexError�OSErrorr )�source� destination�_action�sources�filename�exr r r �mvj s
rC c G sX xR| D ]J}xDt � |�D ]6}t|�r.t�|� qt|�rBt�|� qt�|� qW qW dS )z+A shell-like rm, supporting wildcards.
N)r8 r r& �remover r r )�filesZwfilerA r r r �rm s
rF c C s t | |tjd� dS )z+A shell-like cp, supporting wildcards.
)r? N)rC r �copy)r= r> r r r �cp� s rH Fc s� t �t�r�f�|r dd� �ndd� �g }xRt�| �D ]D\� }}t|||� t� �}||kr^q6|�� ��fdd�|D �� q6W |S )a� Recursively find files ending with the given extensions from the directory.
:type directory: str
:param directory:
directory where the search should start
:type exts: basestring or list or tuple
:param exts:
extensions or lists or extensions to search
:type exclude: boolean
:param exts:
if this argument is True, returning files NOT ending with the given
extensions
:type blacklist: list or tuple
:param blacklist:
optional list of files or directory to ignore, default to the value of
`logilab.common.STD_BLACKLIST`
:rtype: list
:return:
the list of all matching files
c S s x|D ]}| � |�rdS qW dS )NFT)�endswith)rA �exts�extr r r �match� s
zfind.<locals>.matchc S s x|D ]}| � |�rdS qW dS )NTF)rI )rA rJ rK r r r rL � s
c s g | ]}�|��rt � |��qS r )r )�.0�f)�dirpathrJ rL r r �
<listcomp>� s zfind.<locals>.<listcomp>)�
isinstancer r& �walkr r �extend)r$ rJ �exclude� blacklistrE �dirnames� filenames�dirnamer )rO rJ rL r �find� s
rY c c sL xFt �| �D ]8\}}}t|||� x t�||�D ]}t||�V q0W q W dS )aL Recursively finds files matching glob `pattern` under `directory`.
This is an alternative to `logilab.common.shellutils.find`.
:type directory: str
:param directory:
directory where the search should start
:type pattern: basestring
:param pattern:
the glob pattern (e.g *.py, foo*.py, etc.)
:type blacklist: list or tuple
:param blacklist:
optional list of files or directory to ignore, default to the value of
`logilab.common.STD_BLACKLIST`
:rtype: iterator
:return:
iterator over the list of all matching files
N)r& rR r �fnmatch�filterr )r$ �patternrU �curdirrV rW �fnamer r r �globfind� s r_ c C s~ dd l }t|�st�|� |�| �}xT|�� D ]H}|�d�rNt�t||�� q.tt||�d�}|� |�
|�� |� � q.W d S )Nr �/�wb) �zipfiler r& �mkdir�ZipFile�namelistrI r �open�write�read�close)�archiveZdestdirrb Zzfobj�name�outfiler r r �unzip� s
rm c @ s e Zd ZdZdd� ZdS )�ExecutezsThis is a deadlock safe version of popen2 (no stdin), that returns
an object with errorlevel, out and err.
c C s8 t j|dt jt jd�}|�� \| _| _t�|j�| _ d S )NT)�shell�stdout�stderr)
�
subprocess�Popen�PIPE� communicate�out�errr& � WEXITSTATUS�
returncode�status)r �command�cmdr r r r% � s zExecute.__init__N)r r r �__doc__r% r r r r rn � s rn z Use subprocess.Popen instead�
� c
C s\ t |�}�xL|�rJy@t�| tjtjB tjB �}t�|ttt� � ��� t�
|� dS t k
�rF } z�|j t j
k�r"ybt| d�}t|�� �}t|�}t�� t�| �tj } | | dkr�td|�� || | d f ��W nH tk
r� � Y n4 tk
�r
t�| � Y n tk
�r Y nX n� |d8 }t�|� W dd}~X Y q X q W td| ��dS )z�Acquire a lock represented by a file on the file system
If the process written in lock file doesn't exist anymore, we remove the
lock file immediately
If age of the lock_file is greater than max_delay, then we raise a UserWarning
T�rr7 z=Command '%s' (pid %s) has locked the file '%s' for %s minutes�< NzUnable to acquire %s)�absr& rf �O_EXCL�O_RDWR�O_CREATrg r
�str�getpidri r< �errno�EEXISTr+ �readliner �time�stat�ST_MTIME� UserWarningrk r rD � Exception�sleep)
� lock_fileZmax_try�delayZ max_delay�count�fd�er �piZager r r � acquire_lock� s6
r� c C s t �| � dS )z8Release a lock represented by a file on the file system.N)r& rD )r� r r r � release_lock" s r� c @ sb e Zd ZdZdejdfdd�Zdd� Zdd d
�Zd d � Z e
eee �Z ddd�Z dd� Z
dd� ZdS )� ProgressBarz A simple text progression bar.� � c C sT |rd|t |�f | _ndt |� | _|| _|| _|| _d| _d| _d | _d| _d S )Nz
%s [%%-%ss]z
[%%-%ss]r ) r+ �_fstr�_stream�_total�_size�_current� _progress�
_current_text�_last_text_write_size)r �nbops�size�stream�titler r r r% * s zProgressBar.__init__c C s | j S )N)r� )r r r r � _get_text7 s zProgressBar._get_textNc C s || j kr|| _ | �� d S )N)r� �refresh)r �textr r r � _set_text: s
zProgressBar._set_textc C s
d | _ d S )N)r� )r r r r � _del_text? s zProgressBar._del_textr7 Fc C sT |r || _ n| j |7 _ tt| j �t| j� | j �}|| jkrP|| _| �� dS )z�Move FORWARD to new cursor position (cursor will never go backward).
:offset: fraction of ``size``
:exact:
- False: offset relative to current cursor position if True
- True: offset as an asbsolute position
N)r� r+ �floatr� r� r� r� )r �offset�exact�progressr r r �updateD s
zProgressBar.updatec C st | j �| jdt| j| j� � | js,| jrfd| j }| j}|dkrHd}| j �|| � t|� � �| _| j �
� dS )z$Refresh the progression bar display.�=z %%-%isNr� ) r� rg r� �minr� r� r� r� r9 �rstrip�flush)r �templater� r r r r� Y s
zProgressBar.refreshc C s | j �d� | j �� d S )N�
)r� rg r� )r r r r �finishe s zProgressBar.finish)N)r7 F)r r r r} �sysrp r% r� r� r� �propertyr� r� r� r� r r r r r� ' s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DummyProgressBar)r� c C s d S )Nr )r r r r r� m s zDummyProgressBar.refreshc C s d S )Nr )r r r r r� o s zDummyProgressBar.updatec C s d S )Nr )r r r r r� q s zDummyProgressBar.finishN)r r r Z__slot__r� r� r� r r r r r� j s r� c @ s0 e Zd Zeeeedfdd�Zdd� Zdd� ZdS ) r� Tc C s" || _ || _|| _|| _|| _d S )N)r� r� r� r� �enabled)r r� r� r� r� r� r r r r% x s
zprogress.__init__c C sN | j r@i }x&dD ] }t| |�}|tk r|||< qW tf |�| _nt� | _| jS )N)r� r� r� r� )r� �getattr�_MARKERr� �pbr� )r �kwargs�attrr r r r r s
zprogress.__enter__c C s | j �� d S )N)r� r� )r �exc_type�exc_val�exc_tbr r r r" � s zprogress.__exit__N)r r r r� r% r r" r r r r r� v s r� c @ s( e Zd Zd dd�Zdd� Zd
dd�ZdS ) �RawInputNc C s |pt | _|| _d S )N)� raw_input�_input�_print)r r
�printerr r r r% � s
zRawInput.__init__c s: ||ks t �g }x`|D ]X}||kr0|d �� }n |d �� }t|�dkr`|d|dd � �� 7 }|�||f� qW d|d�dd� |D ��f }d}x�|dk�r,| �|��� �� � � s�|S � fd d�|D �} t| �dkr�| d S t| �dkr�d
� }
nd � d �| �f }
| j�r| �|
� nt |
� |d8 }q�W t
d
��d S )Nr r7 z(%s)z %s [%s]: r` c S s g | ] }|d �qS )r7 r )rM �optr r r rP � s z RawInput.ask.<locals>.<listcomp>� c s" g | ]\}}|� � �� �r|�qS r )�lower�
startswith)rM �option�label)�answerr r rP � s z%s is not an option.z+%s is an ambiguous answer, do you mean %s ?z or z unable to get a sensible answer) r: �upperr� r9 �appendr r� �stripr� �printr� ) r �question�options�default�choicesr� r� �promptZtries�possible�msgr )r� r �ask� s8
z RawInput.askTc C s" |rdp
d}| � |d|�}|dkS )N�y�n)r� r� )r� )r r� Zdefault_is_yesr� r� r r r �confirm� s zRawInput.confirm)NN)T)r r r r% r� r� r r r r r� � s
"r� c C s2 t jdkr$ddl} | �t�� �d S tjd S dS )z�avoid using os.getlogin() because of strange tty / stdin problems
(man 3 getlogin)
Another solution would be to use $LOGNAME, $USER or $USERNAME
�win32r NZUSERNAME)r� �platformr- �getpwuidr& �getuid�environ)r- r r r �getlogin� s
r� � c C s( d}x t | �D ]}|t�|�7 }qW |S )z!dumb password generation functionr� )r �random�choice)�lengthZvocabr- �ir r r �generate_password� s r� )NN)r~ r~ r )@r} �
__future__r �
__docformat__r& r8 r r� r� r r� rZ r� �stringr� rr �os.pathr r r r r Zsixr Z six.movesr r
r� Zlogilab.commonr r Zlogilab.common.compatr
Zlogilab.common.deprecationr Zlogilab.common.procr r � ImportErrorr� �objectr r# r3 �moverC rF rH rY r_ rm rn r� r� r� r� r� r� r� ZASKr� �
ascii_letters�digitsr� r r r r �<module> s\
2
*C -