cleanup
[infrastructure.git] / nagios / configurator / configurator.py
1 #!/usr/bin/env python
2
3 # WARNING - you might need to set your locale before you can run this script
4 # e.g.
5 # export LANG=en_US.UTF-8
6 # nagios.py ...
7
8 import sys 
9 import getopt
10
11 from NagiosConfig import NagiosConfig
12
13 SERVER="www.planet-lab.org"
14 DIRNAME="/etc/nagios"
15
16 options='h:o:Os:d:nNv'
17
18 usage_string="""Usage : %s [options] plc-id plc-passwd role
19 Outputs a nagios config for watching a given set of sites
20 Also writes a static HTML page for bookmarking site-centric queries to comon
21 Options:
22   -h plc-hostname       To provide an alternate plc hostname
23                         (defaults to %s)
24   -o e-mail             If you provide an e-mail address, all notifications
25                         will be sent there. Otherwise, you'll need admin role
26                         so that we can determine the respective PI's e-mails
27   -O                    same as -e with the plc-id as dest. e-mail
28   -s site-file          You can provide a list of sites in a separate file
29                         expected format is one line per site, site_id first
30                         the default is to list all sites
31   -d dirname            To specify an alternate output dir
32                         (defaults to %s)
33   -n                    Does not generate the nagios config
34   -N                    Does not generate the HTML page
35   -v                    Runs in verbose mode
36 """%(sys.argv[0],SERVER,DIRNAME)
37
38 def usage ():
39     print usage_string
40     sys.exit(1)
41
42
43 def main ():
44     command = sys.argv[0]
45     argv = sys.argv[1:]
46
47     try:
48         opts,args = getopt.getopt(argv, options)
49     except getopt.GetoptError:
50         print "Unknown option"
51         usage()
52
53     plc_server=SERVER
54     output_mail=None
55     opt_same_email=False
56     site_file=None
57     dirname=DIRNAME
58     opt_nagios = True
59     opt_html = True
60     opt_verbose = False
61     
62     for o,a in opts:
63         if o in ['-h']:
64             plc_server = a
65         elif o in ['-o']:
66             output_mail = a
67         elif o in ['-O']:
68             opt_same_email = True
69         elif o in ['-s']:
70             site_file = a
71         elif o in ['-d']:
72             dirname = a
73         elif o in ['-n']:
74             opt_nagios = False
75         elif o in ['-N']:
76             opt_html = False
77         elif o in ['-v']:
78             opt_verbose = True
79         else:
80             print "Unknown option",o
81             usage()
82
83     if not len(args) == 3:
84         usage()
85
86     (plc_id,passwd,role) = args
87     if opt_same_email:
88         output_mail=plc_id
89
90 #    print plc_server,output_mail,dirname,opt_nagios,opt_html
91 #    print plc_id,passwd,role,site_file
92
93     print "Connecting to %s and gathering data"%plc_server
94     config = NagiosConfig (plc_server,plc_id,passwd,role)
95     if opt_verbose:
96         config.set_verbose()
97     config.GatherData (output_mail,site_file)
98
99     if opt_nagios:
100         print "Creating nagios config in %s"%dirname
101         config.WriteNagiosConfig (dirname)
102
103     if opt_html:
104         html_page = "%s/comon.html"%dirname
105         config.WriteComonLinks (html_page)
106
107     
108 ####################
109 if __name__ == "__main__":
110     main ()
111