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

  • Committer: Jelmer Vernooij
  • Date: 2007-02-01 15:50:40 UTC
  • Revision ID: jelmer@samba.org-20070201155040-3hq4mfbxs99kzazy
add framework for tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
23
23
    pass
24
24
 
25
25
import gtk
26
 
 
27
 
from errors import show_bzr_error
 
26
import gtk.glade
28
27
 
29
28
from bzrlib.branch import Branch
30
 
from bzrlib.config import GlobalConfig
 
29
import bzrlib.bzrdir as bzrdir
 
30
import bzrlib.errors as errors
 
31
import bzrlib.osutils
31
32
 
32
33
from dialog import error_dialog
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):
 
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):
40
40
        """ Initialize the Checkout dialog. """
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()
 
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()
162
64
        if location is '':
163
65
            error_dialog(_('Missing branch location'),
164
66
                         _('You must specify a branch location.'))
165
67
            return
166
68
        
167
 
        destination = self._filechooser.get_filename()
 
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
        
168
77
        try:
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("/\\")))
 
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()