def download()

in download.py [0:0]

• 13 lines of code
• 4 McCabe index (conditional complexity)

def download(query: str, for_date: date, release_version: int) -> tuple[bigquery.table.RowIterator, bigquery.table.RowIterator]:
    client = bigquery.Client(PROJECT)
    job_config = bigquery.QueryJobConfig(
        query_parameters = [
            bigquery.ScalarQueryParameter("date", bigquery.SqlParameterScalarTypes.DATE, for_date),
            bigquery.ScalarQueryParameter("release_version", bigquery.SqlParameterScalarTypes.INT64, release_version),
        ],
    )
    query_job = client.query(query, job_config = job_config)
    # Wait for job to complete (to ensure all child jobs are present)
    query_job.result()
    child_jobs = client.list_jobs(parent_job = query_job)
    [pings_job, config_job] = [job for job in child_jobs if job.statement_type == "SELECT"]
    return config_job.result(), pings_job.result()