From 149bafa97b3c93b5116090d2834f0c19e6face95 Mon Sep 17 00:00:00 2001
From: Josh Ourisman <josh@joshourisman.com>
Date: Fri, 1 Oct 2010 16:17:04 -0400
Subject: added ability to append new column passing a callable as the value
 that will be applied to every row; w/ test

---
 tablib/core.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'tablib')

diff --git a/tablib/core.py b/tablib/core.py
index 8f74975..dfb8027 100644
--- a/tablib/core.py
+++ b/tablib/core.py
@@ -177,6 +177,15 @@ class Dataset(object):
 			self._validate(row)
 			self._data.append(tuple(row))
 		elif col is not None:
+			col = list(col)
+			if self.headers:
+				header = [col.pop(0)]
+			else:
+				header = []
+			if len(col) == 1 and callable(col[0]):
+				col = map(col[0], self._data)
+			col = tuple(header + col)
+				
 			self._validate(col=col)
 
 			if self.headers:
-- 
cgit v1.2.1