summaryrefslogtreecommitdiffstats
path: root/examples/glpi.py
diff options
context:
space:
mode:
authorUltrabug <ultrabug@gentoo.org>2013-08-29 13:03:29 +0200
committerUltrabug <ultrabug@gentoo.org>2013-08-29 13:03:29 +0200
commit59f36cf1433f0b5602d6d3f150afcaf18b70e2d3 (patch)
tree055c7572fdfcc90be6698f4538f173d154a2b9c5 /examples/glpi.py
parente8c35a25f04316c622739bf527c4c53059b0e28f (diff)
make sure all examples are PEP8 compatible
Diffstat (limited to 'examples/glpi.py')
-rw-r--r--examples/glpi.py56
1 files changed, 29 insertions, 27 deletions
diff --git a/examples/glpi.py b/examples/glpi.py
index 30e148e..929749d 100644
--- a/examples/glpi.py
+++ b/examples/glpi.py
@@ -1,3 +1,7 @@
+# You need MySQL-python from http://pypi.python.org/pypi/MySQL-python
+import MySQLdb
+
+
class Py3status:
"""
This example class demonstrates how to display the current total number of
@@ -10,7 +14,7 @@ class Py3status:
py3status automagically.
"""
def count_glpi_open_tickets(self, json, i3status_config):
- response = {'full_text' : '', 'name' : 'glpi_tickets'}
+ response = {'full_text': '', 'name': 'glpi_tickets'}
# user-defined variables
CRIT_THRESHOLD = 20
@@ -21,30 +25,28 @@ class Py3status:
MYSQL_USER = ''
POSITION = 0
- 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=5,
+ )
+ 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()
- mydb = MySQLdb.connect(
- host=MYSQL_HOST,
- user=MYSQL_USER,
- passwd=MYSQL_PASSWD,
- db=MYSQL_DB,
- connect_timeout=5,
- )
- 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 as e:
- pass
- finally:
- return (POSITION, response)
+ return (POSITION, response)