/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 __init__.py

Fix regressions in plugins tab.

Show diffs side-by-side

added added

removed removed

Lines of Context:
__init__.py
15
15
"""Graphical support for Bazaar using GTK.
16
16
 
17
17
This plugin includes:
 
18
commit-notify     Start the graphical notifier of commits.
18
19
gannotate         GTK+ annotate. 
19
20
gbranch           GTK+ branching. 
20
21
gcheckout         GTK+ checkout. 
21
22
gcommit           GTK+ commit dialog.
22
23
gconflicts        GTK+ conflicts. 
23
24
gdiff             Show differences in working tree in a GTK+ Window. 
 
25
ghandle-patch     Display and optionally merge a merge directive or patch.
24
26
ginit             Initialise a new branch.
25
 
gmerge            GTK+ merge dialog
26
27
gmissing          GTK+ missing revisions dialog. 
27
28
gpreferences      GTK+ preferences dialog. 
28
29
gpush             GTK+ push.
36
37
 
37
38
import bzrlib
38
39
 
39
 
version_info = (0, 95, 0, 'dev', 1)
 
40
version_info = (0, 94, 0, 'dev', 0)
40
41
 
41
42
if version_info[3] == 'final':
42
43
    version_string = '%d.%d.%d' % version_info[:3]
104
105
    bzrlib.ui.ui_factory = GtkUIFactory()
105
106
 
106
107
 
107
 
def data_basedirs():
108
 
    return [os.path.dirname(__file__),
 
108
def data_path():
 
109
    return os.path.dirname(__file__)
 
110
 
 
111
 
 
112
def icon_path(*args):
 
113
    basedirs = [os.path.join(data_path()),
109
114
             "/usr/share/bzr-gtk", 
110
115
             "/usr/local/share/bzr-gtk"]
111
 
 
112
 
 
113
 
def data_path(*args):
114
 
    for basedir in data_basedirs():
115
 
        path = os.path.join(basedir, *args)
 
116
    for basedir in basedirs:
 
117
        path = os.path.join(basedir, 'icons', *args)
116
118
        if os.path.exists(path):
117
119
            return path
118
120
    return None
119
121
 
120
122
 
121
 
def icon_path(*args):
122
 
    return data_path(os.path.join('icons', *args))
123
 
 
124
 
 
125
 
def open_display():
126
 
    pygtk = import_pygtk()
127
 
    try:
128
 
        import gtk
129
 
    except RuntimeError, e:
130
 
        if str(e) == "could not open display":
131
 
            raise NoDisplayError
132
 
    set_ui_factory()
133
 
    return gtk
134
 
 
135
 
 
136
123
class GTKCommand(Command):
137
124
    """Abstract class providing GTK specific run commands."""
138
125
 
 
126
    def open_display(self):
 
127
        pygtk = import_pygtk()
 
128
        try:
 
129
            import gtk
 
130
        except RuntimeError, e:
 
131
            if str(e) == "could not open display":
 
132
                raise NoDisplayError
 
133
        set_ui_factory()
 
134
        return gtk
 
135
 
139
136
    def run(self):
140
 
        open_display()
 
137
        self.open_display()
141
138
        dialog = self.get_gtk_dialog(os.path.abspath('.'))
142
139
        dialog.run()
143
140
 
171
168
 
172
169
    def run(self, location="."):
173
170
        (br, path) = branch.Branch.open_containing(location)
174
 
        open_display()
 
171
        self.open_display()
175
172
        from push import PushDialog
176
173
        dialog = PushDialog(br.repository, br.last_revision(), br)
177
174
        dialog.run()
196
193
            if revision is not None:
197
194
                if len(revision) == 1:
198
195
                    tree1 = wt
199
 
                    revision_id = revision[0].as_revision_id(tree1.branch)
 
196
                    revision_id = revision[0].in_history(branch).rev_id
200
197
                    tree2 = branch.repository.revision_tree(revision_id)
201
198
                elif len(revision) == 2:
202
 
                    revision_id_0 = revision[0].as_revision_id(branch)
 
199
                    revision_id_0 = revision[0].in_history(branch).rev_id
203
200
                    tree2 = branch.repository.revision_tree(revision_id_0)
204
 
                    revision_id_1 = revision[1].as_revision_id(branch)
 
201
                    revision_id_1 = revision[1].in_history(branch).rev_id
205
202
                    tree1 = branch.repository.revision_tree(revision_id_1)
206
203
            else:
207
204
                tree1 = wt
264
261
            if revision is None:
265
262
                revids.append(br.last_revision())
266
263
            else:
267
 
                revids.append(revision[0].as_revision_id(br))
 
264
                (revno, revid) = revision[0].in_history(br)
 
265
                revids.append(revid)
268
266
        import gtk
269
267
        pp = start_viz_window(br, revids, limit)
270
268
        pp.connect("destroy", lambda w: gtk.main_quit())
289
287
    aliases = ["gblame", "gpraise"]
290
288
    
291
289
    def run(self, filename, all=False, plain=False, line='1', revision=None):
292
 
        gtk = open_display()
 
290
        gtk = self.open_display()
293
291
 
294
292
        try:
295
293
            line = int(line)
314
312
        if revision is not None:
315
313
            if len(revision) != 1:
316
314
                raise BzrCommandError("Only 1 revion may be specified.")
317
 
            revision_id = revision[0].as_revision_id(br)
 
315
            revision_id = revision[0].in_history(br).rev_id
318
316
            tree = br.repository.revision_tree(revision_id)
319
317
        else:
320
318
            revision_id = getattr(tree, 'get_revision_id', lambda: None)()
321
319
 
322
 
        window = GAnnotateWindow(all, plain, branch=br)
 
320
        window = GAnnotateWindow(all, plain)
323
321
        window.connect("destroy", lambda w: gtk.main_quit())
324
322
        config = GAnnotateConfig(window)
325
323
        window.show()
348
346
 
349
347
    def run(self, filename=None):
350
348
        import os
351
 
        open_display()
 
349
        self.open_display()
352
350
        from commit import CommitDialog
353
351
        from bzrlib.errors import (BzrCommandError,
354
352
                                   NotBranchError,
361
359
            br = wt.branch
362
360
        except NoWorkingTree, e:
363
361
            from dialog import error_dialog
364
 
            error_dialog(_i18n('Directory does not have a working tree'),
365
 
                         _i18n('Operation aborted.'))
 
362
            error_dialog(_('Directory does not have a working tree'),
 
363
                         _('Operation aborted.'))
366
364
            return 1 # should this be retval=3?
367
365
 
368
366
        # It is a good habit to keep things locked for the duration, but it
385
383
    
386
384
    aliases = [ "gst" ]
387
385
    takes_args = ['PATH?']
388
 
    takes_options = ['revision']
 
386
    takes_options = []
389
387
 
390
 
    def run(self, path='.', revision=None):
 
388
    def run(self, path='.'):
391
389
        import os
392
 
        gtk = open_display()
 
390
        gtk = self.open_display()
393
391
        from status import StatusDialog
394
392
        (wt, wt_path) = workingtree.WorkingTree.open_containing(path)
395
 
        
396
 
        if revision is not None:
397
 
            try:
398
 
                revision_id = revision[0].as_revision_id(wt.branch)
399
 
            except:
400
 
                from bzrlib.errors import BzrError
401
 
                raise BzrError('Revision %r doesn\'t exist' % revision[0].user_spec )
402
 
        else:
403
 
            revision_id = None
404
 
 
405
 
        status = StatusDialog(wt, wt_path, revision_id)
 
393
        status = StatusDialog(wt, wt_path)
406
394
        status.connect("destroy", gtk.main_quit)
407
395
        status.run()
408
396
 
413
401
    """
414
402
    def run(self):
415
403
        (br, path) = branch.Branch.open_containing(".")
416
 
        gtk = open_display()
 
404
        gtk = self.open_display()
417
405
        from bzrlib.plugins.gtk.mergedirective import SendMergeDirectiveDialog
418
406
        from StringIO import StringIO
419
407
        dialog = SendMergeDirectiveDialog(br)
435
423
    """
436
424
    def run(self):
437
425
        (wt, path) = workingtree.WorkingTree.open_containing('.')
438
 
        open_display()
 
426
        self.open_display()
439
427
        from bzrlib.plugins.gtk.conflicts import ConflictsDialog
440
428
        dialog = ConflictsDialog(wt)
441
429
        dialog.run()
446
434
 
447
435
    """
448
436
    def run(self):
449
 
        open_display()
 
437
        self.open_display()
450
438
        from bzrlib.plugins.gtk.preferences import PreferencesWindow
451
439
        dialog = PreferencesWindow()
452
440
        dialog.run()
453
441
 
454
442
 
455
 
class cmd_gmerge(Command):
456
 
    """ GTK+ merge dialog
457
 
    
458
 
    """
459
 
    takes_args = ["merge_from_path?"]
460
 
    def run(self, merge_from_path=None):
461
 
        from bzrlib import workingtree
462
 
        from bzrlib.plugins.gtk.dialog import error_dialog
463
 
        from bzrlib.plugins.gtk.merge import MergeDialog
464
 
        
465
 
        (wt, path) = workingtree.WorkingTree.open_containing('.')
466
 
        old_tree = wt.branch.repository.revision_tree(wt.branch.last_revision())
467
 
        delta = wt.changes_from(old_tree)
468
 
        if len(delta.added) or len(delta.removed) or len(delta.renamed) or len(delta.modified):
469
 
            error_dialog(_i18n('There are local changes in the branch'),
470
 
                         _i18n('Please commit or revert the changes before merging.'))
471
 
        else:
472
 
            parent_branch_path = wt.branch.get_parent()
473
 
            merge = MergeDialog(wt, path, parent_branch_path)
474
 
            response = merge.run()
475
 
            merge.destroy()
476
 
 
477
 
 
478
443
class cmd_gmissing(Command):
479
444
    """ GTK+ missing revisions dialog.
480
445
 
513
478
 
514
479
class cmd_ginit(GTKCommand):
515
480
    def run(self):
516
 
        open_display()
 
481
        self.open_display()
517
482
        from initialize import InitDialog
518
483
        dialog = InitDialog(os.path.abspath(os.path.curdir))
519
484
        dialog.run()
523
488
    def run(self):
524
489
        br = branch.Branch.open_containing('.')[0]
525
490
        
526
 
        gtk = open_display()
 
491
        gtk = self.open_display()
527
492
        from tags import TagsWindow
528
493
        window = TagsWindow(br)
529
494
        window.show()
538
503
    cmd_gconflicts, 
539
504
    cmd_gdiff,
540
505
    cmd_ginit,
541
 
    cmd_gmerge,
542
506
    cmd_gmissing, 
543
507
    cmd_gpreferences, 
544
508
    cmd_gpush, 
552
516
    register_command(cmd)
553
517
 
554
518
 
 
519
class cmd_commit_notify(GTKCommand):
 
520
    """Run the bzr commit notifier.
 
521
 
 
522
    This is a background program which will pop up a notification on the users
 
523
    screen when a commit occurs.
 
524
    """
 
525
 
 
526
    def run(self):
 
527
        from notify import NotifyPopupMenu
 
528
        gtk = self.open_display()
 
529
        menu = NotifyPopupMenu()
 
530
        icon = gtk.status_icon_new_from_file(icon_path("bzr-icon-64.png"))
 
531
        icon.connect('popup-menu', menu.display)
 
532
 
 
533
        import cgi
 
534
        import dbus
 
535
        import dbus.service
 
536
        import pynotify
 
537
        from bzrlib.bzrdir import BzrDir
 
538
        from bzrlib import errors
 
539
        from bzrlib.osutils import format_date
 
540
        from bzrlib.transport import get_transport
 
541
        if getattr(dbus, 'version', (0,0,0)) >= (0,41,0):
 
542
            import dbus.glib
 
543
        BROADCAST_INTERFACE = "org.bazaarvcs.plugins.dbus.Broadcast"
 
544
        bus = dbus.SessionBus()
 
545
 
 
546
        def catch_branch(revision_id, urls):
 
547
            # TODO: show all the urls, or perhaps choose the 'best'.
 
548
            url = urls[0]
 
549
            try:
 
550
                if isinstance(revision_id, unicode):
 
551
                    revision_id = revision_id.encode('utf8')
 
552
                transport = get_transport(url)
 
553
                a_dir = BzrDir.open_from_transport(transport)
 
554
                branch = a_dir.open_branch()
 
555
                revno = branch.revision_id_to_revno(revision_id)
 
556
                revision = branch.repository.get_revision(revision_id)
 
557
                summary = 'New revision %d in %s' % (revno, url)
 
558
                body  = 'Committer: %s\n' % revision.committer
 
559
                body += 'Date: %s\n' % format_date(revision.timestamp,
 
560
                    revision.timezone)
 
561
                body += '\n'
 
562
                body += revision.message
 
563
                body = cgi.escape(body)
 
564
                nw = pynotify.Notification(summary, body)
 
565
                def start_viz(notification=None, action=None, data=None):
 
566
                    """Start the viz program."""
 
567
                    pp = start_viz_window(branch, revision_id)
 
568
                    pp.show()
 
569
                def start_branch(notification=None, action=None, data=None):
 
570
                    """Start a Branch dialog"""
 
571
                    from bzrlib.plugins.gtk.branch import BranchDialog
 
572
                    bd = BranchDialog(remote_path=url)
 
573
                    bd.run()
 
574
                nw.add_action("inspect", "Inspect", start_viz, None)
 
575
                nw.add_action("branch", "Branch", start_branch, None)
 
576
                nw.set_timeout(5000)
 
577
                nw.show()
 
578
            except Exception, e:
 
579
                print e
 
580
                raise
 
581
        bus.add_signal_receiver(catch_branch,
 
582
                                dbus_interface=BROADCAST_INTERFACE,
 
583
                                signal_name="Revision")
 
584
        pynotify.init("bzr commit-notify")
 
585
        gtk.main()
 
586
 
 
587
register_command(cmd_commit_notify)
 
588
 
 
589
 
555
590
class cmd_gselftest(GTKCommand):
556
591
    """Version of selftest that displays a notification at the end"""
557
592
 
648
683
register_command(cmd_test_gtk)
649
684
 
650
685
 
 
686
class cmd_ghandle_patch(GTKCommand):
 
687
    """Display a patch or merge directive, possibly merging.
 
688
 
 
689
    This is a helper, meant to be launched from other programs like browsers
 
690
    or email clients.  Since these programs often do not allow parameters to
 
691
    be provided, a "handle-patch" script is included.
 
692
    """
 
693
 
 
694
    takes_args = ['path']
 
695
 
 
696
    def run(self, path):
 
697
        try:
 
698
            from bzrlib.plugins.gtk.diff import (DiffWindow,
 
699
                                                 MergeDirectiveWindow)
 
700
            lines = open(path, 'rb').readlines()
 
701
            lines = [l.replace('\r\n', '\n') for l in lines]
 
702
            try:
 
703
                directive = merge_directive.MergeDirective.from_lines(lines)
 
704
            except errors.NotAMergeDirective:
 
705
                window = DiffWindow()
 
706
                window.set_diff_text(path, lines)
 
707
            else:
 
708
                window = MergeDirectiveWindow(directive, path)
 
709
                window.set_diff_text(path, directive.patch.splitlines(True))
 
710
            window.show()
 
711
            gtk = self.open_display()
 
712
            window.connect("destroy", gtk.main_quit)
 
713
        except Exception, e:
 
714
            from dialog import error_dialog
 
715
            error_dialog('Error', str(e))
 
716
            raise
 
717
        gtk.main()
 
718
 
 
719
 
 
720
register_command(cmd_ghandle_patch)
 
721
 
651
722
 
652
723
import gettext
653
724
gettext.install('olive-gtk')
654
725
 
655
 
# Let's create a specialized alias to protect '_' from being erased by other
656
 
# uses of '_' as an anonymous variable (think pdb for one).
657
 
_i18n = gettext.gettext
658
726
 
659
727
class NoDisplayError(BzrCommandError):
660
728
    """gtk could not find a proper display"""