Allowing for direct iteration over rows in a table

Browse Source
This commit is contained in:
Gregor Aisch 2013-04-03 12:28:32 +02:00
parent 7638e898fd
commit 258e3f3059
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
dataset/persistence/table.py
View File

@ -231,6 +231,9 @@ class Table(object):
yield row yield row
def __len__(self): def __len__(self):
"""
Returns the number of rows in the table.
"""
d = self.database.query(self.table.count()).next() d = self.database.query(self.table.count()).next()
return d.values().pop() return d.values().pop()
@ -268,3 +271,14 @@ class Table(object):
rows = table.all()""" rows = table.all()"""
return self.find() return self.find()
def __iter__(self):
"""
Allows for iterating over all rows in the table without explicelty calling :py:meth:`all() <dataset.Table.all>`.
::
for row in table:
print row
"""
for row in self.all():
yield row