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
|
#!/usr/bin/env python
import os
import sys
import json
import pickle
import hashlib
import logging
import argparse
import binascii
import ConfigParser
import requests
from twisted.internet import reactor, ssl
from twisted.web import server
from jobmanager import JobManager
from statustracker import StatusTracker
from servers import StatusSite, PingSite
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Check your stuff.")
parser.add_argument('-m', '--mode', choices=['daemon', 'cron'], required=True, help='The mode the application will run it.')
parser.add_argument('-c', '--crontime', choices=['minute', 'hour', 'day', 'day_noon'], help='When in cron mode, the increment of cron.')
parser.add_argument('-v', action="store_true", help="Print verbose debugging information to stderr")
args = parser.parse_args()
config = ConfigParser.ConfigParser()
configfile = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'settings.cfg')
if args.v:
print "[Pre-Logging] Reading config from", configfile
config.read(configfile)
if not config.get('email', 'user') or \
not config.get('email', 'pass') or \
not config.get('email', 'smtpserver') or \
not config.get('email', 'smtpport') or \
not config.get('email', 'imapserver'):
print "Sending email address is not configured"
sys.exit(1)
if not config.get('general', 'servername') or \
not config.get('general', 'alertcontact'):
print "Default alert contact is not configured"
sys.exit(1)
requests_log = logging.getLogger("requests.packages.urllib3")
requests_log.setLevel(logging.CRITICAL)
logging.basicConfig(format="%(asctime)s:%(levelname)s: %(message)s")
log = logging.getLogger()
if args.v:
log.setLevel(logging.DEBUG)
if args.mode == 'daemon':
log.info("Starting up daemon")
statusTracker = StatusTracker(config)
reactor.listenTCP(5000, server.Site(StatusSite(statusTracker)))
reactor.listenTCP(5001, server.Site(PingSite(statusTracker)), interface='127.0.0.1')
reactor.run()
elif args.mode == 'cron':
jobManager = JobManager(config)
if not args.crontime:
log.warn("Did not run cron, no crontime specified")
parser.print_help()
sys.exit(-1)
else:
log.info("Running cron at frequency " + args.crontime)
if jobManager.execute_jobs(args.crontime):
jobManager.mark_jobs_ran()
else:
jobManager.mark_jobs_ran_with_error()
|
