循环通过peewee导致烧瓶

Question

基于此回复：

cursor = db.execute_sql('select * from tweets;')
for row in cursor.fetchall():
    print row

cursor = db.execute_sql('select count(*) from tweets;')
res = cursor.fetchone()
print 'Total: ', res[0]

来自：Python Peewee execute_sql() example

如何将其带到flask app，然后显示在网页中？

这对吗：

模型.py

def get_statistics():
        cursor = finDB.execute_sql('CALL Allstatistics;')
        for row in cursor.fetchall():
                return row

app.py

@app.route("/finance")
def finance():
        stats = model.get_statistics()
        return render_template('/finance.html', stats=stats)

但是如何在表格中显示呢？

Answer 1

问题在于您对以下内容的适应：

for row in cursor.fetchall():
    print row

这将一一打印fetchall()返回的所有行。

你试图把它改编成一个函数returning所有行：

def get_statistics():
    cursor = finDB.execute_sql('CALL Allstatistics;')
    for row in cursor.fetchall():
        return row

现在这将仅return 第一行，因为 return 语句在第一次迭代时终止您的循环。

你真正想要的是这样的：

def get_statistics():
    cursor = finDB.execute_sql('CALL Allstatistics;')
    return cursor.fetchall()

这将正确返回游标中的所有行，如果没有结果行，则返回None。

通过检查是否有非空结果，而不是 None 返回一个空列表，您可以这样做：

def get_statistics():
    cursor = finDB.execute_sql('CALL Allstatistics;')
    rows = cursor.fetchall()
    if rows:
        return rows
    return []

关于cursor.fetchone()，这将返回光标的下一个可用行，如果没有更多行可用，则返回None。例如，您可以像这样遍历光标中的所有可用行：

rows = []
row = cursor.fetchone() # fetch first row, or None if empty result
while row is not None:
    rows.append(row)
    row = cursor.fetchone() # fetch the next row, if None loop terminates
return rows # return all collected results

对于您的用例，为您的结果构建一个更方便的数据结构可能会很有趣，例如list of dicts:

rows = []
row = cursor.fetchone()
while row is not None:
    rows.append({'foo': row[0], 'bar': row[1], 'baz': row[2]})
    row = cursor.fetchone()
return rows

请注意，这可以类似地实现：

rows = []
for row in cursor.fetchall():
    rows.append({'foo': row[0], 'bar': row[1], 'baz': row[2]})
return rows

然后你可以在你的模板中写，循环for row in rows：

foo is {{row['foo']}} and bar is {{row['bar']}}

或者你可以构造一个namedtuple的列表，允许你写在模板中：

foo is {{row.foo}} and bar is {{foo.bar}}