MyMonPlugins/check_cups.py

#!/usr/bin/env python2
# -*- encoding: utf-8 -*-

#####################################################################
# (c) 2016 by Sven Velt, Germany                                    #
#             sven-mymonplugins@velt.biz                            #
#                                                                   #
# This file is part of "velt.biz - My Monitoring Plugins"           #
# a fork of "team(ix) Monitoring Plugins" in 2015                   #
# URL: https://gogs.velt.biz/velt.biz/MyMonPlugins/                 #
#                                                                   #
# This file is free software: you can redistribute it and/or modify #
# it under the terms of the GNU General Public License as published #
# by the Free Software Foundation, either version 2 of the License, #
# or (at your option) any later version.                            #
#                                                                   #
# This file is distributed in the hope that it will be useful, but  #
# WITHOUT ANY WARRANTY; without even the implied warranty of        #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the      #
# GNU General Public License for more details.                      #
#                                                                   #
# You should have received a copy of the GNU General Public License #
# along with this file. If not, see <http://www.gnu.org/licenses/>. #
#####################################################################

import datetime
import re
import subprocess
import sys
import time

try:
	from monitoringplugin import MonitoringPlugin
except ImportError:
	print '=========================='
	print 'AIKS! Python import error!'
	print '==========================\n'
	print 'Could not find "monitoringplugin.py"!\n'
	print 'Did you download "%s"' % os.path.basename(sys.argv[0])
	print 'without "monitoringplugin.py"?\n'
	print 'Please go back to'
	print 'https://gogs.velt.biz/velt.biz/MyMonPlugins and download it,'
	print 'or even better:'
	print 'get a full archive at http://gogs.velt.biz/velt.biz/MyMonPlugins/releases'
	print 'or a master snapshot at http://gogs.velt.biz/velt.biz/MyMonPlugins/archive/master.tar.gz\n'
	sys.exit(127)


plugin = MonitoringPlugin(
		pluginname='check_cups',
		tagforstatusline='CUPS',
		description='Check CUPS jobs and printer queues',
		version='0.1',
	)


plugin.add_cmdlineoption('-P', '--check-printers', 'check_printers', 'check printer queue', default=False, action='store_true' )
plugin.add_cmdlineoption('-J', '--check-jobs', 'check_jobs', 'check job queue', default=False, action='store_true')
plugin.add_cmdlineoption('-w', '', 'warn', 'warning thresold for old jobs (seconds)', default='3600')
plugin.add_cmdlineoption('-c', '', 'crit', 'warning thresold for old jobs (seconds)', default='86400')

plugin.add_cmdlineoption('', '--mymontools-testmode', 'mymontools_testmode', None, default=False, action='store_true')

plugin.parse_cmdlineoptions()

##############################################################################
##### Testmode

if plugin.options.mymontools_testmode:
	# Because we need tzlocal from dateutil.tz
	mymontools_testmode = {}

	mymontools_testmode['lpstat -a'] = '''Printer_One accepting requests since 1970-01-01T00:00:00 CEST
Printer_Two accepting requests since 2016-01-01T00:00:00 CEST
Printer_Three accepting requests since 2116-01-01T00:00:00 CEST
Printer_Rejecting not accepting requests since 2016-02-01T00:00:00 CEST -
Rejecting Jobs'''.split('\n')

	from dateutil.tz import tzlocal
	mymontools_testmode['lpstat -o'] = '''Printer_One-1234 username 12345 1970-01-01T00:00:00 CEST
Printer_Two-1234 username 12345 %s''' % datetime.datetime.now(tzlocal()).strftime('%FT%T %Z')
	mymontools_testmode['lpstat -o'] = mymontools_testmode['lpstat -o'].split('\n')

##############################################################################

def check_printer_queue(output_printer_queue):
	lidx = 0
	printers = []
	printers_bad = []

	while lidx < len(output_printer_queue):
		printer =  output_printer_queue[lidx].split(' ')[0]
		if output_printer_queue[lidx].find('not accepting') > 0:
			reason = output_printer_queue[(lidx+1)].replace('\t', '').lstrip().rstrip()
			printers_bad.append( (printer, reason,) )
			lidx += 1

		printers.append( printer )
		lidx += 1

	plugin.remember_check('All Printers', 0, '%d printer%s found' % (len(printers), len(printers) != 1 and 's' or ''), perfdata=[ {'label':'printers', 'value':len(printers), 'unit':''}, ] )

	for p in printers_bad:
		plugin.remember_check('%s' % p[0], 1, p[1])


##############################################################################

def check_job_queue(output_job_queue):
	try:
		import dateutil.parser
		from dateutil.tz import tzlocal
	except ImportError:
		print('Python module "dateutil" required, please install!')
		sys.exit(3)

	m = re.compile('(\d{4}\D\d{2}\D\d{2}\D\d{2}\D\d{2}\D\d{2}\D\w{3,5})')
	nowsecs = long( time.time() ) 

	jobs_warn = []
	jobs_crit = []
	for line in output_job_queue:
		f = m.search(line)
		if not f:
			continue
		tstamp = dateutil.parser.parse(f.group(0))

		tsecs = long( tstamp.strftime('%s') )
		rc = plugin.value_wc_to_returncode((nowsecs-tsecs), plugin.options.warn, plugin.options.crit)
		if rc == 1:
			jobs_warn.append(line)
		elif rc == 2:
			jobs_crit.append(line)

	jobs = len(output_job_queue)
	jobs_old = len(jobs_warn) + len(jobs_crit)
	perfdata = []
	perfdata.append( { 'label':'jobs', 'value':'%d' % jobs, 'unit':'', } )
	perfdata.append( { 'label':'jobs_old', 'value':'%d' % jobs_old, 'unit':'', } )
	plugin.remember_check('Jobs', 0, '%d job%s found' % (jobs, jobs != 1 and 's' or ''), perfdata=perfdata )

	for j in jobs_crit:
		plugin.remember_check('Job %s' % j.split(' ')[0], 2, m.search(j).group(0) )
	for j in jobs_warn:
		plugin.remember_check('Job %s' % j.split(' ')[0], 1, m.search(j).group(0) )

##############################################################################

def call_cmd(cmdline):
	if plugin.options.mymontools_testmode:
		return (mymontools_testmode.get(' '.join(cmdline)), '', 0)

	try:
		cmd = subprocess.Popen(cmdline, stdout=subprocess.PIPE)
		(sout, serr) = cmd.communicate()
		if sout:
			sout = sout.lstrip().rstrip().split('\n')
		else:
			sout = []
	except OSError:
		plugin.back2nagios(3, 'Could not execute "%s"' % ' '.join(cmdline))
	return (sout, serr, cmd.returncode)

##############################################################################

allchecks = not( plugin.options.check_printers or plugin.options.check_jobs )

if allchecks or plugin.options.check_printers:
	(sout, serr, retcode) = call_cmd(['lpstat', '-a'])
	check_printer_queue(sout)
if allchecks or  plugin.options.check_jobs:
	(sout, serr, retcode) = call_cmd(['lpstat', '-o'])
	check_job_queue(sout)

# Exit
plugin.brain2output()
plugin.exit()