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

  • Committer: Curtis Hovey
  • Date: 2011-09-04 20:04:03 UTC
  • mto: This revision was merged to the branch mainline in revision 741.
  • Revision ID: sinzui.is@verizon.net-20110904200403-t38t3o2q1j600dho
Added missing tests for BranchTreeModel and CellRendererGraph.

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
import os
18
18
 
19
 
try:
20
 
    import pygtk
21
 
    pygtk.require("2.0")
22
 
except:
23
 
    pass
24
 
 
25
 
import gtk
26
 
import gtk.glade
 
19
from gi.repository import GObject
 
20
from gi.repository import Gtk
27
21
 
28
22
from bzrlib.branch import Branch
29
 
import bzrlib.bzrdir as bzrdir
30
 
import bzrlib.errors as errors
31
 
import bzrlib.osutils
32
 
 
33
 
from dialog import error_dialog
34
 
from guifiles import GLADEFILENAME
35
 
 
36
 
 
37
 
class OliveCheckout:
38
 
    """ Display checkout dialog and perform the needed operations. """
39
 
    def __init__(self, path=None):
 
23
from bzrlib.config import GlobalConfig
 
24
 
 
25
from bzrlib.plugins.gtk.dialog import error_dialog
 
26
from bzrlib.plugins.gtk.errors import show_bzr_error
 
27
from bzrlib.plugins.gtk.history import UrlHistory
 
28
from bzrlib.plugins.gtk.i18n import _i18n
 
29
 
 
30
 
 
31
class CheckoutDialog(Gtk.Dialog):
 
32
    """ New implementation of the Checkout dialog. """
 
33
 
 
34
    def __init__(self, path=None, parent=None, remote_path=None):
40
35
        """ Initialize the Checkout dialog. """
41
 
        self.glade = gtk.glade.XML(GLADEFILENAME, 'window_checkout', 'olive-gtk')
42
 
        
43
 
        self.window = self.glade.get_widget('window_checkout')
44
 
        
45
 
        # Dictionary for signal_autoconnect
46
 
        dic = { "on_button_checkout_checkout_clicked": self.checkout,
47
 
                "on_button_checkout_cancel_clicked": self.close }
48
 
        
49
 
        # Connect the signals to the handlers
50
 
        self.glade.signal_autoconnect(dic)
51
 
        
52
 
        # Save FileChooser state
53
 
        self.filechooser = self.glade.get_widget('filechooserbutton_checkout')
54
 
        if path is not None:
55
 
            self.filechooser.set_filename(path)
56
 
 
57
 
    def display(self):
58
 
        """ Display the Checkout dialog. """
59
 
        self.window.show_all()
60
 
    
61
 
    def checkout(self, widget):
62
 
        entry_location = self.glade.get_widget('entry_checkout_location')
63
 
        location = entry_location.get_text()
 
36
        GObject.GObject.__init__(self, title="Checkout - Olive",
 
37
                                  parent=parent,
 
38
                                  flags=0,
 
39
                                  buttons=(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL))
 
40
 
 
41
        # Get arguments
 
42
        self.path = path
 
43
 
 
44
        # Create the widgets
 
45
        self._button_checkout = Gtk.Button(_i18n("Check_out"), use_underline=True)
 
46
        self._button_revision = Gtk.Button('')
 
47
        self._image_browse = Gtk.Image()
 
48
        self._filechooser = Gtk.FileChooserButton(_i18n("Please select a folder"))
 
49
        self._combo = Gtk.ComboBox.new_with_entry()
 
50
        self._label_location = Gtk.Label(label=_i18n("Branch location:"))
 
51
        self._label_destination = Gtk.Label(label=_i18n("Destination:"))
 
52
        self._label_nick = Gtk.Label(label=_i18n("Branch nick:"))
 
53
        self._label_revision = Gtk.Label(label=_i18n("Revision:"))
 
54
        self._hbox_revision = Gtk.HBox()
 
55
        self._entry_revision = Gtk.Entry()
 
56
        self._entry_nick = Gtk.Entry()
 
57
        self._check_lightweight = Gtk.CheckButton(_i18n("_Lightweight checkout"),
 
58
                                                  use_underline=True)
 
59
 
 
60
        # Set callbacks
 
61
        self._button_checkout.connect('clicked', self._on_checkout_clicked)
 
62
        self._button_revision.connect('clicked', self._on_revision_clicked)
 
63
        self._combo.get_child().connect('focus-out-event', self._on_combo_changed)
 
64
 
 
65
        # Create the table and pack the widgets into it
 
66
        self._table = Gtk.Table(rows=3, columns=2)
 
67
        self._table.attach(self._label_location, 0, 1, 0, 1)
 
68
        self._table.attach(self._label_destination, 0, 1, 1, 2)
 
69
        self._table.attach(self._label_nick, 0, 1, 2, 3)
 
70
        self._table.attach(self._label_revision, 0, 1, 3, 4)
 
71
        self._table.attach(self._combo, 1, 2, 0, 1)
 
72
        self._table.attach(self._filechooser, 1, 2, 1, 2)
 
73
        self._table.attach(self._entry_nick, 1, 2, 2, 3)
 
74
        self._table.attach(self._hbox_revision, 1, 2, 3, 4)
 
75
        self._table.attach(self._check_lightweight, 1, 2, 4, 5)
 
76
 
 
77
        # Set properties
 
78
        self._image_browse.set_from_stock(Gtk.STOCK_OPEN, Gtk.IconSize.BUTTON)
 
79
        self._button_revision.set_image(self._image_browse)
 
80
        self._button_revision.set_sensitive(False)
 
81
        self._filechooser.set_action(Gtk.FileChooserAction.SELECT_FOLDER)
 
82
        self._label_location.set_alignment(0, 0.5)
 
83
        self._label_destination.set_alignment(0, 0.5)
 
84
        self._label_nick.set_alignment(0, 0.5)
 
85
        self._label_revision.set_alignment(0, 0.5)
 
86
        self._table.set_row_spacings(3)
 
87
        self._table.set_col_spacings(3)
 
88
        self.vbox.set_spacing(3)
 
89
        if self.path is not None:
 
90
            self._filechooser.set_filename(self.path)
 
91
        if remote_path is not None:
 
92
            self._combo.get_child().set_text(remote_path)
 
93
 
 
94
        # Pack some widgets
 
95
        self._hbox_revision.pack_start(self._entry_revision, True, True)
 
96
        self._hbox_revision.pack_start(self._button_revision, False, False)
 
97
        self.vbox.add(self._table)
 
98
        self.action_area.pack_end(self._button_checkout)
 
99
 
 
100
        # Show the dialog
 
101
        self.vbox.show_all()
 
102
 
 
103
        # Build checkout history
 
104
        self._history = UrlHistory(GlobalConfig(), 'branch_history')
 
105
        self._build_history()
 
106
 
 
107
    def _build_history(self):
 
108
        """ Build up the checkout history. """
 
109
        self._combo_model = Gtk.ListStore(str)
 
110
 
 
111
        for item in self._history.get_entries():
 
112
            self._combo_model.append([ item ])
 
113
 
 
114
        self._combo.set_model(self._combo_model)
 
115
        self._combo.set_entry_text_column(0)
 
116
 
 
117
    def _get_last_revno(self):
 
118
        """ Get the revno of the last revision (if any). """
 
119
        location = self._combo.get_child().get_text()
 
120
        try:
 
121
            br = Branch.open(location)
 
122
        except:
 
123
            return None
 
124
        else:
 
125
            return br.revno()
 
126
 
 
127
    def _on_revision_clicked(self, button):
 
128
        """ Browse for revision button clicked handler. """
 
129
        from revbrowser import RevisionBrowser
 
130
 
 
131
        location = self._combo.get_child().get_text()
 
132
 
 
133
        try:
 
134
            br = Branch.open(location)
 
135
        except:
 
136
            return
 
137
        else:
 
138
            revb = RevisionBrowser(br, self)
 
139
            response = revb.run()
 
140
            if response != Gtk.ResponseType.NONE:
 
141
                revb.hide()
 
142
 
 
143
                if response == Gtk.ResponseType.OK:
 
144
                    if revb.selected_revno is not None:
 
145
                        self._entry_revision.set_text(revb.selected_revno)
 
146
 
 
147
                revb.destroy()
 
148
 
 
149
    @show_bzr_error
 
150
    def _on_checkout_clicked(self, button):
 
151
        """ Checkout button clicked handler. """
 
152
        location = self._combo.get_child().get_text()
64
153
        if location is '':
65
 
            error_dialog(_('Missing branch location'),
66
 
                         _('You must specify a branch location.'))
 
154
            error_dialog(_i18n('Missing branch location'),
 
155
                         _i18n('You must specify a branch location.'))
67
156
            return
68
 
        
69
 
        destination = self.filechooser.get_filename()
70
 
        
71
 
        spinbutton_revno = self.glade.get_widget('spinbutton_checkout_revno')
72
 
        revno = spinbutton_revno.get_value_as_int()
73
 
        
74
 
        checkbutton_lightweight = self.glade.get_widget('checkbutton_checkout_lightweight')
75
 
        lightweight = checkbutton_lightweight.get_active()
76
 
        
 
157
 
 
158
        destination = self._filechooser.get_filename()
77
159
        try:
78
 
            source = Branch.open(location)
79
 
            rev_id = source.get_rev_id(revno)
80
 
            
81
 
            # if the source and destination are the same, 
82
 
            # and there is no working tree,
83
 
            # then reconstitute a branch
84
 
            if (bzrlib.osutils.abspath(destination) ==
85
 
                bzrlib.osutils.abspath(location)):
86
 
                try:
87
 
                    source.bzrdir.open_workingtree()
88
 
                except errors.NoWorkingTree:
89
 
                    source.bzrdir.create_workingtree()
90
 
                    return
91
 
 
92
 
            destination = destination + '/' + os.path.basename(location.rstrip("/\\"))
93
 
            
94
 
            os.mkdir(destination)
95
 
 
96
 
            old_format = bzrlib.bzrdir.BzrDirFormat.get_default_format()
97
 
            bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
98
 
 
99
 
            try:
100
 
                if lightweight:
101
 
                    checkout = bzrdir.BzrDirMetaFormat1().initialize(destination)
102
 
                    bzrlib.branch.BranchReferenceFormat().initialize(checkout, source)
103
 
                else:
104
 
                    checkout_branch = bzrlib.bzrdir.BzrDir.create_branch_convenience(
105
 
                        destination, force_new_tree=False)
106
 
                    checkout = checkout_branch.bzrdir
107
 
                    checkout_branch.bind(source)
108
 
                    if rev_id is not None:
109
 
                        rh = checkout_branch.revno_history()
110
 
                        checkout_branch.set_revno_history(rh[:rh.index(rev_id) + 1])
111
 
 
112
 
                checkout.create_workingtree(rev_id)
113
 
            finally:
114
 
                bzrlib.bzrdir.BzrDirFormat.set_default_format(old_format)
115
 
        except errors.NotBranchError, errmsg:
116
 
            error_dialog(_('Location is not a branch'),
117
 
                         _('The specified location has to be a branch.'))
118
 
            return
119
 
        except errors.TargetAlreadyExists, errmsg:
120
 
            error_dialog(_('Target already exists'),
121
 
                         _('Target directory (%s)\nalready exists. Please select another target.') % errmsg)
122
 
            return
123
 
        except errors.NonExistingParent, errmsg:
124
 
            error_dialog(_('Non existing parent directory'),
125
 
                         _("The parent directory (%s)\ndoesn't exist.") % errmsg)
126
 
            return
127
 
        
128
 
        self.close()
129
 
        self.comm.refresh_right()
130
 
 
131
 
    def close(self, widget=None):
132
 
        self.window.destroy()
 
160
            revno = int(self._entry_revision.get_text())
 
161
        except:
 
162
            revno = None
 
163
 
 
164
        nick = self._entry_nick.get_text()
 
165
        if nick is '':
 
166
            nick = os.path.basename(location.rstrip("/\\"))
 
167
 
 
168
        br_from = Branch.open(location)
 
169
 
 
170
        revision_id = br_from.get_rev_id(revno)
 
171
        lightweight = self._check_lightweight.get_active()
 
172
        to_location = destination + os.sep + nick
 
173
 
 
174
        os.mkdir(to_location)
 
175
 
 
176
        br_from.create_checkout(to_location, revision_id, lightweight)
 
177
 
 
178
        self._history.add_entry(location)
 
179
 
 
180
        self.response(Gtk.ResponseType.OK)
 
181
 
 
182
    def _on_combo_changed(self, widget, event):
 
183
        """ We try to get the last revision if focus lost. """
 
184
        rev = self._get_last_revno()
 
185
        if rev is None:
 
186
            self._entry_revision.set_text(_i18n('N/A'))
 
187
            self._button_revision.set_sensitive(False)
 
188
        else:
 
189
            self._entry_revision.set_text(str(rev))
 
190
            self._button_revision.set_sensitive(True)
 
191
            if self._entry_nick.get_text() == '':
 
192
                self._entry_nick.set_text(os.path.basename(self._combo.get_child().get_text().rstrip("/\\")))