From ce107151c020b5bf822ba50004a285bd976ea447 Mon Sep 17 00:00:00 2001
From: Joe Germuska <joe@germuska.com>
Date: Mon, 14 Nov 2011 16:38:58 -0600
Subject: [PATCH] basic implementation of dict reader/writer by subclassing
 python's csv classes

---
 csvkit/unicsv.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/csvkit/unicsv.py b/csvkit/unicsv.py
index 6122429df..33292dc79 100644
--- a/csvkit/unicsv.py
+++ b/csvkit/unicsv.py
@@ -49,6 +49,10 @@ def next(self):
     def __iter__(self):
         return self
 
+    @property
+    def line_num(self):
+        return self.reader.line_num
+
 class UnicodeCSVWriter(object):
     """
     A CSV writer which will write rows to CSV file "f",
@@ -76,3 +80,31 @@ def writerow(self, row):
     def writerows(self, rows):
         for row in rows:
             self.writerow(row)
+
+class UnicodeCSVDictReader(csv.DictReader):
+    """Defer almost all implementation to csv.DictReader, but wrapping our unicode reader instead
+       of csv.reader
+    """
+    def __init__(self, f, fieldnames=None, restkey=None, restval=None, *args, **kwds):
+        self._fieldnames = fieldnames   # list of keys for the dict
+        self.restkey = restkey          # key to catch long rows
+        self.restval = restval          # default value for short rows
+        self.reader = UnicodeCSVReader(f, *args, **kwds)
+        self.line_num = 0
+
+
+class UnicodeCSVDictWriter(csv.DictWriter):
+    """Defer almost all implementation to csv.DictReader, but wrapping our unicode reader instead
+       of csv.reader
+    """
+    def __init__(self, f, fieldnames, writeheader=False, restval="", extrasaction="raise", *args, **kwds):
+        self.fieldnames = fieldnames    # list of keys for the dict
+        self.restval = restval          # for writing short dicts
+        if extrasaction.lower() not in ("raise", "ignore"):
+            raise ValueError, \
+                  ("extrasaction (%s) must be 'raise' or 'ignore'" %
+                   extrasaction)
+        self.extrasaction = extrasaction
+        self.writer = UnicodeCSVWriter(f, *args, **kwds)
+        if writeheader:
+            self.writerow(dict(zip(self.fieldnames,self.fieldnames)))