# Copyright 2015 The Chromium OS Authors. All rights reserved.
|
"""Utilities for locking machines."""
|
|
from __future__ import print_function
|
|
import time
|
|
import afe_lock_machine
|
|
import logger
|
|
|
def AcquireLock(machines, chromeos_root, timeout=1200):
|
"""Acquire lock for machine(s) with timeout, using AFE server for locking."""
|
start_time = time.time()
|
locked = True
|
sleep_time = min(10, timeout / 10.0)
|
while True:
|
try:
|
afe_lock_machine.AFELockManager(machines, False, chromeos_root,
|
None).UpdateMachines(True)
|
break
|
except Exception as e:
|
if time.time() - start_time > timeout:
|
locked = False
|
logger.GetLogger().LogWarning(
|
'Could not acquire lock on {0} within {1} seconds: {2}'.format(
|
repr(machines), timeout, str(e)))
|
break
|
time.sleep(sleep_time)
|
return locked
|
|
|
def ReleaseLock(machines, chromeos_root):
|
"""Release locked machine(s), using AFE server for locking."""
|
unlocked = True
|
try:
|
afe_lock_machine.AFELockManager(machines, False, chromeos_root,
|
None).UpdateMachines(False)
|
except Exception as e:
|
unlocked = False
|
logger.GetLogger().LogWarning('Could not unlock %s. %s' %
|
(repr(machines), str(e)))
|
return unlocked
|
