/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

Commit messages never contain config options

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
        super(CheckoutDialog, self).__init__(
 
37
            title="Checkout - Olive", parent=parent, flags=0,
 
38
            buttons=(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL))
 
39
 
 
40
        # Get arguments
 
41
        self.path = path
 
42
 
 
43
        # Create the widgets
 
44
        self._button_checkout = Gtk.Button(_i18n("Check_out"), use_underline=True)
 
45
        self._button_revision = Gtk.Button('')
 
46
        self._image_browse = Gtk.Image()
 
47
        self._filechooser = Gtk.FileChooserButton(_i18n("Please select a folder"))
 
48
        self._combo = Gtk.ComboBox.new_with_entry()
 
49
        self._label_location = Gtk.Label(label=_i18n("Branch location:"))
 
50
        self._label_destination = Gtk.Label(label=_i18n("Destination:"))
 
51
        self._label_nick = Gtk.Label(label=_i18n("Branch nick:"))
 
52
        self._label_revision = Gtk.Label(label=_i18n("Revision:"))
 
53
        self._hbox_revision = Gtk.HBox()
 
54
        self._entry_revision = Gtk.Entry()
 
55
        self._entry_nick = Gtk.Entry()
 
56
        self._check_lightweight = Gtk.CheckButton(_i18n("_Lightweight checkout"),
 
57
                                                  use_underline=True)
 
58
 
 
59
        # Set callbacks
 
60
        self._button_checkout.connect('clicked', self._on_checkout_clicked)
 
61
        self._button_revision.connect('clicked', self._on_revision_clicked)
 
62
        self._combo.get_child().connect('focus-out-event', self._on_combo_changed)
 
63
 
 
64
        # Create the table and pack the widgets into it
 
65
        self._table = Gtk.Table(rows=3, columns=2)
 
66
        self._table.attach(self._label_location, 0, 1, 0, 1)
 
67
        self._table.attach(self._label_destination, 0, 1, 1, 2)
 
68
        self._table.attach(self._label_nick, 0, 1, 2, 3)
 
69
        self._table.attach(self._label_revision, 0, 1, 3, 4)
 
70
        self._table.attach(self._combo, 1, 2, 0, 1)
 
71
        self._table.attach(self._filechooser, 1, 2, 1, 2)
 
72
        self._table.attach(self._entry_nick, 1, 2, 2, 3)
 
73
        self._table.attach(self._hbox_revision, 1, 2, 3, 4)
 
74
        self._table.attach(self._check_lightweight, 1, 2, 4, 5)
 
75
 
 
76
        # Set properties
 
77
        self._image_browse.set_from_stock(Gtk.STOCK_OPEN, Gtk.IconSize.BUTTON)
 
78
        self._button_revision.set_image(self._image_browse)
 
79
        self._button_revision.set_sensitive(False)
 
80
        self._filechooser.set_action(Gtk.FileChooserAction.SELECT_FOLDER)
 
81
        self._label_location.set_alignment(0, 0.5)
 
82
        self._label_destination.set_alignment(0, 0.5)
 
83
        self._label_nick.set_alignment(0, 0.5)
 
84
        self._label_revision.set_alignment(0, 0.5)
 
85
        self._table.set_row_spacings(3)
 
86
        self._table.set_col_spacings(3)
 
87
        self.get_content_area().set_spacing(3)
 
88
        if self.path is not None:
 
89
            self._filechooser.set_filename(self.path)
 
90
        if remote_path is not None:
 
91
            self._combo.get_child().set_text(remote_path)
 
92
 
 
93
        # Pack some widgets
 
94
        self._hbox_revision.pack_start(self._entry_revision, True, True, 0)
 
95
        self._hbox_revision.pack_start(self._button_revision, False, False, 0)
 
96
        self.get_content_area().add(self._table)
 
97
        self.action_area.pack_end(self._button_checkout, False, False, 0)
 
98
 
 
99
        # Show the dialog
 
100
        self.get_content_area().show_all()
 
101
 
 
102
        # Build checkout history
 
103
        self._history = UrlHistory(GlobalConfig(), 'branch_history')
 
104
        self._build_history()
 
105
 
 
106
    def _build_history(self):
 
107
        """ Build up the checkout history. """
 
108
        self._combo_model = Gtk.ListStore(str)
 
109
 
 
110
        for item in self._history.get_entries():
 
111
            self._combo_model.append([ item ])
 
112
 
 
113
        self._combo.set_model(self._combo_model)
 
114
        self._combo.set_entry_text_column(0)
 
115
 
 
116
    def _get_last_revno(self):
 
117
        """ Get the revno of the last revision (if any). """
 
118
        location = self._combo.get_child().get_text()
 
119
        try:
 
120
            br = Branch.open(location)
 
121
        except:
 
122
            return None
 
123
        else:
 
124
            return br.revno()
 
125
 
 
126
    def _on_revision_clicked(self, button):
 
127
        """ Browse for revision button clicked handler. """
 
128
        from revbrowser import RevisionBrowser
 
129
 
 
130
        location = self._combo.get_child().get_text()
 
131
 
 
132
        try:
 
133
            br = Branch.open(location)
 
134
        except:
 
135
            return
 
136
        else:
 
137
            revb = RevisionBrowser(br, self)
 
138
            response = revb.run()
 
139
            if response != Gtk.ResponseType.NONE:
 
140
                revb.hide()
 
141
 
 
142
                if response == Gtk.ResponseType.OK:
 
143
                    if revb.selected_revno is not None:
 
144
                        self._entry_revision.set_text(revb.selected_revno)
 
145
 
 
146
                revb.destroy()
 
147
 
 
148
    @show_bzr_error
 
149
    def _on_checkout_clicked(self, button):
 
150
        """ Checkout button clicked handler. """
 
151
        location = self._combo.get_child().get_text()
64
152
        if location is '':
65
 
            error_dialog(_('Missing branch location'),
66
 
                         _('You must specify a branch location.'))
 
153
            error_dialog(_i18n('Missing branch location'),
 
154
                         _i18n('You must specify a branch location.'))
67
155
            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
 
        
 
156
 
 
157
        destination = self._filechooser.get_filename()
77
158
        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()
 
159
            revno = int(self._entry_revision.get_text())
 
160
        except:
 
161
            revno = None
 
162
 
 
163
        nick = self._entry_nick.get_text()
 
164
        if nick is '':
 
165
            nick = os.path.basename(location.rstrip("/\\"))
 
166
 
 
167
        br_from = Branch.open(location)
 
168
 
 
169
        revision_id = br_from.get_rev_id(revno)
 
170
        lightweight = self._check_lightweight.get_active()
 
171
        to_location = destination + os.sep + nick
 
172
 
 
173
        os.mkdir(to_location)
 
174
 
 
175
        br_from.create_checkout(to_location, revision_id, lightweight)
 
176
 
 
177
        self._history.add_entry(location)
 
178
 
 
179
        self.response(Gtk.ResponseType.OK)
 
180
 
 
181
    def _on_combo_changed(self, widget, event):
 
182
        """ We try to get the last revision if focus lost. """
 
183
        rev = self._get_last_revno()
 
184
        if rev is None:
 
185
            self._entry_revision.set_text(_i18n('N/A'))
 
186
            self._button_revision.set_sensitive(False)
 
187
        else:
 
188
            self._entry_revision.set_text(str(rev))
 
189
            self._button_revision.set_sensitive(True)
 
190
            if self._entry_nick.get_text() == '':
 
191
                self._entry_nick.set_text(os.path.basename(self._combo.get_child().get_text().rstrip("/\\")))