#!/usr/bin/python
from ovirtsdk.api import API
from ovirtsdk.xml import params
import time

user = 'admin@internal'
password = '123456'
host = 'https://10.35.161.55/api'


disk_alias = 'lsm_Disk1'
SD_A = 'fcp_0'
SD_B = 'fcp_1'

def get_disk_to_move(disks):
    return [d for d in disks if d.get_alias() == disk_alias][0]

def wait_for_locked_disk():
    disk_status = 'locked'
    print("Waiting for disk status to reach OK")
    while disk_status == 'locked':
        time.sleep(10)
        disks = api.disks.list()
        disk_status = get_disk_to_move(disks).get_status().get_state()
    print("Disk status is OK")



api = API (url=host, username=user, password=password, insecure=True)

print "Connected to %s successfully!" % api.get_product_info().name
disks = [d for d in api.disks.list()]
disk_to_move = get_disk_to_move(disks)
for i in xrange(100):
    print("Moving disk from domain %s to domain %s", SD_A, SD_B)
    disk_to_move.move(params.Action(storage_domain=params.StorageDomain(name=SD_B)))
    wait_for_locked_disk()
    print("Moving disk from domain %s to domain %s", SD_B, SD_A)
    disk_to_move.move(params.Action(storage_domain=params.StorageDomain(name=SD_A)))
    wait_for_locked_disk()