1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#!/usr/bin/python2.4
#
# Copyright 2009 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Monitor scripts."""
__metaclass__ = type
__all__ = ['check_script']
import _pythonpath
from datetime import datetime, timedelta
from email.MIMEText import MIMEText
from optparse import OptionParser
from time import strftime
import smtplib
import sys
from canonical.database.sqlbase import connect
from canonical.launchpad.scripts import db_options, logger_options, logger
from canonical.launchpad.scripts.scriptmonitor import check_script
def main():
# XXX: Tom Haddon 2007-07-12
# There's a lot of untested stuff here: parsing options and sending
# emails - this should be moved into a testable location.
# Also duplicated code in scripts/script-monitor-nagios.py
parser = OptionParser(
'%prog [options] (minutes) (host:scriptname) [host:scriptname]'
)
db_options(parser)
logger_options(parser)
(options, args) = parser.parse_args()
if len(args) < 2:
parser.error("Must specify at time in minutes and "
"at least one host and script")
# First argument is the number of minutes into the past
# we want to look for the scripts on the specified hosts
try:
minutes_ago, args = int(args[0]), args[1:]
start_date = datetime.now() - timedelta(minutes=minutes_ago)
completed_from = strftime("%Y-%m-%d %H:%M:%S", start_date.timetuple())
completed_to = strftime("%Y-%m-%d %H:%M:%S", datetime.now().timetuple())
hosts_scripts = []
for arg in args:
try:
hostname, scriptname = arg.split(':')
except TypeError:
parser.error(
"%r is not in the format 'host:scriptname'" % (arg,))
hosts_scripts.append((hostname, scriptname))
except ValueError:
parser.error("Must specify time in minutes and "
"at least one host and script")
log = logger(options)
try:
log.debug("Connecting to database")
con = connect(options.dbuser)
error_found = False
msg, subj = [], []
for hostname, scriptname in hosts_scripts:
failure_msg = check_script(con, log, hostname,
scriptname, completed_from, completed_to)
if failure_msg is not None:
msg.append(failure_msg)
subj.append("%s:%s" % (hostname, scriptname))
error_found = 2
if error_found:
# Construct our email
msg = MIMEText('\n'.join(msg))
msg['Subject'] = "Scripts failed to run: %s" % ", ".join(subj)
msg['From'] = 'script-failures@launchpad.net'
msg['Reply-To'] = 'launchpad@lists.canonical.com'
msg['To'] = 'launchpad@lists.canonical.com'
# Send out the email
smtp = smtplib.SMTP()
smtp.connect()
smtp.sendmail('script-failures@launchpad.net', ['launchpad@lists.canonical.com'], msg.as_string())
smtp.close()
return 2
except:
log.exception("Unhandled exception")
return 1
if __name__ == '__main__':
sys.exit(main())
|