iterator decorators / function composition

iterfoo.py

#!/usr/bin/python

import itertools

def iterchain( fn ):
  def new( *iterables ):
    return itertools.imap( fn, *iterables )
  return new 
def itersink( fn ):
  def new( *iterables ):
    return map( fn, *iterables )
  return new 
def it():
  return [1,2,3,4]
@iterchain
def do( i ):
  return i * 2 
@itersink
def go( i ):
  print i

if __name__ == '__main__':
  go( do( it() ) )