检查是否<sqlalchemy.engine.result.ResultProxy>对象为空

Question

我正在通过一个在后端利用pymssql的库从 SQL Server 数据库下载一些数据。 一的结果curson.execute("""<QUERY BODY>""")是一个sqlalchemy.engine.result.ResultProxy对象。 如何检查查询结果是否为空，因此没有行？

cur = ff.sql.create_engine(server=dw.address, db=dw.BI_DW,
                               login=":".join([os.environ["SQL_USER"],
                                               os.environ["SQL_PASSWD"]]))
for n in range(100):
    result = cur.execute("""QUERY BODY;""")
    if result:
        break

不幸的是，即使 SQL 查询没有返回任何行， result也永远不会是None 。

检查的最佳方法是什么？

Answer 1

ResultProxy对象还不包含任何行。 因此，它没有关于它们总量的信息，甚至没有关于它们是否有的信息。 ResultProxy只是一个指向数据库的“指针”。 只有当您通过ResultProxy显式获取行时，您才能获取行。 你可以做到这一点通过迭代过这个对象，或通过.first()方法，或通过.fetchall()方法。

底线：在您实际获取所有行并且ResultProxy对象耗尽之前，您无法知道 fethced 行的数量。

方法#1

您可以一次获取所有行并计算它们，然后对它们执行任何您需要的操作：

rows = result.fetchall()
if len(rows):
    # do something with rows

这种方法的缺点是我们一次将所有行加载到内存中（ rows是一个包含所有获取行的 Python list ）。 如果获取的行数非常大和/或如果您只需要一个一个地独立迭代行（通常是这种情况），这可能是不可取的。

方法#2

如果一次将所有获取的行加载到内存中是不可接受的，那么我们可以这样做：

rows_amount = 0
for row in result:
    rows_amount += 1
    # do something with row
if not rows_amount:
    print('There were zero rows')
else:
    print('{} rows were fetched and processed'.format(rows_amount))

Answer 2

SQLAlchemy < 1.2：您始终可以将 ResultProxy 转换为迭代器：

res = engine.execute(...)
rp_iter = iter(res)
row_count = 0
try:
    row = next(rp_iter)
    row_count += 1
except StopIteration:
    # end of data
    if not row_count:
      # no rows returned, StopIteration was raised on first attempt

在SQLAlchemy >= 1.2 中， ResultProxy 实现了.next()和.__next__() ，所以你不需要创建迭代器：

res = engine.execute()
row_count = 0
try:
    row = next(res)
    row_count += 1
except StopIteration:
    ...