/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to bzr-notify

  • Committer: Mark Lee
  • Date: 2009-07-10 20:07:43 UTC
  • mto: This revision was merged to the branch mainline in revision 661.
  • Revision ID: bzr@lazymalevolence.com-20090710200743-2s9x0xa8jgs9wf8s
Remove credits.pickle (generated file).

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/usr/bin/env python
 
2
 
 
3
"""Run the bzr tray icon.
 
4
 
 
5
This is a background program which will pop up a notification on the users
 
6
screen when a commit occurs.
 
7
"""
 
8
 
 
9
from bzrlib.plugin import load_plugins
 
10
load_plugins()
 
11
 
 
12
from bzrlib.plugins.gtk import open_display, icon_path
 
13
 
 
14
from bzrlib.plugins.gtk.notify import NotifyPopupMenu
 
15
gtk = open_display()
 
16
menu = NotifyPopupMenu()
 
17
icon = gtk.status_icon_new_from_file(icon_path("bzr-icon-64.png"))
 
18
icon.connect('popup-menu', menu.display)
 
19
icon.set_visible(False)
 
20
 
 
21
import cgi
 
22
import dbus
 
23
import dbus.service
 
24
import gobject
 
25
import pynotify
 
26
from bzrlib.bzrdir import BzrDir
 
27
from bzrlib import errors
 
28
from bzrlib.osutils import format_date
 
29
from bzrlib.transport import get_transport
 
30
if getattr(dbus, 'version', (0,0,0)) >= (0,41,0):
 
31
        import dbus.glib
 
32
BROADCAST_INTERFACE = "org.bazaarvcs.plugins.dbus.Broadcast"
 
33
bus = dbus.SessionBus()
 
34
 
 
35
def hide_icon():
 
36
        icon.set_visible(False)
 
37
 
 
38
def catch_branch(revision_id, urls):
 
39
        # TODO: show all the urls, or perhaps choose the 'best'.
 
40
        url = urls[0]
 
41
        try:
 
42
                if isinstance(revision_id, unicode):
 
43
                        revision_id = revision_id.encode('utf8')
 
44
                transport = get_transport(url)
 
45
                a_dir = BzrDir.open_from_transport(transport)
 
46
                branch = a_dir.open_branch()
 
47
                revno = branch.revision_id_to_revno(revision_id)
 
48
                revision = branch.repository.get_revision(revision_id)
 
49
                summary = 'New revision %d in %s' % (revno, url)
 
50
                body  = 'Committer: %s\n' % revision.committer
 
51
                body += 'Date: %s\n' % format_date(revision.timestamp,
 
52
                        revision.timezone)
 
53
                body += '\n'
 
54
                body += revision.message
 
55
                body = cgi.escape(body)
 
56
                nw = pynotify.Notification(summary, body)
 
57
                def start_viz(notification=None, action=None, data=None):
 
58
                        """Start the viz program."""
 
59
                        pp = start_viz_window(branch, revision_id)
 
60
                        pp.show()
 
61
                def start_branch(notification=None, action=None, data=None):
 
62
                        """Start a Branch dialog"""
 
63
                        from bzrlib.plugins.gtk.branch import BranchDialog
 
64
                        bd = BranchDialog(remote_path=url)
 
65
                        bd.run()
 
66
                if "actions" in pynotify.get_server_caps():
 
67
                        nw.add_action("inspect", "Inspect", start_viz, None)
 
68
                        nw.add_action("branch", "Branch", start_branch, None)
 
69
                icon.set_visible(True)
 
70
                gobject.timeout_add(5000, hide_icon)
 
71
                nw.set_timeout(5000)
 
72
                nw.show()
 
73
        except Exception, e:
 
74
                print e
 
75
                raise
 
76
bus.add_signal_receiver(catch_branch,
 
77
                                                dbus_interface=BROADCAST_INTERFACE,
 
78
                                                signal_name="Revision")
 
79
pynotify.init("bzr-notify")
 
80
gtk.main()