70 lines
2 KiB
Python
70 lines
2 KiB
Python
# Copyright 2010 Google Inc. All Rights Reserved.
|
|
|
|
__author__ = 'asharif@google.com (Ahmad Sharif)'
|
|
|
|
from fnmatch import fnmatch
|
|
|
|
|
|
class Machine(object):
|
|
"""Stores information related to machine and its state."""
|
|
|
|
def __init__(self, hostname, label, cpu, cores, os, username):
|
|
self.hostname = hostname
|
|
self.label = label
|
|
self.cpu = cpu
|
|
self.cores = cores
|
|
self.os = os
|
|
self.username = username
|
|
|
|
# MachineManager related attributes.
|
|
self.uses = 0
|
|
self.locked = False
|
|
|
|
def Acquire(self, exclusively):
|
|
assert not self.locked
|
|
|
|
if exclusively:
|
|
self.locked = True
|
|
self.uses += 1
|
|
|
|
def Release(self):
|
|
assert self.uses > 0
|
|
|
|
self.uses -= 1
|
|
|
|
if not self.uses:
|
|
self.locked = False
|
|
|
|
def __repr__(self):
|
|
return '{%s: %s@%s}' % (self.__class__.__name__, self.username,
|
|
self.hostname)
|
|
|
|
def __str__(self):
|
|
return '\n'.join(
|
|
['Machine Information:', 'Hostname: %s' % self.hostname, 'Label: %s' %
|
|
self.label, 'CPU: %s' % self.cpu, 'Cores: %d' % self.cores, 'OS: %s' %
|
|
self.os, 'Uses: %d' % self.uses, 'Locked: %s' % self.locked])
|
|
|
|
|
|
class MachineSpecification(object):
|
|
"""Helper class used to find a machine matching your requirements."""
|
|
|
|
def __init__(self, hostname='*', label='*', os='*', lock_required=False):
|
|
self.hostname = hostname
|
|
self.label = label
|
|
self.os = os
|
|
self.lock_required = lock_required
|
|
self.preferred_machines = []
|
|
|
|
def __str__(self):
|
|
return '\n'.join(['Machine Specification:', 'Name: %s' % self.name, 'OS: %s'
|
|
% self.os, 'Lock required: %s' % self.lock_required])
|
|
|
|
def IsMatch(self, machine):
|
|
return all([not machine.locked, fnmatch(machine.hostname, self.hostname),
|
|
fnmatch(machine.label, self.label), fnmatch(machine.os,
|
|
self.os)])
|
|
|
|
def AddPreferredMachine(self, hostname):
|
|
if hostname not in self.preferred_machines:
|
|
self.preferred_machines.append(hostname)
|