Decorator: csv reader

csv_decorator.py

import csv
import string

def make_csv():
    with open('data.csv', 'wb') as fh:
        writer = csv.writer(fh)
        for i in range(10):
            writer.writerow([i, string.lowercase[i:i+5]])

def process_line(line):
    return [int(line[0]), line[1].title()]

def csv_wrapper(func):
    def opener(line_transform):
        with open('data.csv', 'rb') as fh:
            reader = csv.reader(fh)
            data = list(func(reader, line_transform))
        return data
    return opener


@csv_wrapper
def process_reader(reader, line_transform):
    for line in reader:
        yield line_transform(line)