diff options
| author | Kenneth Reitz <me@kennethreitz.com> | 2010-09-25 17:18:51 -0400 |
|---|---|---|
| committer | Kenneth Reitz <me@kennethreitz.com> | 2010-09-25 17:18:51 -0400 |
| commit | 9f26c23eb50f6d00e04b4fe4bd5e5e98b512cfbe (patch) | |
| tree | 043f062e3eeb93939ec368dcd8c40b0e63b418a8 /tablib/formats/_csv.py | |
| parent | dc21825f346e2c422c7a3ebc180cce62726aaaa7 (diff) | |
| parent | 8136f4b09eac5ed160fd49ec70f2c1338df6ec8d (diff) | |
| download | tablib-0.8.0.tar.gz | |
Merge branch 'release/0.8.0'v0.8.0
Diffstat (limited to 'tablib/formats/_csv.py')
| -rw-r--r-- | tablib/formats/_csv.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/tablib/formats/_csv.py b/tablib/formats/_csv.py new file mode 100644 index 0000000..8b19da7 --- /dev/null +++ b/tablib/formats/_csv.py @@ -0,0 +1,42 @@ +# -*- coding: utf-8 -*- + +""" Tablib - CSV Support. +""" + +import cStringIO +import csv +import os + +import simplejson as json + +import tablib + + +title = 'csv' +extentions = ('csv',) + + + +def export_set(dataset): + """Returns CSV representation of Dataset.""" + stream = cStringIO.StringIO() + _csv = csv.writer(stream) + + for row in dataset._package(dicts=False): + _csv.writerow(row) + + return stream.getvalue() + + +def import_set(dset, in_stream, headers=True): + """Returns dataset from CSV stream.""" + + dset.wipe() + + rows = csv.reader(in_stream.split()) + for i, row in enumerate(rows): + + if (i == 0) and (headers): + dset.headers = row + else: + dset.append(row) |
