Übergabe von Array-Parameter an SQL für BigQuery in Python

Question

Ich habe eine Reihe von IDs (~ 200k) und ich muss alle Zeilen in einer BigQuery-Tabelle mit diesen IDs abrufen. Ich habe versucht, eine Liste in Python zu konstruieren und übergeben sie als Parameter an die SQL-Abfrage mit @, aber ich bekomme TypeError: 'ArrayQueryParameter' object is not iterable Fehler. Hier ist der Code Ich habe versucht (sehr ähnlich https://cloud.google.com/bigquery/querying-data#running_parameterized_queries):

id_list = ['id1', 'id2'] 
query = """ 
    SELECT id 
    FROM `my-db` 
    WHERE id in UNNEST(@ids) 
""" 
query_job = client.run_async_query(
    str(uuid.uuid4()), 
    query, 
    query_parameters=(
     bigquery.ArrayQueryParameter('ids', 'ARRAY<STRING>', id_list) 
    ) 
) 

Answer 1

Wahrscheinlich ist das Problem hier, dass Sie kein Tupel an die Funktion übergeben.

Versuchen ein Komma vor dem Schließen der Klammer hinzufügen, etwa so:

id_list = ['id1', 'id2'] 
query = """ 
    SELECT id 
    FROM `my-db` 
    WHERE id in UNNEST(@ids) 
""" 
query_job = client.run_async_query(
    str(uuid.uuid4()), 
    query, 
    query_parameters=(
     bigquery.ArrayQueryParameter('ids', 'STRING', id_list), 
    ) 
) 

In Python, wenn Sie tun:

t = (1) 

und dann laufen:

type(t) 

Sie werden feststellen, Das Ergebnis ist int. Aber wenn Sie das tun:

t = (1,) 

Dann ergibt sich ein Tupel.

Answer 2

Sie müssen 'STRING' verwenden, anstatt 'ARRAY<STRING>' für den Array-Element-Typen, zB:

query_parameters=(
    bigquery.ArrayQueryParameter('ids', 'STRING', id_list) 

Das Beispiel aus dem querying data Thema :

def query_array_params(gender, states): 
    client = bigquery.Client() 
    query = """ 
     SELECT name, sum(number) as count 
     FROM `bigquery-public-data.usa_names.usa_1910_2013` 
     WHERE gender = @gender 
     AND state IN UNNEST(@states) 
     GROUP BY name 
     ORDER BY count DESC 
     LIMIT 10; 
     """ 
    query_job = client.run_async_query(
     str(uuid.uuid4()), 
     query, 
     query_parameters=(
      bigquery.ScalarQueryParameter('gender', 'STRING', gender), 
      bigquery.ArrayQueryParameter('states', 'STRING', states))) 
    query_job.use_legacy_sql = False 

    # Start the query and wait for the job to complete. 
    query_job.begin() 
    wait_for_job(query_job) 
    print_results(query_job.results())