view numerics/csvwriter.py @ 172:dc0a620a0368

add another example
author Jeff Hammel <k0scist@gmail.com>
date Fri, 03 Jul 2015 11:11:01 -0700
parents 7927d7127f9a
children
line wrap: on
line source

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import csv
string = (str, unicode)
__all__ = ['CSVWriter']

class CSVWriter(object):
    """a more sensible front-end to writing CSV files"""

    def __init__(self, f, mode='a'):
        if isinstance(f, string):
            f = open(f, mode)
        self.f = f
        self.writer = csv.writer(f)

    def __call__(self, *data):
        self.writer.writerow(data)
        self.f.flush()