/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: Mateusz Korniak
  • Date: 2007-07-21 13:16:33 UTC
  • mto: This revision was merged to the branch mainline in revision 248.
  • Revision ID: matkor@laptop-hp-20070721131633-t40kxs20j1q2fvvc
Context menu "Remove and delete added"
Acts like "Remove" but also deletes file locally.

Show diffs side-by-side

added added

removed removed

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