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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
#!/usr/bin/python
import sys
import os
import locale
from subprocess import *
def get_memory_usage():
memusg = {}
# Get memory usage information, according to /proc/meminfo
f = open('/proc/meminfo', 'r')
swap_free = 0
mem_physical_free = 0
mem_buffers = 0
mem_cached = 0
for line in f.readlines():
tokens = line.split()
label = tokens[0]
size = tokens[1]
try:
size = int(size)
except:
# The line is an header, skip it.
continue
# We approximate kb to bytes
size = size * 1024
if label == 'MemTotal:':
memusg['ram_total'] = size
elif label == 'MemFree:':
mem_physical_free = size
elif label == 'Buffers:':
mem_buffers = size
elif label == 'Cached:':
mem_cached = size
elif label == 'SwapTotal:':
memusg['swap_total'] = size
elif label == 'SwapFree:':
swap_free = size
f.close()
memusg['ram_used'] = memusg['ram_total'] - mem_physical_free - mem_buffers - mem_cached
memusg['swap_used'] = memusg['swap_total'] - swap_free
return memusg
def get_load_average():
# Gets the current system load, according to /proc/loadavg
loadavg = {}
load_file = open('/proc/loadavg')
load_infos = load_file.read().split()
loadavg['one_min_avg'] = load_infos[0]
loadavg['five_min_avg'] = load_infos[1]
loadavg['fifteen_min_avg'] = load_infos[2]
# scheduling_info = load_infos[3] # not used
# last_pid = load_infos[4]
load_file.close()
return loadavg
def compute_server_rating():
"""Compute the server rating from it's state
The rating is computed by using load average and the memory
used. The returned value is betweed 0 and 100, higher is better
"""
max_acceptable_load_avg = 8.0
mem = get_memory_usage()
load = get_load_average()
rating = 100 - int( \
50 * ( float(load['fifteen_min_avg']) / max_acceptable_load_avg ) + \
50 * ( float(mem['ram_used']) / float(mem['ram_total']) ) \
)
if rating < 0:
rating = 0
return rating
def get_sessions (dir):
"""Get a list of available sessions.
Returns a list of sessions gathered from .desktop files
"""
sessions = []
if os.path.isdir(dir):
for f in os.listdir(dir):
if f.endswith('.desktop') and os.path.isfile(os.path.join(dir, f)):
x=dict()
for line in file(os.path.join(dir, f), 'r').readlines():
if line.count('Exec=') > 0:
variable, value = line.split('=')
x[variable]=value
if x.has_key('TryExec'):
sessions.append(x['TryExec'].rstrip())
elif x.has_key('Exec'):
sessions.append(x['Exec'].rstrip())
return sessions
def get_sessions_with_names (dir):
"""Get a list of available sessions with their name.
Returns a list of sessions gathered from .desktop files
"""
sessions = []
if os.path.isdir(dir):
for f in os.listdir(dir):
if f.endswith('.desktop') and os.path.isfile(os.path.join(dir, f)):
x=dict()
for line in file(os.path.join(dir, f), 'r').readlines():
if (line.count('Exec=') > 0) or (line.count("Name=") > 0):
variable, value = line.split('=')
x[variable]=value
if x.has_key('TryExec'):
if x.has_key('Name'):
thing = x['Name'].rstrip()
else:
thing = f.replace(".desktop", "")
sessions.append(thing + ":" + x['TryExec'].rstrip())
elif x.has_key('Exec'):
if x.has_key('Name'):
thing = x['Name'].rstrip()
else:
thing = f.replace(".desktop", "")
sessions.append(thing + ":" + x['Exec'].rstrip())
return sessions
def get_xsession():
"""Return the full path to the default Xsession script"""
xsessionlist=("/etc/X11/xinit/Xsession",
"/etc/X11/Xsession",
"/usr/lib/X11/xdm/Xsession",
"/etc/X11/xdm/Xsession")
for xsession in xsessionlist:
# check if file exists and is executable
if os.access(xsession, 5):
return xsession
return None
if __name__ == "__main__":
# Get the server's default locale
# We want it to appear first in the list
try:
lines = Popen(['locale'], stdout=PIPE).communicate()[0]
except OSError:
print "ERROR: failed to run locale"
sys.exit(0)
for line in lines.split():
if line.startswith('LANG='):
defaultlocale = line.split('=')[1].strip('"')
defaultlocale = defaultlocale.replace('UTF8', 'UTF-8')
print "language:" + defaultlocale
# Get list of valid locales from locale -a
try:
lines = Popen(['locale', '-a'], stdout=PIPE).communicate()[0]
except OSError:
print "ERROR"
sys.exit(0)
langs = lines.split(None)
langs.sort()
for lang in langs:
lang = lang.rstrip()
if lang.endswith('.utf8'):
# locale returns .utf8 when we want .UTF-8
lang = lang.replace('.utf8','.UTF-8')
else:
# if locale did not end with .utf8, do not add to list
continue
if lang != 'POSIX' and lang != 'C' and lang != defaultlocale:
print "language:" + lang
try:
lines = get_sessions('/usr/share/xsessions/')
except OSError:
print "ERROR"
sys.exit(0)
for line in lines:
print "session:" + line
try:
lines = get_sessions_with_names('/usr/share/xsessions/')
except OSError:
print "ERROR"
sys.exit(0)
for line in lines:
print "session-with-name:" + line
try:
xsession = get_xsession()
except:
print "ERROR"
sys.exit(0)
if xsession:
print "xsession:" + xsession
# Get the rating of this server
rate = 0
try:
rate = compute_server_rating()
except:
print "ERROR"
sys.exit(0)
print "rating:" + str(compute_server_rating())
|