abbcee8bb78f44df9b3245131c5086f5875b7e82
[monitor.git] / nodehistory.py
1 #!/usr/bin/python
2
3 import plc
4 api = plc.getAuthAPI()
5
6 import database
7 import reboot
8 import time
9 from datetime import datetime, timedelta
10 import calendar
11
12 import sys
13 import time
14 from model import *
15 from nodecommon import *
16
17 def get_filefromglob(d, str):
18         import os
19         import glob
20         # TODO: This is aweful.
21         path = "archive-pdb"
22         archive = database.SPickle(path)
23         glob_str = "%s*.%s.pkl" % (d.strftime("%Y-%m-%d"), str)
24         os.chdir(path)
25         #print glob_str
26         file = glob.glob(glob_str)[0]
27         #print "loading %s" % file
28         os.chdir("..")
29         return file[:-4]
30         #fb = archive.load(file[:-4])
31
32
33 def fb_print_nodeinfo(fbnode, verbose, date=None):
34         if verbose: print "              state |  ssh  |  pcu  | bootcd | category | kernel"
35         if 'checked' in fbnode:
36                 print "%11.11s " % diff_time(fbnode['checked']),
37         else:
38                 if date: print date,
39                 else: print "Unknown",
40                 
41         if fbnode['bootcd']:
42                 fbnode['bootcd'] = fbnode['bootcd'].split()[-1]
43         else:
44                 fbnode['bootcd'] = "unknown"
45         fbnode['state'] = color_boot_state(get_current_state(fbnode))
46         if len(fbnode['kernel'].split()) >= 3:
47                 fbnode['kernel'] = fbnode['kernel'].split()[2]
48         print "    %(state)5s | %(ssh)5.5s | %(pcu)5.5s | %(bootcd)6.6s | %(category)8.8s | %(kernel)s" % fbnode
49
50 def pcu_print_info(pcuinfo, hostname):
51         print "   Checked: ",
52         if 'checked' in pcuinfo:
53                 print "%11.11s " % diff_time(pcuinfo['checked'])
54         else:
55                 print "Unknown"
56
57         print "\t            user   |          password | port | hostname "
58         print "\t %17s | %17s | %4s | %30s | %s" % \
59                 (pcuinfo['username'], pcuinfo['password'], 
60                  pcuinfo[hostname], reboot.pcu_name(pcuinfo), pcuinfo['model'])
61
62         if 'portstatus' in pcuinfo and pcuinfo['portstatus'] != {}:
63                 if pcuinfo['portstatus']['22'] == "open":
64                         print "\t ssh -o PasswordAuthentication=yes -o PubkeyAuthentication=no %s@%s" % (pcuinfo['username'], reboot.pcu_name(pcuinfo))
65                 if pcuinfo['portstatus']['23'] == "open":
66                         print "\t telnet %s" % (reboot.pcu_name(pcuinfo))
67                 if pcuinfo['portstatus']['80'] == "open" or \
68                         pcuinfo['portstatus']['443'] == "open":
69                         print "\t http://%s" % (reboot.pcu_name(pcuinfo))
70                 if pcuinfo['portstatus']['443'] == "open":
71                         print "\t racadm.py -r %s -u %s -p '%s'" % (pcuinfo['ip'], pcuinfo['username'], pcuinfo['password'])
72                         print "\t cmdhttps/locfg.pl -s %s -f iloxml/Reset_Server.xml -u %s -p '%s' | grep MESSAGE" % \
73                                 (reboot.pcu_name(pcuinfo), pcuinfo['username'], pcuinfo['password'])
74
75 def main():
76         import parser as parsermodule
77
78         parser = parsermodule.getParser()
79         parser.set_defaults(node=None, fields='state', fromtime=None)
80         parser.add_option("", "--node", dest="node", metavar="nodename.edu", 
81                                                 help="A single node name to add to the nodegroup")
82         parser.add_option("", "--fields", dest="fields", metavar="key",
83                                                 help="Which record field to extract from all files.")
84         parser.add_option("", "--fromtime", dest="fromtime", metavar="YYYY-MM-DD",
85                                                 help="Specify a starting date from which to begin the query.")
86         config = parsermodule.parse_args(parser)
87
88         path = "archive-pdb"
89         archive = database.SPickle(path)
90
91         if config.fromtime:
92                 begin = config.fromtime
93         else:
94                 begin = "2007-11-06"
95
96         if config.node is None and len(config.args) > 0:
97                 config.node = config.args[0]
98         elif config.node is None:
99                 print "Add a hostname to arguments"
100                 print "exit."
101                 sys.exit(1)
102
103         d = datetime_fromstr(begin)
104         tdelta = timedelta(1)
105         verbose = 1
106
107         while True:
108                 file = get_filefromglob(d, "production.findbad")
109                 #file = "%s.production.findbad" % d.strftime("%Y-%m-%d")
110                 
111                 try:
112                         fb = archive.load(file)
113                         if config.node in fb['nodes']:
114                                 fb_nodeinfo  = fb['nodes'][config.node]['values']
115                                 fb_print_nodeinfo(fb_nodeinfo, verbose, d.strftime("%Y-%m-%d"))
116
117                         del fb
118                         verbose = 0
119                 except KeyboardInterrupt:
120                         sys.exit(1)
121                 except:
122                         #import traceback; print traceback.print_exc()
123                         print d.strftime("%Y-%m-%d"), "No record"
124
125                 d = d + tdelta
126                 if d > datetime.now(): break
127
128 if __name__ == "__main__":
129         main()