Fix a couple of more iterator changes
This commit is contained in:
parent
426095634c
commit
2870fdfb0b
@ -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()
|
||||
|
||||
|
||||
|
||||
@ -46,4 +46,3 @@ class JSONSerializer(Serializer):
|
||||
data)
|
||||
fh.write(data)
|
||||
fh.close()
|
||||
|
||||
|
||||
@ -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]
|
||||
|
||||
@ -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):
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user