From 6313437a27d68e42247802f7f68ec3270cb9a2da Mon Sep 17 00:00:00 2001 From: Mark Walling Date: Fri, 1 Jul 2011 17:51:43 -0400 Subject: Added support for detecting unicode column headers Also added tests! Fix for kennethreitz#26 --- tablib/core.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tablib') diff --git a/tablib/core.py b/tablib/core.py index 53f3767..00e55af 100644 --- a/tablib/core.py +++ b/tablib/core.py @@ -162,7 +162,7 @@ class Dataset(object): def __getitem__(self, key): - if isinstance(key, str): + if isinstance(key, str) or isinstance(key, unicode): if key in self.headers: pos = self.headers.index(key) # get 'key' index from each data return [row[pos] for row in self._data] @@ -182,7 +182,7 @@ class Dataset(object): def __delitem__(self, key): - if isinstance(key, str): + if isinstance(key, str) or isinstance(key, unicode): if key in self.headers: @@ -730,7 +730,7 @@ class Dataset(object): sorted. """ - if isinstance(col, str): + if isinstance(col, str) or isinstance(col, unicode): if not self.headers: raise HeadersNeeded -- cgit v1.2.1