在 python 编码的 Apache-Beam 管道中提供 BigQuery 凭证

Question

我正在尝试使用云数据流运行器从我的光束管道中的 bigquery 读取数据。 我想提供访问该项目的凭据。

我在 Java 中看到了示例，但在 Python 中没有看到。

我发现的唯一可能性是使用： --service_account_email参数但是如果我想在代码本身的所有选项中给出 .json 关键信息，例如：google_cloud_options.service_account = '/path/to/credential.json'

options = PipelineOptions(flags=argv)
google_cloud_options = options.view_as(GoogleCloudOptions)
google_cloud_options.project = 'project_name'
google_cloud_options.job_name = 'job_name'
google_cloud_options.staging_location = 'gs://bucket'
google_cloud_options.temp_location = 'gs://bucket'
options.view_as(StandardOptions).runner = 'DataflowRunner'

with beam.Pipeline(options=options) as pipeline:
    query = open('query.sql', 'r')
    bq_source = beam.io.BigQuerySource(query=query.read(), use_standard_sql=True)
    main_table = \
        pipeline \
        | 'ReadAccountViewAll' >> beam.io.Read(bq_source) \

Java 有一个方法getGcpCredential但在 Python 中找不到...

有任何想法吗？

Answer 1

--service_account_email是这里提到的推荐方法。 不建议下载密钥并将其存储在本地或 GCE 上。

Answer 2

对于需要在代码中为 json 文件使用不同路径的情况，您可以尝试以下python 身份验证解决方法：

client = Client.from_service_account_json('/path/to/keyfile.json')

或者

client = Client(credentials=credentials)

以下是从文件创建自定义凭据的示例：

credentials = service_account.Credentials.from_service_account_file(
    key_path,
    scopes=["https://www.googleapis.com/auth/cloud-platform"],
)