/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: Szilveszter Farkas (Phanatic)
  • Date: 2007-02-01 10:52:33 UTC
  • mto: (157.1.2 trunk) (170.1.3 trunk)
  • mto: This revision was merged to the branch mainline in revision 141.
  • Revision ID: szilveszter.farkas@gmail.com-20070201105233-q0azmkxxawlyhemi
Improved Branch dialog. Refactored Checkout dialog.

Show diffs side-by-side

added added

removed removed

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