python - BigQuery - 如何导入 WRITE_TRUNCATE 进行大查询数据覆盖？

Question

我正在使用 python BigQuery 客户端 ( https://github.com/tylertreat/BigQuery-Python ) 将数据从 Google 云存储上传到表。我想覆盖数据，因此需要将 writeDisposition 设置为 WRITE_TRUNCATE。我应该从哪里导入 WRITE_TRUNCATE？

job = client.import_data_from_uris( gs_file_path,
                               'dataset_name',
                               'table_name', 
                               schema, 
                               source_format=JOB_SOURCE_FORMAT_CSV,
                               writeDisposition=WRITE_TRUNCATE,
                               field_delimiter='\t')

这是当前错误 - NameError:名称“WRITE_TRUNCATE”未定义。

此外，我有兴趣了解其他类似名称常量的导入。

Answer 1

也许在你运行任何东西之前:

from bigquery.client import JOB_WRITE_TRUNCATE

然后运行:

job = client.import_data_from_uris( gs_file_path,
                                   'dataset_name',
                                   'table_name', 
                                   schema, 
                                   source_format=JOB_SOURCE_FORMAT_CSV,
                                   writeDisposition=JOB_WRITE_TRUNCATE,
                                   field_d

elimiter='\t')

它可能已经适合你了。不过，我认为自己定义这个变量并根据需要使用它可能会以相同的方式工作并且更简单一些，如下所示:

write_disposition = 'WRITE_TRUNCATE'
job = client.import_data_from_uris( gs_file_path,
                               'dataset_name',
                               'table_name', 
                               schema, 
                               source_format=JOB_SOURCE_FORMAT_CSV,
                               writeDisposition=write_disposition,
                               field_delimiter='\t')

如果您想要一些附加操作，只需将值更改为 write_disposition='WRITE_APPEND' 即可。