在Google BigQuery中，如何使用Google Python客户端使用JavaScript UDF

Question

我正在使用标准SQL和JavaScript UDF在BigQuery编写查询。我可以使用WebUI和BigQuery命令行工具实现此问题，但我的要求是使用Google Python客户端进行此查询。无法实现这一目标。请有人帮助。

from google.cloud import bigquery
bigquery_client = bigquery.Client()
client = bigquery.Client()
query_results = client.run_sync_query("""
CREATE TEMPORARY FUNCTION CategoriesToNumerical(a array<STRING>,b    array<STRING>)
RETURNS string
LANGUAGE js AS """
var values = {};
 var counter = 0;
 for(i=0;i<a.length;i++)
 { var temp;
   temp = a[i];
   a[i] = counter;
   values[temp] = counter;
   counter ++;
  }
for(i=0;i<b.length;i++)
  {
for(var key in values)
  {
   if(b[i] == key)
   {
    b[i] = values[key];
   }
  }
}
return b;
""";
SELECT
CategoriesToNumerical(ARRAY(SELECT DISTINCT ProspectStage from   lsq.lsq_dest),ARRAY(SELECT ProspectStage from lsq.lsq_dest)) as prospectstageds
;""")
query_results.use_legacy_sql = False
query_results.run()
page_token = None
while True:
    rows1, total_rows, page_token = query_results.fetch_data(
        max_results=100,
        page_token=page_token)
    for row1 in rows1:
        print "row",row1
    if not page_token:
        break

这对我不起作用。请有人可以帮助我如何处理。

Answer 1

问题似乎是您在这里有2组冲突的"。将其中一组替换为三重'''，并且代码应起作用。

所以而不是

query_results = client.run_sync_query("""
CREATE TEMPORARY FUNCTION CategoriesToNumerical(a array<STRING>,b    array<STRING>)
RETURNS string
LANGUAGE js AS """
  javacript code
"""
SELECT *
FROM 
"""

写

query_results = client.run_sync_query('''
CREATE TEMPORARY FUNCTION CategoriesToNumerical(a array<STRING>,b    array<STRING>)
RETURNS string
LANGUAGE js AS """
  javacript code
"""
SELECT *
FROM 
'''