python - 打印以查看 BigQuery 查询的结果

Question

我在函数内有一个 BigQuery 查询，如下所示:

def get_data_from_bigquery():
    """query bigquery to get data to import to PSQL"""
    app_id = get_app_id()
    bq = bigquery.Client(project=app_id)
    query = """SELECT * FROM `Backup.Prod_UserUserMetadata`"""
    query_job = bq.query(query)
    data = query_job.result()
    rows = list(data)
    print(rows)
    return rows

我的路线如下:

@app.route('/azure-import-data')
def print_data():
    return 'This is the data:\n \n'

函数 get_data_from_bigquery() 正在以下类的 get() 函数中调用:

class AzureImportProcess(Resource):
    def get(self):
        get_data_from_bigquery()
...
api.add_resource(AzureImportProcess, '/azure-import-data')

print(rows) 函数似乎不起作用，因为它没有在 Cloud Console 或浏览器中打印。有没有其他方法可以打印数据以查看其格式如何？

Answer 1

首先，您不需要在 get_data_from_bigquery 函数中使用 print 语句。其次，get 函数需要有一个 return 语句才能显示某些内容。

编辑函数定义如下:

def get_data_from_bigquery():
    """query bigquery to get data to import to PSQL"""
    app_id = get_app_id()
    bq = bigquery.Client(project=app_id)
    query = """SELECT * FROM `Backup.Prod_UserUserMetadata`"""
    query_job = bq.query(query)
    data = query_job.result()
    rows = list(data)
    return rows

编辑Resource类定义，如下所示:

class AzureImportProcess(Resource):
    def get(self):
        bq_data = get_data_from_bigquery()
        return 'This is the data:\n \n{}'.format(bq_data)

现在您可以将资源添加到 API 对象中:

api.add_resource(AzureImportProcess, '/azure-import-data')

您不需要在此处定义 print_data 函数，除非您正在处理更复杂的 API。

我希望这会有所帮助。