summaryrefslogtreecommitdiffstats
path: root/examples/glpi.py
blob: 0afdacebb6c315731ee18325f9fcb167456bf47a (plain)
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
class Py3status:
	"""
	This example class demonstrates how to display the current total number of
	open tickets from GLPI in your i3bar.

	It features thresholds to colorize the output and forces a low timeout to
	limit the impact of a server connectivity problem on your i3bar freshness.

	Note that we don't have to implement a cache layer as it is handled by
	py3status automagically.
	"""
	def count_glpi_open_tickets(self, json, i3status_config):
		response = {'full_text' : '', 'name' : 'glpi_tickets'}

		# user-defined variables
		CRIT_THRESHOLD = 20
		WARN_THRESHOLD = 15
		MYSQL_DB = ''
		MYSQL_HOST = ''
		MYSQL_PASSWD = ''
		MYSQL_USER = ''

		try:
			# You need MySQL-python from http://pypi.python.org/pypi/MySQL-python
			import MySQLdb

			mydb = MySQLdb.connect(
				host=MYSQL_HOST,
				user=MYSQL_USER,
				passwd=MYSQL_PASSWD,
				db=MYSQL_DB,
				connect_timeout=1, # dont cause a lag on i3bar freshness
				)
			mycr = mydb.cursor()
			mycr.execute('select count(*) from glpi_tickets where closedate is NULL and solvedate is NULL;')
			row = mycr.fetchone()
			if row:
				open_tickets = int(row[0])
				if i3status_config['colors']:
					if open_tickets > CRIT_THRESHOLD:
						response.update({'color': i3status_config['color_bad']})
					elif open_tickets > WARN_THRESHOLD:
						response.update({'color': i3status_config['color_degraded']})
				response['full_text'] = '%s tickets' % open_tickets
			mydb.close()
		except Exception, e:
			pass
		finally:
			return (0, response)