mirror of
https://github.com/vmware/vsphere-automation-sdk-python.git
synced 2024-11-22 01:39:58 -05:00
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
|
"""
|
||
|
* *******************************************************
|
||
|
* Copyright (c) VMware, Inc. 2017. All Rights Reserved.
|
||
|
* SPDX-License-Identifier: MIT
|
||
|
* *******************************************************
|
||
|
*
|
||
|
* DISCLAIMER. THIS PROGRAM IS PROVIDED TO YOU "AS IS" WITHOUT
|
||
|
* WARRANTIES OR CONDITIONS OF ANY KIND, WHETHER ORAL OR WRITTEN,
|
||
|
* EXPRESS OR IMPLIED. THE AUTHOR SPECIFICALLY DISCLAIMS ANY IMPLIED
|
||
|
* WARRANTIES OR CONDITIONS OF MERCHANTABILITY, SATISFACTORY QUALITY,
|
||
|
* NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE.
|
||
|
"""
|
||
|
|
||
|
__author__ = 'VMware, Inc.'
|
||
|
|
||
|
from time import sleep
|
||
|
from tabulate import tabulate
|
||
|
|
||
|
from com.vmware.vmc.model_client import Task
|
||
|
|
||
|
|
||
|
def wait_for_task(task_client, org_id, task_id, interval_sec=60):
|
||
|
"""
|
||
|
Helper method to wait for a task to finish
|
||
|
:param task_client: task client to query the task object
|
||
|
:param org_id: organization id
|
||
|
:param task_id: task id
|
||
|
:param interval_sec: task pulling interval_sec in sec
|
||
|
:return: True if task finished successfully, False otherwise.
|
||
|
"""
|
||
|
print('Wait for task {} to finish'.format(task_id))
|
||
|
print('Checking task status every {} seconds'.format(interval_sec))
|
||
|
|
||
|
while True:
|
||
|
task = task_client.get(org_id, task_id)
|
||
|
|
||
|
if task.status == Task.STATUS_FINISHED:
|
||
|
print('\nTask {} finished successfully'.format(task_id))
|
||
|
return True
|
||
|
elif task.status == Task.STATUS_FAILED:
|
||
|
print('\nTask {} failed'.format(task_id))
|
||
|
return False
|
||
|
elif task.status == Task.STATUS_CANCELED:
|
||
|
print('\nTask {} cancelled'.format(task_id))
|
||
|
return False
|
||
|
else:
|
||
|
print("Estimated time remaining: {} minutes".
|
||
|
format(task.estimated_remaining_minutes))
|
||
|
sleep(interval_sec)
|
||
|
|
||
|
|
||
|
def list_all_tasks(task_client, org_id):
|
||
|
"""
|
||
|
List all tasks in a given org
|
||
|
:param task_client: task client to query the task object
|
||
|
:param org_id: organization id
|
||
|
"""
|
||
|
tasks = task_client.list(org_id)
|
||
|
headers = ['ID', 'Status', 'Progress', 'Started', 'User']
|
||
|
table = []
|
||
|
for task in tasks:
|
||
|
table.append([task.id, task.status, task.progress_percent,
|
||
|
task.start_time, task.user_name])
|
||
|
print(tabulate(table, headers))
|