summaryrefslogtreecommitdiffstats
path: root/examples/glpi.py
blob: 30e148ea46167fadb9cf35dd1eb8c2e258032d2b (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
50
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 = ''
        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()
        except Exception as e:
            pass
        finally:
            return (POSITION, response)