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
+2 -4
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()
-1
View File
@@ -46,4 +46,3 @@ class JSONSerializer(Serializer):
data)
fh.write(data)
fh.close()
+1 -1
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]
+2 -2
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):
+2 -2
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