notifications.py 1.97 KB
Newer Older
1 2
# vim: set fileencoding=utf-8 :
#
Guido Günther's avatar
Guido Günther committed
3
# (C) 2011 Guido Günther <agx@sigxcpu.org>
4 5 6 7 8 9 10 11 12 13 14
#    This program is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 2 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
15 16
#    along with this program; if not, please see
#    <http://www.gnu.org/licenses/>
17 18 19 20 21

import warnings

notify_module = None

22

23 24 25 26 27 28 29
def enable_notifications():
    global notify_module

    with warnings.catch_warnings():
        # Avoid GTK+ cannot open display warning:
        warnings.simplefilter("ignore")
        try:
Guido Günther's avatar
Guido Günther committed
30 31
            import notify2
            notify_module = notify2
32
        except (ImportError, RuntimeError):
33 34
            return False

Guido Günther's avatar
Guido Günther committed
35 36 37 38
    try:
        return notify_module.init("git-buildpackage")
    except Exception:
        return False
39 40 41 42 43


def build_msg(cp, success):
    summary = "Gbp %s" % ["failed", "successful"][success]
    msg = ("Build of %s %s %s" %
44
           (cp['Source'], cp['Version'], ["failed", "succeeded"][success]))
45 46 47 48 49 50

    return summary, msg


def send_notification(summary, msg):
    n = notify_module.Notification(summary, msg)
51
    n.set_hint('transient', True)
52 53 54
    try:
        if not n.show():
            return False
55
    except Exception:
56 57 58 59
        return False
    return True


60
def notify(summary, message, notify_opt):
61 62 63 64 65 66 67 68 69 70 71 72
    """
    Send a notifications
    @return: False on error
    """

    if notify_opt.is_off():
        return True

    enable = enable_notifications()
    if not enable:
        return [True, False][notify_opt.is_on()]

73
    return notify_opt.do(send_notification, summary, message)