git-buildpackage: Add support for sending notifications via libnotify

after the build finished.

2 files changed, 76 insertions, 2 deletions

diff --git a/gbp/config.py b/gbp/config.py
index e623c3b2..8448762c 100644
--- a/gbp/config.py
+++ b/gbp/config.py
@@ -94,6 +94,7 @@ class GbpOptionParser(OptionParser):
                  'customizations'  : '',
                  'spawn-editor'    : 'release',
                  'patch-numbers'   : 'True',
+                 'notify'          : 'auto',
              }
     help = {
              'debian-branch':
@@ -145,9 +146,11 @@ class GbpOptionParser(OptionParser):
              'color':
                   "color output, default is '%(color)s'",
              'spawn-editor':
-                  "Wether to spawn an editor after adding the changelog entry, default is '%(spawn-editor)s'",
+                  "Whether to spawn an editor after adding the changelog entry, default is '%(spawn-editor)s'",
              'patch-numbers':
-                  "Wether to number patch files, default is %(patch-numbers)s",
+                  "Whether to number patch files, default is %(patch-numbers)s",
+             'notify':
+                  "Whether to send a desktop notification after the build, default is '%(notify)s'",
            }
     config_files = [ '/etc/git-buildpackage/gbp.conf',
                      os.path.expanduser('~/.gbp.conf'),
diff --git a/gbp/notifications.py b/gbp/notifications.py
new file mode 100644
index 00000000..b2534e3f
--- /dev/null
+++ b/gbp/notifications.py
@@ -0,0 +1,71 @@
+# vim: set fileencoding=utf-8 :
+#
+# (C) 2011 Guido Guenther <agx@sigxcpu.org>
+#    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
+#    along with this program; if not, write to the Free Software
+#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+import log
+import warnings
+
+notify_module = None
+
+def enable_notifications():
+    global notify_module
+
+    with warnings.catch_warnings():
+        # Avoid GTK+ cannot open display warning:
+        warnings.simplefilter("ignore")
+        try:
+            import pynotify
+            notify_module = pynotify
+        except ImportError:
+            return False
+
+    return notify_module.init("git-buildpackage")
+
+
+def build_msg(cp, success):
+    summary = "Gbp %s" % ["failed", "successful"][success]
+    msg = ("Build of %s %s %s" %
+            (cp['Source'], cp['Version'], ["failed", "succeeded"][success]))
+
+    return summary, msg
+
+
+def send_notification(summary, msg):
+    n = notify_module.Notification(summary, msg)
+    try:
+        if not n.show():
+            return False
+    except:
+        return False
+    return True
+
+
+def notify(cp, success, notify_opt):
+    """
+    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()]
+
+    summary, msg = build_msg(cp, success)
+    return notify_opt.do(send_notification, summary, msg)
+