3
\�" � @ sl d dl Z d dlZd dlZd dlmZ dgZdZdZdZdZ dZ
d Z eeee e
e gZ G d
d� d�Z
d d � ZdS )
� N)�quote�TemplateZffz-fzf-z--z.-z-.c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )r c C s d| _ | j� d S )Nr )� debugging�reset)�self� r � /usr/lib64/python3.6/pipes.py�__init__U s zTemplate.__init__c C s d| j f S )Nz )�steps)r r r r �__repr__Z s zTemplate.__repr__c C s
g | _ d S )N)r
)r r r r r ^ s zTemplate.resetc C s" t � }| jd d � |_| j|_|S )N)r r
r )r �tr r r �cloneb s zTemplate.clonec C s
|| _ d S )N)r )r �flagr r r �debugj s zTemplate.debugc C s� t |�t d�k rtd��|tkr.td|f ��|tkr>td��| jr^| jd
d tkr^td��|d dkr�tjd |� r�td
��|d dkr�tjd |� r�td ��| jj ||f� d S )N� z%Template.append: cmd must be a stringz Template.append: bad kind %rz-Template.append: SOURCE can only be prepended� z'Template.append: already ends with SINKr �fz\$IN\bz#Template.append: missing $IN in cmdz\$OUT\bz$Template.append: missing $OUT in cmd�����)
�type� TypeError� stepkinds�
ValueError�SOURCEr
�SINK�re�search�append)r �cmd�kindr r r r n s zTemplate.appendc C s� t |�t d�k rtd��|tkr.td|f ��|tkr>td��| jr^| jd d tkr^td��|d dkr�tjd |� r�td
��|d dkr�tjd |� r�td ��| jj d||f� d S )
Nr z&Template.prepend: cmd must be a stringz Template.prepend: bad kind %rz+Template.prepend: SINK can only be appendedr r z,Template.prepend: already begins with SOURCEr z\$IN\bz$Template.prepend: missing $IN in cmdz\$OUT\bz%Template.prepend: missing $OUT in cmd)
r r r r r r
r r r �insert)r r r r r r �prepend~ s zTemplate.prependc C s6 |dkr| j |�S |dkr$| j|�S td|f ��d S )N�r�wz,Template.open: rw must be 'r' or 'w', not %r)�open_r�open_wr )r �fileZrwr r r �open� s
z
Template.openc C sB | j st|d�S | j d d tkr*td��| j|d�}tj|d�S )Nr! r z)Template.open_r: pipeline ends width SINKr r )r
r& r r � makepipeline�os�popen)r r% r r r r r# � s
zTemplate.open_rc C sB | j st|d�S | j d d tkr*td��| jd|�}tj|d�S )Nr" r r z,Template.open_w: pipeline begins with SOURCEr )r
r& r r r' r( r) )r r% r r r r r$ � s
zTemplate.open_wc C s t j| j||��S )N)r( �systemr' )r �infile�outfiler r r �copy� s z
Template.copyc C s( t || j|�}| jr$t|� d| }|S )Nzset -x; )r' r
r �print)r r+ r, r r r r r' � s
zTemplate.makepipelineN)�__name__�
__module__� __qualname__r r r r
r r r r& r# r$ r- r' r r r r r R s
c C s� g }x"|D ]\}}|j d||dg� q
W |s>|j ddddg� |d dd� \}}|d dkrx| rx|jdddddg� | |d d< |d dd� \}}|d dkr�| r�|j ddddg� ||d d< g }x�tdt|��D ]v}||d d }|| d } |d dk�s| d dkr�tj� \}
} tj|
� |j | � | ||d d< || d< q�W x�|D ]�} | \}
}}}|d dk�r�d t|� d
| }|d dk�r�d t|
� d
| }|d d k�r�|
�r�|d
t|
� }|d d k�r�|�r�|d t|� }|| d< �q\W |d d }xd|dd � D ]T} | dd� \}}| d dk�rfd|k�rXd| d }|d | }n |d | }�q W |�r�d}x |D ]}|d t|� }�q�W dt|d � d }|d | d | }|S ) Nr �catz--r r � r � zOUT=z; zIN=�-z z{ z; }z |
�
zrm -f� ztrap z; exitz 1 2 3 13 14 15r r r r ) r r �range�len�tempfileZmkstempr( �closer )r+ r
r, �listr r Zgarbage�iZlkindZrkind�fdZtemp�item�infZoutfZcmdlistZrmcmdr% Ztrapcmdr r r r' � s`
r' )r r( r: Zshlexr �__all__ZFILEIN_FILEOUTZ
STDIN_FILEOUTZ
FILEIN_STDOUTZ STDIN_STDOUTr r r r r' r r r r �< s c