7 from datetime import datetime
8 from monitor import database
9 from unified_model import PersistFlags
10 esc = struct.pack('i', 27)
12 GREEN = esc + "[1;32m"
13 YELLOW = esc + "[1;33m"
15 LIGHTBLUE = esc + "[1;36m"
16 NORMAL = esc + "[0;39m"
19 return RED + str + NORMAL
22 return YELLOW + str + NORMAL
25 return GREEN + str + NORMAL
28 return LIGHTBLUE + str + NORMAL
31 return BLUE + str + NORMAL
33 def get_current_state(fbnode):
35 state = fbnode['state']
39 if l == "debug": l = 'dbg '
42 def color_pcu_state(fbnode):
44 if 'plcnode' in fbnode and 'pcu_ids' in fbnode['plcnode'] and len(fbnode['plcnode']['pcu_ids']) > 0 :
45 values = reboot.get_pcu_values(fbnode['plcnode']['pcu_ids'][0])
49 if 'pcu' not in fbnode:
54 if 'reboot' in values:
56 if rb == 0 or rb == "0":
57 return fbnode['pcu'] + "OK "
58 #return fbnode['pcu'] + "OK "
59 #return green(fbnode['pcu'])
60 elif "NetDown" == rb or "Not_Run" == rb:
61 return fbnode['pcu'] + "DOWN"
62 #return yellow(fbnode['pcu'])
64 return fbnode['pcu'] + "BAD "
65 #return red(fbnode['pcu'])
67 #return red(fbnode['pcu'])
68 return fbnode['pcu'] + "BAD "
70 def color_boot_state(l):
71 if l == "dbg": return yellow("debg")
72 elif l == "dbg ": return yellow("debg")
73 elif l == "diag": return lightblue(l)
74 elif l == "disable": return red("dsbl")
75 elif l == "down": return red(l)
76 elif l == "boot": return green(l)
77 elif l == "rins": return blue(l)
78 elif l == "reinstall": return blue(l)
82 def diff_time(timestamp, abstime=True):
88 diff = now - timestamp
91 # return the number of seconds as a difference from current time.
93 if diff < 60: # sec in min.
95 t_str = "%s sec ago" % int(math.ceil(t))
96 elif diff < 60*60: # sec in hour
98 t_str = "%s min ago" % int(math.ceil(t))
99 elif diff < 60*60*24: # sec in day
101 t_str = "%s hrs ago" % int(math.ceil(t))
102 elif diff < 60*60*24*14: # sec in week
103 t = diff / (60*60*24)
104 t_str = "%s days ago" % int(math.ceil(t))
105 elif diff <= 60*60*24*30: # approx sec in month
106 t = diff / (60*60*24*7)
107 t_str = "%s wks ago" % int(math.ceil(t))
108 elif diff > 60*60*24*30: # approx sec in month
109 t = diff / (60*60*24*30)
110 t_str = "%s mnths ago" % int(t)
113 def getvalue(fb, path):
114 indexes = path.split("/")
116 for index in indexes:
118 values = values[index]
123 def nodegroup_display(node, fb, conf=None):
124 if node['hostname'] in fb['nodes']:
125 node['current'] = get_current_state(fb['nodes'][node['hostname']]['values'])
127 node['current'] = 'none'
129 if fb['nodes'][node['hostname']]['values'] == []:
132 s = fb['nodes'][node['hostname']]['values']['kernel'].split()
134 node['kernel'] = s[2]
136 node['kernel'] = fb['nodes'][node['hostname']]['values']['kernel']
138 if '2.6' not in node['kernel']: node['kernel'] = ""
139 if conf and not conf.nocolor:
140 node['boot_state'] = color_boot_state(node['boot_state'])
141 node['current'] = color_boot_state(node['current'])
142 #node['boot_state'] = node['boot_state']
143 #node['current'] = node['current']
144 node['pcu'] = fb['nodes'][node['hostname']]['values']['pcu']
145 node['lastupdate'] = diff_time(node['last_contact'])
146 pf = PersistFlags(node['hostname'], 1, db='node_persistflags')
148 node['lc'] = diff_time(pf.last_changed)
151 ut = fb['nodes'][node['hostname']]['values']['comonstats']['uptime']
153 ut = diff_time(float(fb['nodes'][node['hostname']]['values']['comonstats']['uptime']), False)
156 return "%(hostname)-42s %(boot_state)8s %(current)5s %(pcu)6s %(key)10.10s... %(kernel)35.35s %(lastupdate)12s, %(lc)s, %(uptime)s" % node
158 def datetime_fromstr(str):
161 tup = time.strptime(str, "%Y-%m-%d")
163 tup = time.strptime(str, "%Y-%m-%d-%H:%M")
165 tup = time.strptime(str, "%m/%d/%Y")
167 tup = time.strptime(str, "%m/%d/%Y")
168 ret = datetime.fromtimestamp(time.mktime(tup))
171 def get_nodeset(config):
173 Given the config values passed in, return the set of hostnames that it
176 api = plc.getAuthAPI()
177 l_nodes = database.dbLoad("l_plcnodes")
180 f_nodes = util.file.getListFromFile(config.nodelist)
181 l_nodes = filter(lambda x: x['hostname'] in f_nodes, l_nodes)
183 f_nodes = [config.node]
184 l_nodes = filter(lambda x: x['hostname'] in f_nodes, l_nodes)
185 elif config.nodegroup:
186 ng = api.GetNodeGroups({'groupname' : config.nodegroup})
187 l_nodes = api.GetNodes(ng[0]['node_ids'], ['hostname'])
189 site = api.GetSites(config.site)
190 l_nodes = api.GetNodes(site[0]['node_ids'], ['hostname'])
192 l_nodes = [node['hostname'] for node in l_nodes]
194 # perform this query after the above options, so that the filter above
196 if config.nodeselect:
197 fb = database.dbLoad("findbad")
198 l_nodes = node_select(config.nodeselect, fb['nodes'].keys(), fb)
202 def email_exception(content=None):
204 from unified_model import Message
206 msg=traceback.format_exc()
208 msg = content + "\n" + msg
209 m=Message("exception running monitor", msg, False)
210 m.send([config.cc_email])