/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

  • Committer: Jelmer Vernooij
  • Date: 2006-09-30 10:21:43 UTC
  • Revision ID: jelmer@samba.org-20060930102143-c0ef64d6ca860c21
Merge some files from Olive and bzr-gtk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
12
12
# along with this program; if not, write to the Free Software
13
13
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
14
14
 
15
 
"""Graphical support for Bazaar using GTK.
16
 
 
17
 
This plugin includes:
18
 
gannotate         GTK+ annotate. 
19
 
gbranch           GTK+ branching. 
20
 
gcheckout         GTK+ checkout. 
21
 
gcommit           GTK+ commit dialog.
22
 
gconflicts        GTK+ conflicts. 
23
 
gdiff             Show differences in working tree in a GTK+ Window. 
24
 
ginit             Initialise a new branch.
25
 
ginfo             GTK+ branch info dialog
26
 
gloom             GTK+ loom browse dialog
27
 
gmerge            GTK+ merge dialog
28
 
gmissing          GTK+ missing revisions dialog. 
29
 
gpreferences      GTK+ preferences dialog. 
30
 
gpush             GTK+ push.
31
 
gsend             GTK+ send merge directive.
32
 
gstatus           GTK+ status dialog.
33
 
gtags             Manage branch tags.
34
 
visualise         Graphically visualise this branch. 
35
 
"""
36
 
 
37
 
import os
38
 
import sys
39
 
 
40
 
if getattr(sys, "frozen", None) is not None: # we run bzr.exe
41
 
 
42
 
    # FIXME: Unless a better packaging solution is found, the following
43
 
    # provides a workaround for https://bugs.launchpad.net/bzr/+bug/388790 Also
44
 
    # see https://code.edge.launchpad.net/~vila/bzr-gtk/388790-windows-setup
45
 
    # for more details about while it's needed.
46
 
 
47
 
    # NOTE: _lib must be ahead of bzrlib or sax.saxutils (in olive) fails
48
 
    here = os.path.dirname(__file__)
49
 
    sys.path.insert(0, os.path.join(here, '_lib'))
50
 
    sys.path.append(os.path.join(here, '_lib/gtk-2.0'))
51
 
 
52
 
 
53
 
import bzrlib
54
 
import bzrlib.api
55
 
from bzrlib import (
56
 
    branch,
57
 
    config,
58
 
    errors,
59
 
    )
60
 
from bzrlib.commands import plugin_cmds
61
 
 
62
 
 
63
 
version_info = (0, 99, 0, 'dev', 1)
64
 
 
65
 
if version_info[3] == 'final':
66
 
    version_string = '%d.%d.%d' % version_info[:3]
67
 
else:
68
 
    version_string = '%d.%d.%d%s%d' % version_info
69
 
__version__ = version_string
70
 
 
71
 
COMPATIBLE_BZR_VERSIONS = [(1, 6, 0), (1, 7, 0), (1, 8, 0), (1, 9, 0),
72
 
                           (1, 10, 0), (1, 11, 0), (1, 12, 0), (1, 13, 0),
73
 
                           (1, 15, 0),
74
 
                           (1, 17, 0),
75
 
                           (2, 1, 0),
76
 
                           (2, 2, 0),
77
 
                           ]
78
 
 
79
 
bzrlib.api.require_any_api(bzrlib, COMPATIBLE_BZR_VERSIONS)
80
 
 
81
 
if __name__ != 'bzrlib.plugins.gtk':
82
 
    from bzrlib.trace import warning
83
 
    warning("Not running as bzrlib.plugins.gtk, things may break.")
84
 
 
85
 
def import_pygtk():
86
 
    try:
 
15
"""GTK+ frontends to Bazaar commands """
 
16
 
 
17
from bzrlib.commands import Command, register_command, display_command
 
18
from bzrlib.errors import NotVersionedError, BzrCommandError, NoSuchFile
 
19
from bzrlib.commands import Command, register_command
 
20
from bzrlib.option import Option
 
21
from bzrlib.branch import Branch
 
22
from bzrlib.workingtree import WorkingTree
 
23
from bzrlib.bzrdir import BzrDir
 
24
 
 
25
__version__ = '0.11.0'
 
26
 
 
27
class cmd_gbranch(Command):
 
28
    """GTK+ branching.
 
29
    
 
30
    """
 
31
 
 
32
    def run(self):
87
33
        import pygtk
88
 
    except ImportError:
89
 
        raise errors.BzrCommandError("PyGTK not installed.")
90
 
    pygtk.require('2.0')
91
 
    return pygtk
92
 
 
93
 
 
94
 
def set_ui_factory():
95
 
    import_pygtk()
96
 
    from ui import GtkUIFactory
97
 
    import bzrlib.ui
98
 
    bzrlib.ui.ui_factory = GtkUIFactory()
99
 
 
100
 
 
101
 
def data_basedirs():
102
 
    return [os.path.dirname(__file__),
103
 
             "/usr/share/bzr-gtk", 
104
 
             "/usr/local/share/bzr-gtk"]
105
 
 
106
 
 
107
 
def data_path(*args):
108
 
    for basedir in data_basedirs():
109
 
        path = os.path.join(basedir, *args)
110
 
        if os.path.exists(path):
111
 
            return path
112
 
    return None
113
 
 
114
 
 
115
 
def icon_path(*args):
116
 
    return data_path(os.path.join('icons', *args))
117
 
 
118
 
 
119
 
def open_display():
120
 
    pygtk = import_pygtk()
121
 
    try:
 
34
        pygtk.require("2.0")
 
35
        try:
 
36
            import gtk
 
37
        except RuntimeError, e:
 
38
            if str(e) == "could not open display":
 
39
                raise NoDisplayError
 
40
 
 
41
        from clone import CloneDialog
 
42
 
 
43
        window = CloneDialog()
 
44
        if window.run() == gtk.RESPONSE_OK:
 
45
            bzrdir = BzrDir.open(window.url)
 
46
            bzrdir.sprout(window.dest_path)
 
47
 
 
48
register_command(cmd_gbranch)
 
49
 
 
50
class cmd_gdiff(Command):
 
51
    """Show differences in working tree in a GTK+ Window.
 
52
    
 
53
    Otherwise, all changes for the tree are listed.
 
54
    """
 
55
    takes_args = ['filename?']
 
56
    takes_options = ['revision']
 
57
 
 
58
    @display_command
 
59
    def run(self, revision=None, filename=None):
 
60
        wt = WorkingTree.open_containing(".")[0]
 
61
        branch = wt.branch
 
62
        if revision is not None:
 
63
            if len(revision) == 1:
 
64
                tree1 = wt
 
65
                revision_id = revision[0].in_history(branch).rev_id
 
66
                tree2 = branch.repository.revision_tree(revision_id)
 
67
            elif len(revision) == 2:
 
68
                revision_id_0 = revision[0].in_history(branch).rev_id
 
69
                tree2 = branch.repository.revision_tree(revision_id_0)
 
70
                revision_id_1 = revision[1].in_history(branch).rev_id
 
71
                tree1 = branch.repository.revision_tree(revision_id_1)
 
72
        else:
 
73
            tree1 = wt
 
74
            tree2 = tree1.basis_tree()
 
75
 
 
76
        from bzrlib.plugins.gtk.viz.diffwin import DiffWindow
122
77
        import gtk
123
 
    except RuntimeError, e:
124
 
        if str(e) == "could not open display":
125
 
            raise NoDisplayError
126
 
    set_ui_factory()
127
 
    return gtk
128
 
 
129
 
 
130
 
commands = {
131
 
    "gannotate": ["gblame", "gpraise"],
132
 
    "gbranch": [],
133
 
    "gcheckout": [],
134
 
    "gcommit": ["gci"],
135
 
    "gconflicts": [],
136
 
    "gdiff": [],
137
 
    "ginit": [],
138
 
    "ginfo": [],
139
 
    "gmerge": [],
140
 
    "gmissing": [],
141
 
    "gpreferences": [],
142
 
    "gpush": [],
143
 
    "gselftest": [],
144
 
    "gsend": [],
145
 
    "gstatus": ["gst"],
146
 
    "gtags": [],
147
 
    "visualise": ["visualize", "vis", "viz", 'glog'],
148
 
    }
149
 
 
150
 
try:
151
 
    from bzrlib.plugins import loom
152
 
except ImportError:
153
 
    pass # Loom plugin doesn't appear to be present
154
 
else:
155
 
    commands["gloom"] = []
156
 
 
157
 
for cmd, aliases in commands.iteritems():
158
 
    plugin_cmds.register_lazy("cmd_%s" % cmd, aliases,
159
 
                              "bzrlib.plugins.gtk.commands")
160
 
 
161
 
def save_commit_messages(*args):
162
 
    from bzrlib.plugins.gtk import commit
163
 
    commit.save_commit_messages(*args)
164
 
 
165
 
branch.Branch.hooks.install_named_hook('post_uncommit',
166
 
                                       save_commit_messages,
167
 
                                       "Saving commit messages for gcommit")
168
 
 
169
 
import gettext
170
 
gettext.install('olive-gtk')
171
 
 
172
 
# Let's create a specialized alias to protect '_' from being erased by other
173
 
# uses of '_' as an anonymous variable (think pdb for one).
174
 
_i18n = gettext.gettext
175
 
 
176
 
class NoDisplayError(errors.BzrCommandError):
 
78
        window = DiffWindow()
 
79
        window.connect("destroy", lambda w: gtk.main_quit())
 
80
        window.set_diff("Working Tree", tree1, tree2)
 
81
        if filename is not None:
 
82
            tree_filename = tree1.relpath(filename)
 
83
            try:
 
84
                window.set_file(tree_filename)
 
85
            except NoSuchFile:
 
86
                if (tree1.inventory.path2id(tree_filename) is None and 
 
87
                    tree2.inventory.path2id(tree_filename) is None):
 
88
                    raise NotVersionedError(filename)
 
89
                raise BzrCommandError('No changes found for file "%s"' % 
 
90
                                      filename)
 
91
        window.show()
 
92
 
 
93
        gtk.main()
 
94
 
 
95
register_command(cmd_gdiff)
 
96
 
 
97
class cmd_visualise(Command):
 
98
    """Graphically visualise this branch.
 
99
 
 
100
    Opens a graphical window to allow you to see the history of the branch
 
101
    and relationships between revisions in a visual manner,
 
102
 
 
103
    The default starting point is latest revision on the branch, you can
 
104
    specify a starting point with -r revision.
 
105
    """
 
106
    takes_options = [
 
107
        "revision",
 
108
        Option('limit', "maximum number of revisions to display",
 
109
               int, 'count')]
 
110
    takes_args = [ "location?" ]
 
111
    aliases = [ "visualize", "vis", "viz" ]
 
112
 
 
113
    def run(self, location=".", revision=None, limit=None):
 
114
        (branch, path) = Branch.open_containing(location)
 
115
        branch.lock_read()
 
116
        branch.repository.lock_read()
 
117
        try:
 
118
            if revision is None:
 
119
                revid = branch.last_revision()
 
120
                if revid is None:
 
121
                    return
 
122
            else:
 
123
                (revno, revid) = revision[0].in_history(branch)
 
124
 
 
125
            from viz.bzrkapp import BzrkApp
 
126
                
 
127
            app = BzrkApp()
 
128
            app.show(branch, revid, limit)
 
129
        finally:
 
130
            branch.repository.unlock()
 
131
            branch.unlock()
 
132
        app.main()
 
133
 
 
134
 
 
135
register_command(cmd_visualise)
 
136
 
 
137
class cmd_gannotate(Command):
 
138
    """GTK+ annotate.
 
139
    
 
140
    Browse changes to FILENAME line by line in a GTK+ window.
 
141
    """
 
142
 
 
143
    takes_args = ["filename", "line?"]
 
144
    takes_options = [
 
145
        Option("all", help="show annotations on all lines"),
 
146
        Option("plain", help="don't highlight annotation lines"),
 
147
        Option("line", type=int, argname="lineno",
 
148
               help="jump to specified line number")
 
149
    ]
 
150
    aliases = ["gblame", "gpraise"]
 
151
    
 
152
    def run(self, filename, all=False, plain=False, line='1'):
 
153
        import pygtk
 
154
        pygtk.require("2.0")
 
155
 
 
156
        try:
 
157
            import gtk
 
158
        except RuntimeError, e:
 
159
            if str(e) == "could not open display":
 
160
                raise NoDisplayError
 
161
 
 
162
        try:
 
163
            line = int(line)
 
164
        except ValueError:
 
165
            raise BzrCommandError('Line argument ("%s") is not a number.' % 
 
166
                                  line)
 
167
 
 
168
        from annotate.gannotate import GAnnotateWindow
 
169
        from annotate.config import GAnnotateConfig
 
170
 
 
171
        (wt, path) = WorkingTree.open_containing(filename)
 
172
        branch = wt.branch
 
173
 
 
174
        file_id = wt.path2id(path)
 
175
 
 
176
        if file_id is None:
 
177
            raise NotVersionedError(filename)
 
178
 
 
179
        window = GAnnotateWindow(all, plain)
 
180
        window.connect("destroy", lambda w: gtk.main_quit())
 
181
        window.set_title(path + " - gannotate")
 
182
        config = GAnnotateConfig(window)
 
183
        window.show()
 
184
        branch.lock_read()
 
185
        try:
 
186
            window.annotate(branch, file_id)
 
187
        finally:
 
188
            branch.unlock()
 
189
        window.jump_to_line(line)
 
190
        
 
191
        gtk.main()
 
192
 
 
193
register_command(cmd_gannotate)
 
194
 
 
195
class cmd_gcommit(Command):
 
196
    """GTK+ commit dialog
 
197
 
 
198
    Graphical user interface for committing revisions"""
 
199
    
 
200
    takes_args = []
 
201
    takes_options = []
 
202
 
 
203
    def run(self, filename=None):
 
204
        import pygtk
 
205
        pygtk.require("2.0")
 
206
 
 
207
        try:
 
208
            import gtk
 
209
        except RuntimeError, e:
 
210
            if str(e) == "could not open display":
 
211
                raise NoDisplayError
 
212
 
 
213
        from commit import GCommitDialog
 
214
        from bzrlib.commit import Commit
 
215
        from bzrlib.errors import (BzrCommandError, PointlessCommit, ConflictsInTree, 
 
216
           StrictCommitFailed)
 
217
 
 
218
        (wt, path) = WorkingTree.open_containing(filename)
 
219
        branch = wt.branch
 
220
 
 
221
        file_id = wt.path2id(path)
 
222
 
 
223
        if file_id is None:
 
224
            raise NotVersionedError(filename)
 
225
 
 
226
        dialog = GCommitDialog(wt)
 
227
        dialog.set_title(path + " - Commit")
 
228
        if dialog.run() != gtk.RESPONSE_CANCEL:
 
229
            Commit().commit(working_tree=wt,message=dialog.message,
 
230
                specific_files=dialog.specific_files)
 
231
 
 
232
register_command(cmd_gcommit)
 
233
 
 
234
class NoDisplayError(BzrCommandError):
177
235
    """gtk could not find a proper display"""
178
236
 
179
237
    def __str__(self):
180
 
        return "No DISPLAY. Unable to run GTK+ application."
181
 
 
182
 
 
183
 
credential_store_registry = getattr(config, "credential_store_registry", None)
184
 
if credential_store_registry is not None:
185
 
    try:
186
 
        credential_store_registry.register_lazy(
187
 
            "gnome-keyring", "bzrlib.plugins.gtk.keyring", "GnomeKeyringCredentialStore",
188
 
            help="The GNOME Keyring.", fallback=True)
189
 
    except TypeError:
190
 
    # Fallback credentials stores were introduced in Bazaar 1.15
191
 
        credential_store_registry.register_lazy(
192
 
            "gnome-keyring", "bzrlib.plugins.gtk.keyring", "GnomeKeyringCredentialStore",
193
 
            help="The GNOME Keyring.")
194
 
 
195
 
 
196
 
def load_tests(basic_tests, module, loader):
197
 
    testmod_names = [
198
 
        'tests',
199
 
        ]
200
 
    import sys
201
 
    default_encoding = sys.getdefaultencoding()
202
 
    try:
203
 
        result = basic_tests
204
 
        try:
205
 
            import_pygtk()
206
 
        except errors.BzrCommandError:
207
 
            return basic_tests
208
 
        basic_tests.addTest(loader.loadTestsFromModuleNames(
209
 
                ["%s.%s" % (__name__, tmn) for tmn in testmod_names]))
210
 
    finally:
211
 
        if sys.getdefaultencoding() != default_encoding:
212
 
            reload(sys)
213
 
            sys.setdefaultencoding(default_encoding)
214
 
    return basic_tests
 
238
        return "No DISPLAY. gannotate is disabled."