diff options
Diffstat (limited to 'examples/glpi.py')
| -rw-r--r-- | examples/glpi.py | 86 |
1 files changed, 43 insertions, 43 deletions
diff --git a/examples/glpi.py b/examples/glpi.py index 0afdace..9d84f81 100644 --- a/examples/glpi.py +++ b/examples/glpi.py @@ -1,49 +1,49 @@ class Py3status: - """ - This example class demonstrates how to display the current total number of - open tickets from GLPI in your i3bar. + """ + 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. + 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'} + 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 = '' + # 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 + 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) + 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 as e: + pass + finally: + return (0, response) |