hope I got the merge wright...

[monitor.git] / commands / siteinfo.py

diff --git a/commands/siteinfo.py b/commands/siteinfo.py
new file mode 100755 (executable)
index 0000000..4b4daf7
--- /dev/null
+++ b/commands/siteinfo.py
@@ -0,0 +1,102 @@
+#!/usr/bin/python
+
+from monitor.wrapper import plc
+api = plc.getAuthAPI()
+
+from monitor import database
+
+import time
+from monitor.common import *
+
+from monitor import util
+from monitor import parser as parsermodule
+from monitor.model import *
+
+
+parser = parsermodule.getParser()
+parser.set_defaults(site=None, 
+                                       findbad=False,
+                                       enable=False,
+                                       disable=False
+                                       )
+parser.add_option("", "--site", dest="site", metavar="login_base", 
+                                       help="The sitename to present")
+parser.add_option("", "--findbad", dest="findbad", action="store_true", 
+                                       help="Re-run findbad on the nodes we're going to check before acting.")
+parser.add_option("", "--enable", dest="enable", action="store_true",
+                                       help="")
+parser.add_option("", "--disable", dest="disable", action="store_true",
+                                       help="")
+config = parsermodule.parse_args(parser)
+
+def color_sitestatus(status):
+       if status == "good":
+               return green(status)
+       elif status == "down":
+               return red(status)
+       else:
+               return status
+               
+
+def pf_print_siteinfo(sitename):
+       pf = PersistFlags(sitename, 1, db='site_persistflags')
+       if pf.checkattr('last_changed'):
+               print "   Checked: %s" % diff_time(pf.last_checked)
+               print "\t status | nodes up / total | last_change"
+               print "\t   %6s | %8s / %5s | %s" % \
+                       ( color_sitestatus(pf.status), pf.nodes_up, pf.nodes_total, diff_time(pf.last_changed) )
+       else:
+               print "no  such site in pf"
+       del pf
+
+
+def plc_print_siteinfo(plcsite):
+       print ""
+       print "   Checked: %s" % time.ctime()
+       print "\t login_base    | used / max | enabled | last_updated "
+       print "\t %13s | %4s / %3s | %7s | %12s" % \
+                       (plcsite['login_base'], 
+                        len(plcsite['slice_ids']),
+                        plcsite['max_slices'],
+                        plcsite['enabled'],
+                        diff_time(plcsite['last_updated']))
+
+       print ""
+       nodes = plccache.GetNodesByIds(plcsite['node_ids'])
+       print "   Checked: %s" % time.ctime()
+       print "\t                               host     | state | obs   |   created   |   updated   | last_contact "
+       for plcnode in nodes:
+               fbnode = FindbadNodeRecord.get_latest_by(hostname=plcnode['hostname']).to_dict()
+               plcnode['state'] = color_boot_state(get_current_state(fbnode))
+               print "\t  %37s |  %5s |  %5s | %11.11s | %11.11s | %12s " % \
+               (plcnode['hostname'], color_boot_state(plcnode['boot_state']), plcnode['state'], 
+                       diff_time(plcnode['date_created']), diff_time(plcnode['last_updated']), 
+               diff_time(plcnode['last_contact']))
+
+
+act_all = database.dbLoad("act_all")
+
+for site in config.args:
+       config.site = site
+
+       plc_siteinfo = plccache.GetSitesByName([config.site])
+       url = "https://www.planet-lab.org/db/sites/index.php?site_pattern="
+       plc_siteinfo['url'] = url + plc_siteinfo['login_base']
+
+       if config.findbad:
+               # rerun findbad with the nodes in the given nodes.
+               import os
+               file = "findbad.txt"
+               nodes = plccache.GetNodesByIds(plc_siteinfo['node_ids'])
+               nodes = [ n['hostname'] for n in nodes ]
+               util.file.setFileFromList(file, nodes)
+               os.system("./findbad.py --cachenodes --debug=0 --dbname=findbad --increment --nodelist %s" % file)
+
+       print "%(login_base)s %(url)s" % plc_siteinfo
+       pf_print_siteinfo(config.site)
+       plc_print_siteinfo(plc_siteinfo)
+
+       if config.enable:
+               api.UpdateSite(config.site, {'enabled' : True})
+       if config.disable:
+               api.UpdateSite(config.site, {'enabled' : False})