Update update_attempt.py

fixed bug
treasure-data · May 28, 2024 · 24f5dfe · 24f5dfe
1 parent a0d1bb3
commit 24f5dfe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scenarios/monitoring/workflow_monitoring/scripts/update_attempt.py b/scenarios/monitoring/workflow_monitoring/scripts/update_attempt.py
@@ -44,10 +44,10 @@ def insert_attempt_info(import_unixtime, endpoint, apikey, dest_db, dest_table,
   client.load_table_from_dataframe(df, dest_table, if_exists='append', fmt='msgpack')
 
 def run(session_unixtime, dest_db, dest_table, ids, api_endpoint='api.treasuredata.com', workflow_endpoint='api-workflow.treasuredata.com'):
-  id_list = ids.split(',')
-  if len(id_list) == 0:
+  if len(ids) == 0:
     print('no update record')
     return
+  id_list = ids.split(',')
   delete_attempt_info(api_endpoint, os.environ['TD_API_KEY'], dest_db, dest_table, id_list)
 
   workflow_url = 'https://%s/api/attempts' % workflow_endpoint + '/%s'
@@ -56,4 +56,4 @@ def run(session_unixtime, dest_db, dest_table, ids, api_endpoint='api.treasureda
   if len(l) == 0:
     print('no update record')
     return
-  insert_attempt_info(session_unixtime, 'https://%s' % api_endpoint, os.environ['TD_API_KEY'], dest_db, dest_table, l)
+  insert_attempt_info(session_unixtime, 'https://%s' % api_endpoint, os.environ['TD_API_KEY'], dest_db, dest_table, l)