Added a __len__ method to the columns container.

miracle2k · miracle2k · commit 5ee008025add · 2010-04-06T15:18:00.000+02:00
diff --git a/django_tables/base.py b/django_tables/base.py
@@ -249,6 +249,10 @@ def __contains__(self, item):
         else:
             return item in self.all()
 
+    def __len__(self):
+        self._spawn_columns()
+        return len([1 for c in self._columns.values() if c.visible])
+
     def __getitem__(self, name):
         """Return a column by name."""
         self._spawn_columns()
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -53,6 +53,15 @@ class StateTable2(GeoAreaTable, tables.ModelTable):
     assert 'motto' in StateTable2.base_columns
 
 
+def test_column_count():
+    class MyTable(TestTable):
+        visbible = tables.Column(visible=True)
+        hidden = tables.Column(visible=False)
+
+    # The columns container supports the len() builtin
+    assert len(MyTable([]).columns) == 1
+
+
 def test_pagination():
     class BookTable(TestTable):
         name = tables.Column()
