You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.0 KiB
70 lines
2.0 KiB
# 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)
|
|
|