32 lines
974 B
Python
32 lines
974 B
Python
|
import time
|
|||
|
|
|
|||
|
|
|
|||
|
|
def wait_for_job_completion(client, batch_job_id, timeout=6000, interval=10):
|
|||
|
|
"""
|
|||
|
|
等待批处理作业完成。
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
- client: 批处理作业客户端对象
|
|||
|
|
- batch_job_id: 批处理作业的ID
|
|||
|
|
- timeout: 最大等待时间（秒）
|
|||
|
|
- interval: 检查状态的间隔时间（秒）
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
- True 如果作业完成，False 如果超时
|
|||
|
|
"""
|
|||
|
|
start_time = time.time()
|
|||
|
|
while time.time() - start_time < timeout:
|
|||
|
|
batch_job = client.batches.retrieve(batch_job_id)
|
|||
|
|
if batch_job.status == 'completed':
|
|||
|
|
print("Batch job completed successfully.")
|
|||
|
|
return True
|
|||
|
|
elif batch_job.status == 'failed':
|
|||
|
|
print("Batch job failed.")
|
|||
|
|
return False
|
|||
|
|
else:
|
|||
|
|
print(f"Current status: {batch_job.status}. Waiting for completion...")
|
|||
|
|
time.sleep(interval)
|
|||
|
|
|
|||
|
|
print("Timeout reached, job did not complete in time.")
|
|||
|
|
return False
|