fastapi雪花连接只拉1条记录

Question

我正在尝试使用 FASTAPI 从雪花数据库中读取数据。我能够创建能够从雪花中提取数据的连接。

我现在面临的问题是我只得到 1 条记录（而不是 10 条记录）。我怀疑我在返回数据时没有使用正确的关键字。感谢任何帮助。

这是我的代码：-

from fastapi import FastAPI
import snowflake.connector as sf
import configparser


username='username_value'
password='password_value'
account= 'account_value'
warehouse= 'test_wh'
database= 'test_db'


ctx=sf.connect(user=username,password=password,account=account,warehouse=warehouse,database=database)

app = FastAPI()
@app.get('/test API')
async def fetchdata():
  cursor = ctx.cursor()
  cursor.execute("USE WAREHOUSE test_WH ")
  cursor.execute("USE DATABASE test_db") 
  cursor.execute("USE SCHEMA test_schema")
  sql = cursor.execute ("SELECT DISTINCT ID,NAME,AGE,CITY FROM TEST_TABLE WHERE AGE > 60")

for data in sql:
  return data

Answer 1

你在你的内心 for-loop 中使用了 return。这将 return 遇到第一行。

如果你想return所有行作为一个列表，你可能会这样做（我不熟悉雪花连接器）：

return list(data)

而不是 for 循环，或 sql.fetchall()。

fastapi雪花连接只拉1条记录

fastapi snowflake connection only pulling 1 record

python

snowflake-cloud-data-platform

fastapi