def _get_vm(hostname, unlock=True, allow_retired=False):
"""Get a server from Serveradmin by hostname to return VM object
The function is accepting hostnames in any length as long as it resolves
to a single server on Serveradmin.
"""
object_id = Query({
'hostname': Any(hostname, StartsWith(hostname + '.')),
'servertype': 'vm',
}, ['object_id']).get()['object_id']
def vm_query():
return Query({
'object_id': object_id,
}, VM_ATTRIBUTES).get()
dataset_obj = vm_query()
hypervisor = None
if dataset_obj['hypervisor']:
hypervisor = Hypervisor(dataset_obj['hypervisor'])
# XXX: Ugly hack until adminapi supports modifying joined objects
dict.__setitem__(
dataset_obj, 'hypervisor', dataset_obj['hypervisor']['hostname']
)
vm = VM(dataset_obj, hypervisor)
vm.acquire_lock()
try:
if not allow_retired and dataset_obj['state'] == 'retired':
raise InvalidStateError(
'VM {} is in state retired, I refuse to work on it!'.format(
hostname,
)
)
yield vm
except (Exception, KeyboardInterrupt):
VM(vm_query(), hypervisor).release_lock()
raise
else:
# We re-fetch the VM because we can't risk commiting any other changes
# to the VM than unlocking. There can be changes from failed things,
# like setting memory.
# Most operations require unlocking, the only exception is deleting of
# a VM. After object is deleted, it can't be unlocked.
if unlock:
VM(vm_query(), hypervisor).release_lock()
def autocomplete(request):
autocomplete_list = []
if 'hostname' in request.GET:
hostname = request.GET['hostname']
try:
query = Query({'hostname': StartsWith(hostname)}, ['hostname'])
autocomplete_list += islice((h['hostname'] for h in query), 100)
except (DatatypeError, ValidationError):
pass # If there is no valid query, just don't auto-complete
return HttpResponse(
json.dumps({'autocomplete': autocomplete_list}),
content_type='application/x-json',
)
def test_startswith_servertype(self):
q = Query({'servertype': StartsWith('tes')})
self.assertEqual(len(q), 4)
def test_startswith(self):
s = Query({'os': StartsWith('whee')}).get()
self.assertEqual(s['hostname'], 'test0')