supports basic listing of pcus, nodes, and some coloring from the old scripts.
[monitor.git] / nodecommon.py
1
2 import time
3 import struct
4 from pcucontrol import reboot
5
6 from monitor import util
7 from monitor import database
8 from monitor.wrapper import plc, plccache
9
10 from datetime import datetime 
11 from monitor.model import PersistFlags
12
13 esc = struct.pack('i', 27)
14 RED     = esc + "[1;31m"
15 GREEN   = esc + "[1;32m"
16 YELLOW  = esc + "[1;33m"
17 BLUE    = esc + "[1;34m"
18 LIGHTBLUE       = esc + "[1;36m"
19 NORMAL  = esc + "[0;39m"
20
21 def red(str):
22         return RED + str + NORMAL
23
24 def yellow(str):
25         return YELLOW + str + NORMAL
26
27 def green(str):
28         return GREEN + str + NORMAL
29
30 def lightblue(str):
31         return LIGHTBLUE + str + NORMAL
32
33 def blue(str):
34         return BLUE + str + NORMAL
35
36 def get_current_state(fbnode):
37         if 'observed_status' in fbnode:
38                 state = fbnode['observed_status']
39         else:
40                 state = "none"
41         l = state.lower()
42         if l == "debug": l = 'dbg '
43         return l
44
45 def color_pcu_state(fbnode):
46
47         if 'plcnode' in fbnode and 'pcu_ids' in fbnode['plcnode'] and len(fbnode['plcnode']['pcu_ids']) > 0 :
48                 values = reboot.get_pcu_values(fbnode['plcnode']['pcu_ids'][0])
49                 if values == None:
50                         return fbnode['pcu']
51         else:
52                 if 'pcu' not in fbnode:
53                         return 'NOPCU'
54                 else:
55                         return fbnode['pcu']
56
57         if 'reboot' in values:
58                 rb = values['reboot']
59                 if rb == 0 or rb == "0":
60                         return fbnode['pcu'] + "OK  "
61                         #return fbnode['pcu'] + "OK  "
62                         #return green(fbnode['pcu'])
63                 elif "NetDown" == rb  or "Not_Run" == rb:
64                         return fbnode['pcu'] + "DOWN"
65                         #return yellow(fbnode['pcu'])
66                 else:
67                         return fbnode['pcu'] + "BAD "
68                         #return red(fbnode['pcu'])
69         else:
70                 #return red(fbnode['pcu'])
71                 return fbnode['pcu'] + "BAD "
72
73 def color_boot_state(l):
74         if    l == "dbg": return yellow("debg")
75         elif  l == "dbg ": return yellow("debg")
76         elif  l == "diag": return lightblue(l)
77         elif  l == "disable": return red("dsbl")
78         elif  l == "down": return red(l)
79         elif  l == "boot": return green(l)
80         elif  l == "rins": return blue(l)
81         else:
82                 return l
83
84 def diff_time(timestamp, abstime=True):
85         import math
86         now = time.time()
87         if timestamp == None:
88                 return "unknown"
89         if abstime:
90                 diff = now - timestamp
91         else:
92                 diff = timestamp
93         # return the number of seconds as a difference from current time.
94         t_str = ""
95         if diff < 60: # sec in min.
96                 t = diff / 1
97                 t_str = "%s sec ago" % int(math.ceil(t))
98         elif diff < 60*60: # sec in hour
99                 t = diff / (60)
100                 t_str = "%s min ago" % int(math.ceil(t))
101         elif diff < 60*60*24: # sec in day
102                 t = diff / (60*60)
103                 t_str = "%s hrs ago" % int(math.ceil(t))
104         elif diff < 60*60*24*14: # sec in week
105                 t = diff / (60*60*24)
106                 t_str = "%s days ago" % int(math.ceil(t))
107         elif diff <= 60*60*24*30: # approx sec in month
108                 t = diff / (60*60*24*7)
109                 t_str = "%s wks ago" % int(math.ceil(t))
110         elif diff > 60*60*24*30: # approx sec in month
111                 t = diff / (60*60*24*30)
112                 t_str = "%s mnths ago" % int(t)
113         return t_str
114
115 def getvalue(fb, path):
116     indexes = path.split("/")
117     values = fb
118     for index in indexes:
119         if index in values:
120             values = values[index]
121         else:
122             return None
123     return values
124
125 def nodegroup_display(node, fbdata, conf=None):
126         node['current'] = get_current_state(fbdata)
127
128         s = fbdata['kernel_version'].split()
129         if len(s) >=3:
130                 node['kernel_version'] = s[2]
131         else:
132                 node['kernel_version'] = fbdata['kernel_version']
133                 
134         if '2.6' not in node['kernel_version']: node['kernel_version'] = ""
135         if conf and not conf.nocolor:
136             node['boot_state']  = color_boot_state(node['boot_state'])
137             node['current']     = color_boot_state(node['current'])
138
139         if type(fbdata['plc_node_stats']['pcu_ids']) == type([]):
140                 node['pcu'] = "PCU"
141         node['lastupdate'] = diff_time(node['last_contact'])
142
143         pf = PersistFlags(node['hostname'], 1, db='node_persistflags')
144         try:
145                 node['lc'] = diff_time(pf.last_changed)
146         except:
147                 node['lc'] = "err"
148
149         ut = fbdata['comon_stats']['uptime']
150         if ut != "null":
151                 ut = diff_time(float(fbdata['comon_stats']['uptime']), False)
152         node['uptime'] = ut
153
154         return "%(hostname)-42s %(boot_state)8s %(current)5s %(pcu)6s %(key)10.10s... %(kernel_version)35.35s %(lastupdate)12s, %(lc)s, %(uptime)s" % node
155
156 def datetime_fromstr(str):
157         if '-' in str:
158                 try:
159                         tup = time.strptime(str, "%Y-%m-%d")
160                 except:
161                         tup = time.strptime(str, "%Y-%m-%d-%H:%M")
162         elif '/' in str:
163                 tup = time.strptime(str, "%m/%d/%Y")
164         else:
165                 tup = time.strptime(str, "%m/%d/%Y")
166         ret = datetime.fromtimestamp(time.mktime(tup))
167         return ret
168
169 def get_nodeset(config):
170         """
171                 Given the config values passed in, return the set of hostnames that it
172                 evaluates to.
173         """
174         api = plc.getAuthAPI()
175         l_nodes = plccache.l_nodes
176
177         if config.nodelist:
178                 f_nodes = util.file.getListFromFile(config.nodelist)
179                 l_nodes = filter(lambda x: x['hostname'] in f_nodes, l_nodes)
180         elif config.node:
181                 f_nodes = [config.node]
182                 l_nodes = filter(lambda x: x['hostname'] in f_nodes, l_nodes)
183         elif config.nodegroup:
184                 ng = api.GetNodeGroups({'name' : config.nodegroup})
185                 l_nodes = api.GetNodes(ng[0]['node_ids'], ['hostname'])
186         elif config.site:
187                 site = api.GetSites(config.site)
188                 l_nodes = api.GetNodes(site[0]['node_ids'], ['hostname'])
189                 
190         l_nodes = [node['hostname'] for node in l_nodes]
191
192         # perform this query after the above options, so that the filter above
193         # does not break.
194         if config.nodeselect:
195                 fbquery = FindbadNodeRecord.get_all_latest()
196                 node_list = [ n.hostname for n in fbquery ]
197                 l_nodes = node_select(config.nodeselect, node_list, None)
198
199         return l_nodes
200