summaryrefslogtreecommitdiff
path: root/tablib/formats/_csv.py
diff options
context:
space:
mode:
authorKenneth Reitz <me@kennethreitz.com>2010-09-25 17:18:51 -0400
committerKenneth Reitz <me@kennethreitz.com>2010-09-25 17:18:51 -0400
commit9f26c23eb50f6d00e04b4fe4bd5e5e98b512cfbe (patch)
tree043f062e3eeb93939ec368dcd8c40b0e63b418a8 /tablib/formats/_csv.py
parentdc21825f346e2c422c7a3ebc180cce62726aaaa7 (diff)
parent8136f4b09eac5ed160fd49ec70f2c1338df6ec8d (diff)
downloadtablib-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.py42
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)