How to make Python play nicely with pipes

sigpipe.py

import sys
import errno

def main():
    for i in range(20):
        print i

if __name__ == "__main__":
    # ensure buffer is flushed before we handle SIGPIPE
    try:
        try:
            main()

        finally:
            sys.stdout.flush()

    except IOError, e:
        if e.errno != errno.EPIPE:
            raise