�
i f�" � @ s� d Z d d l Z d d l Z d d l Z d d l m Z d g Z d Z d Z d Z d Z
d Z d
Z e e e e
e e g Z
Gd d � d � Z d d
� Z d S)a� Conversion pipeline templates.
The problem:
------------
Suppose you have some data that you want to convert to another format,
such as from GIF image format to PPM image format. Maybe the
conversion involves several steps (e.g. piping it through compress or
uuencode). Some of the conversion steps may require that their input
is a disk file, others may be able to read standard input; similar for
their output. The input to the entire conversion may also be read
from a disk file or from an open file, and similar for its output.
The module lets you construct a pipeline template by sticking one or
more conversion steps together. It will take care of creating and
removing temporary files if they are necessary to hold intermediate
data. You can then use the template to do conversions from many
different sources to many different destinations. The temporary
file names used are different each time the template is used.
The templates are objects so you can create templates for many
different conversion steps and store them in a dictionary, for
instance.
Directions:
-----------
To create a template:
t = Template()
To add a conversion step to a template:
t.append(command, kind)
where kind is a string of two characters: the first is '-' if the
command reads its standard input or 'f' if it requires a file; the
second likewise for the output. The command must be valid /bin/sh
syntax. If input or output files are required, they are passed as
$IN and $OUT; otherwise, it must be possible to use the command in
a pipeline.
To add a conversion step at the beginning:
t.prepend(command, kind)
To convert a file to another file using a template:
sts = t.copy(infile, outfile)
If infile or outfile are the empty string, standard input is read or
standard output is written, respectively. The return value is the
exit status of the conversion pipeline.
To open a file for reading or writing through a conversion pipeline:
fp = t.open(file, mode)
where mode is 'r' to read the file, or 'w' to write it -- just like
for the built-in function open() or for os.popen().
To create a new template object initialized to a given one:
t2 = t.clone()
� N)�quote�TemplateZffz-fzf-z--z.-z-.c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d S)r z'Class representing a pipeline template.c C s d | _ | j � d S)z-Template() returns a fresh pipeline template.r N)� debugging�reset)�self� r �*/opt/alt/python34/lib64/python3.4/pipes.py�__init__U s zTemplate.__init__c C s d | j f S)z t.__repr__() implements repr(t).z <Template instance, steps=%r>)�steps)r r r r �__repr__Z s zTemplate.__repr__c C s
g | _ d S)z<t.reset() restores a pipeline template to its initial state.N)r
)r r r r r ^ s zTemplate.resetc C s/ t � } | j d d � | _ | j | _ | S)zbt.clone() returns a new pipeline template with identical
initial state as the current one.N)r r
r )r �tr r r �cloneb s zTemplate.clonec C s
| | _ d S)z(t.debug(flag) turns debugging on or off.N)r )r Zflagr r r �debugj s zTemplate.debugc C s t | � t d � k r' t d � � n | t k rI t d | f � � n | t k rd t d � � n | j r� | j d d t k r� t d � � n | d d k r� t j d | � r� t d
� � n | d d k r� t j d | � r� t d � � n | j j | | f � d
S)z/t.append(cmd, kind) adds a new step at the end.� 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 cmdN���)
�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 r' t d � � n | t k rI t d | f � � n | t k rd t d � � n | j r� | j d d t k r� t d � � n | d d k r� t j d | � r� t d
� � n | d d k r� t j d | � r� t d � � n | j j d | | f � d
S)z2t.prepend(cmd, kind) adds a new step at the front.r 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 cmdN)
r r r r r r
r r r �insert)r r r r r r �prepend~ s ##zTemplate.prependc C sI | d k r | j | � S| d k r2 | j | � St d | f � � d S)z~t.open(file, rw) returns a pipe or file object open for
reading or writing; the file is the other end of the pipeline.�r�wz,Template.open: rw must be 'r' or 'w', not %rN)�open_r�open_wr )r �fileZrwr r r �open� s
z
Template.openc C s^ | j s t | d � S| j d d t k r< t d � � n | j | d � } t j | d � S)zit.open_r(file) and t.open_w(file) implement
t.open(file, 'r') and t.open(file, 'w') respectively.r 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 s^ | j s t | d � S| j d d t k r< t d � � n | j d | � } t j | 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 s9 t | | j | � } | j r5 t | � d | } n | S)Nzset -x; )r& r
r �print)r r* r+ r r r r r& � s
zTemplate.makepipelineN)�__name__�
__module__� __qualname__�__doc__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 | | d g � q
W| sX | j d d d d g � n | d d d � \ } } | d d k r� | r� | j d d d d d g � n | | d d <| d d d � \ } } | d d k r| r| j d d d d g � n | | d d <g } x� t d t | � � D]� } | | d d } | | d } | d d k ss| d d k r-t j � \ }
} t j |
� | j | � | | | d d <| | d <q-q-Wx� | D]� } | \ }
} } } | d d k rd t | � d
| } n | d d k r9d t |
� d
| } n | d d k rf|
rf| d
t |
� } n | d d k r�| r�| d t | � } n | | d <q�W| d d } xz | d d � D]h } | d d � \ } } | d d k rd | k r d | d } n | d | } q�| d | } q�W| r�d } x" | D] } | d t | � } q?Wd t | d � d } | d | d | } n | S) Nr �catz--r r � r � zOUT=z; zIN=�-z <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& )r1 r r' r: Zshlexr �__all__ZFILEIN_FILEOUTZ
STDIN_FILEOUTZ
FILEIN_STDOUTZ STDIN_STDOUTr r r r r& r r r r �<module>9 s c