add GLPI open tickets counter module example

1 files changed, 49 insertions, 0 deletions

diff --git a/examples/glpi.py b/examples/glpi.py
new file mode 100644
index 0000000..0afdace
--- /dev/null
+++ b/examples/glpi.py
@@ -0,0 +1,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)