BATCAVE/ffstatus-daemon.py

#!/usr/bin/python
from __future__ import print_function
from copy import deepcopy
import daemon
import logging
import sys
import time

from ffstatus import *

INTERVAL = 15
LOGFILE = '/var/log/ffstatus.log'

DUMMY_MODE = 1

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
fh = logging.FileHandler(LOGFILE)
fh.setFormatter(logging.Formatter('%(asctime)s [%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S'))
logger.addHandler(fh)

logger.info('Starting up')

a = AlfredParser()
g = GraphitePush('fdca:ffee:ff12:a254::da7a', 2003)
data = { }

if DUMMY_MODE:
	a.alfred_json = '/home/ffpb-statusbot/status-daemon/alfred-json'
	g.dont_send = True

try:
	a.sanitycheck()
except Exception as err:
	logger.critical('AlfredParser.sanitycheck() failed: ' + str(err))
	print('FAILED SANITY CHECK: ' + str(err))
	sys.exit(1)

daemon_context = daemon.DaemonContext(
	files_preserve = [ fh.stream ],
)
with daemon_context:

	while True:
		try:
			ts = int(time.time())
			logger.debug('Step 1/3: Fetching data ...')
			newdata = a.fetch()

			logger.debug('Step 2/3: Pushing update data ...')
			graphitedata = g.push(newdata, ts=ts)

			logger.info('Sent ' + str(graphitedata.count('\n')+1) + ' lines to Graphite.')

			logger.debug('Step 3/3: Merging current data ...')
			data = dict_merge(data, newdata)
			logger.info('I have data for ' + str(len(data)) + ' nodes.')
		except Exception as err:
			logger.error(str(err))

		time.sleep(INTERVAL)

	logger.info('Shutting down')