1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/python2.6
#
# 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)
|
