File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -1048,7 +1048,12 @@ def get_data_row_ids_for_global_keys(self,
10481048 get_job_params )
10491049
10501050 get_data_rows_str = """query getDataRowsForGlobalKeysPyApi($jobId: ID!) {
1051- dataRowsForGlobalKeysResult(jobId: {id: $jobId}) { data { fetchedDataRows {id}} jobStatus}}
1051+ dataRowsForGlobalKeysResult(jobId: {id: $jobId}) { data {
1052+ fetchedDataRows {id}
1053+ notFoundGlobalKeys
1054+ accessDeniedGlobalKeys
1055+ deletedDataRowGlobalKeys
1056+ } jobStatus}}
10521057 """
10531058 get_data_rows_params = {
10541059 "jobId" :
@@ -1058,6 +1063,7 @@ def get_data_row_ids_for_global_keys(self,
10581063 while timeout >= 0 :
10591064 res = self .execute (get_data_rows_str , get_data_rows_params )
10601065 if res ["dataRowsForGlobalKeysResult" ]['jobStatus' ] == "COMPLETE" :
1066+ # TODO: should consider deleted, not found, global keys return?
10611067 return res ["dataRowsForGlobalKeysResult" ]['data' ][
10621068 'fetchedDataRows' ]
10631069 time .sleep (2 )
You can’t perform that action at this time.
0 commit comments