Fix a couple of more iterator changes

This commit is contained in:
Stefan Wehrmeyer 2013-12-18 11:32:58 +01:00
parent 426095634c
commit 2870fdfb0b
5 changed files with 7 additions and 10 deletions

View File

@ -20,7 +20,7 @@ class CSVSerializer(Serializer):
self.handles = {}
def write(self, path, result):
keys = result.keys()
keys = list(result.keys())
if not path in self.handles:
fh = open(path, 'wb')
writer = csv.writer(fh)
@ -31,7 +31,5 @@ class CSVSerializer(Serializer):
writer.writerow(values)
def close(self):
for (writer, fh) in self.handles.values():
for writer, fh in self.handles.values():
fh.close()

View File

@ -46,4 +46,3 @@ class JSONSerializer(Serializer):
data)
fh.write(data)
fh.close()

View File

@ -7,7 +7,7 @@ class TabsonSerializer(JSONSerializer):
fields = []
data = []
if len(result):
keys = result[0].keys()
keys = list(result[0].keys())
fields = [{'id': k} for k in keys]
for row in result:
d = [row.get(k) for k in keys]

View File

@ -322,7 +322,7 @@ class Table(object):
self._check_dropped()
if not isinstance(order_by, (list, tuple)):
order_by = [order_by]
order_by = filter(lambda o: o in self.table.columns, order_by)
order_by = [o for o in order_by if o in self.table.columns]
order_by = [self._args_to_order_by(o) for o in order_by]
args = self._args_to_clause(_filter)
@ -358,7 +358,7 @@ class Table(object):
"""
Returns the number of rows in the table.
"""
d = self.database.query(self.table.count()).next()
d = next(self.database.query(self.table.count()))
return list(d.values()).pop()
def distinct(self, *columns, **_filter):

View File

@ -33,7 +33,7 @@ class ResultIter(object):
try:
self.rp = next(self.result_proxies)
self.count += self.rp.rowcount
self.keys = self.rp.keys()
self.keys = list(self.rp.keys())
return True
except StopIteration:
return False
@ -42,7 +42,7 @@ class ResultIter(object):
row = self.rp.fetchone()
if row is None:
if self._next_rp():
return self.next()
return next(self)
else:
# stop here
raise StopIteration