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

  • Committer: Jelmer Vernooij
  • Date: 2008-06-29 18:12:29 UTC
  • mto: This revision was merged to the branch mainline in revision 519.
  • Revision ID: jelmer@samba.org-20080629181229-1l2m4cf7vvbyh8qg
Simplify progress bar code, use embedded progress bar inside viz window.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>
 
2
# Copyright (C) 2007 by Jelmer Vernooij <jelmer@samba.org>
2
3
#
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
21
22
    pass
22
23
    
23
24
import gtk
24
 
import gtk.gdk
25
 
import gtk.glade
26
 
 
 
25
 
 
26
from errors import show_bzr_error
 
27
 
 
28
# FIXME: This needs to be public JRV 20070714
 
29
from bzrlib.builtins import _create_prefix
 
30
from bzrlib.config import LocationConfig
27
31
import bzrlib.errors as errors
28
32
 
29
 
from olive import gladefile
30
 
from dialog import error_dialog, info_dialog
31
 
 
32
 
class OlivePush:
33
 
    """ Display Push dialog and perform the needed actions. """
34
 
    def __init__(self, branch):
 
33
from bzrlib.plugins.gtk import _i18n
 
34
from dialog import error_dialog, info_dialog, question_dialog
 
35
 
 
36
from history import UrlHistory
 
37
 
 
38
class PushDialog(gtk.Dialog):
 
39
    """ New implementation of the Push dialog. """
 
40
    def __init__(self, repository, revid, branch=None, parent=None):
35
41
        """ Initialize the Push dialog. """
36
 
        self.glade = gtk.glade.XML(gladefile, 'window_push')
 
42
        gtk.Dialog.__init__(self, title="Push - Olive",
 
43
                                  parent=parent,
 
44
                                  flags=0,
 
45
                                  buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
37
46
        
38
 
        self.window = self.glade.get_widget('window_push')
39
 
 
 
47
        # Get arguments
 
48
        self.repository = repository
 
49
        self.revid = revid
40
50
        self.branch = branch
41
51
        
42
 
        # Dictionary for signal_autoconnect
43
 
        dic = { "on_button_push_push_clicked": self.push,
44
 
                "on_button_push_cancel_clicked": self.close,
45
 
                "on_button_push_test_clicked": self.test,
46
 
                "on_radiobutton_push_stored_toggled": self.stored_toggled,
47
 
                "on_radiobutton_push_specific_toggled": self.specific_toggled, }
48
 
        
49
 
        # Connect the signals to the handlers
50
 
        self.glade.signal_autoconnect(dic)
51
 
        
52
 
        # Get some useful widgets
53
 
        self.radio_stored = self.glade.get_widget('radiobutton_push_stored')
54
 
        self.radio_specific = self.glade.get_widget('radiobutton_push_specific')
55
 
        self.entry_stored = self.glade.get_widget('entry_push_stored')
56
 
        self.entry_location = self.glade.get_widget('entry_push_location')
57
 
        self.check_remember = self.glade.get_widget('checkbutton_push_remember')
58
 
        self.check_overwrite = self.glade.get_widget('checkbutton_push_overwrite')
59
 
        self.check_create = self.glade.get_widget('checkbutton_push_create')
60
 
        self.label_test = self.glade.get_widget('label_push_test')
61
 
        self.image_test = self.glade.get_widget('image_push_test')
62
 
        
63
 
        # Set initial state
64
 
        self.entry_location.set_sensitive(0)
65
 
        self.check_remember.set_sensitive(0)
66
 
        self.check_create.set_sensitive(0)
67
 
        
68
 
        self.entry_stored.set_text(branch.get_push_location())
69
 
    
70
 
    def display(self):
71
 
        """ Display the Push dialog. """
72
 
        self.window.show()
73
 
        self.width, self.height = self.window.get_size()
74
 
    
75
 
    def stored_toggled(self, widget):
76
 
        if widget.get_active():
77
 
            self.entry_stored.set_sensitive(1)
78
 
            self.entry_location.set_sensitive(0)
79
 
            self.check_remember.set_sensitive(0)
80
 
            self.check_create.set_sensitive(0)
81
 
        else:
82
 
            self.entry_stored.set_sensitive(0)
83
 
            self.entry_location.set_sensitive(1)
84
 
            self.check_remember.set_sensitive(1)
85
 
            self.check_create.set_sensitive(1)
86
 
    
87
 
    def specific_toggled(self, widget):
88
 
        if widget.get_active():
89
 
            self.entry_stored.set_sensitive(0)
90
 
            self.entry_location.set_sensitive(1)
91
 
            self.check_remember.set_sensitive(1)
92
 
            self.check_create.set_sensitive(1)
93
 
        else:
94
 
            self.entry_stored.set_sensitive(1)
95
 
            self.entry_location.set_sensitive(0)
96
 
            self.check_remember.set_sensitive(0)
97
 
            self.check_create.set_sensitive(0)
98
 
    
99
 
    def push(self, widget):
 
52
        # Create the widgets
 
53
        self._label_location = gtk.Label(_i18n("Location:"))
 
54
        self._combo = gtk.ComboBoxEntry()
 
55
        self._button_push = gtk.Button(_i18n("_Push"), use_underline=True)
 
56
        self._hbox_location = gtk.HBox()
 
57
        
 
58
        # Set callbacks
 
59
        self._button_push.connect('clicked', self._on_push_clicked)
 
60
        
 
61
        # Set properties
 
62
        self._label_location.set_alignment(0, 0.5)
 
63
        self._hbox_location.set_spacing(3)
 
64
        self.vbox.set_spacing(3)
 
65
        
 
66
        # Pack widgets
 
67
        self._hbox_location.pack_start(self._label_location, False, False)
 
68
        self._hbox_location.pack_start(self._combo, True, True)
 
69
        self.vbox.pack_start(self._hbox_location)
 
70
        self.action_area.pack_end(self._button_push)
 
71
        
 
72
        # Show the dialog
 
73
        self.vbox.show_all()
 
74
        
 
75
        # Build location history
 
76
        self._history = UrlHistory(self.branch.get_config(), 'push_history')
 
77
        self._build_history()
 
78
        
 
79
    def _build_history(self):
 
80
        """ Build up the location history. """
 
81
        self._combo_model = gtk.ListStore(str)
 
82
        for item in self._history.get_entries():
 
83
            self._combo_model.append([ item ])
 
84
        self._combo.set_model(self._combo_model)
 
85
        self._combo.set_text_column(0)
 
86
        
 
87
        if self.branch is not None:
 
88
            location = self.branch.get_push_location()
 
89
            if location is not None:
 
90
                self._combo.get_child().set_text(location)
 
91
    
 
92
    @show_bzr_error
 
93
    def _on_push_clicked(self, widget):
 
94
        """ Push button clicked handler. """
 
95
        location = self._combo.get_child().get_text()
100
96
        revs = 0
101
 
        if self.radio_stored.get_active():
102
 
            try:
103
 
                revs = do_push(self.branch,
104
 
                               overwrite=self.check_overwrite.get_active())
105
 
            except errors.NotBranchError:
106
 
                error_dialog(_('Directory is not a branch'),
107
 
                             _('You can perform this action only in a branch.'))
108
 
                return
109
 
            except errors.DivergedBranches:
110
 
                error_dialog(_('Branches have been diverged'),
111
 
                             _('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
112
 
                return
113
 
        elif self.radio_specific.get_active():
114
 
            location = self.entry_location.get_text()
115
 
            if location == '':
116
 
                error_dialog(_('No location specified'),
117
 
                             _('Please specify a location or use the default.'))
118
 
                return
119
 
            
120
 
            try:
121
 
                revs = do_push(self.branch, location,
122
 
                               self.check_remember.get_active(),
123
 
                               self.check_overwrite.get_active(),
124
 
                               self.check_create.get_active())
125
 
            except errors.NotBranchError:
126
 
                error_dialog(_('Directory is not a branch'),
127
 
                             _('You can perform this action only in a branch.'))
128
 
                return
129
 
            except errors.DivergedBranches:
130
 
                error_dialog(_('Branches have been diverged'),
131
 
                             _('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
132
 
                return
133
 
        
134
 
        self.close()
135
 
        info_dialog(_('Push successful'),
136
 
                    _('%d revision(s) pushed.') % revs)
137
 
    
138
 
    def test(self, widget):
139
 
        """ Test if write access possible. """
140
 
        import re
141
 
        _urlRE = re.compile(r'^(?P<proto>[^:/\\]+)://(?P<path>.*)$')
142
 
        
143
 
        if self.radio_stored.get_active():
144
 
            url = self.entry_stored.get_text()
145
 
        elif self.radio_specific.get_active():
146
 
            url = self.entry_location.get_text()
147
 
        
148
 
        m = _urlRE.match(url)
149
 
        if m:
150
 
            proto = m.groupdict()['proto']
151
 
            if (proto == 'sftp') or (proto == 'file') or (proto == 'ftp'):
152
 
                # have write acces (most probably)
153
 
                self.image_test.set_from_stock(gtk.STOCK_YES, 4)
154
 
                self.label_test.set_markup(_('<b>Write access is probably available</b>'))
155
 
            else:
156
 
                # no write access
157
 
                self.image_test.set_from_stock(gtk.STOCK_NO, 4)
158
 
                self.label_test.set_markup(_('<b>No write access</b>'))
159
 
        else:
160
 
            # couldn't determine
161
 
            self.image_test.set_from_stock(gtk.STOCK_DIALOG_QUESTION, 4)
162
 
            self.label_test.set_markup(_('<b>Could not determine</b>'))
163
 
    
164
 
    def close(self, widget=None):
165
 
        self.window.destroy()
166
 
 
167
 
def do_push(branch, location=None, remember=False, overwrite=False,
168
 
         create_prefix=False):
 
97
        if self.branch is not None and self.branch.get_push_location() is None:
 
98
            response = question_dialog(_i18n('Set default push location'),
 
99
                                       _i18n('There is no default push location set.\nSet %r as default now?') % location)
 
100
            if response == gtk.RESPONSE_OK:
 
101
                self.branch.set_push_location(location)
 
102
 
 
103
        try:
 
104
            revs = do_push(self.branch, location=location, overwrite=False)
 
105
        except errors.DivergedBranches:
 
106
            response = question_dialog(_i18n('Branches have been diverged'),
 
107
                                       _i18n('You cannot push if branches have diverged.\nOverwrite?'))
 
108
            if response == gtk.RESPONSE_YES:
 
109
                revs = do_push(self.branch, location=location, overwrite=True)
 
110
        
 
111
        self._history.add_entry(location)
 
112
        info_dialog(_i18n('Push successful'),
 
113
                    _i18n("%d revision(s) pushed.") % revs)
 
114
        
 
115
        self.response(gtk.RESPONSE_OK)
 
116
 
 
117
def do_push(br_from, location, overwrite):
169
118
    """ Update a mirror of a branch.
170
119
    
171
 
    :param branch: the source branch
 
120
    :param br_from: the source branch
172
121
    
173
122
    :param location: the location of the branch that you'd like to update
174
123
    
175
 
    :param remember: if set, the location will be stored
176
 
    
177
124
    :param overwrite: overwrite target location if it diverged
178
125
    
179
 
    :param create_prefix: create the path leading up to the branch if it doesn't exist
180
 
    
181
126
    :return: number of revisions pushed
182
127
    """
183
 
    from bzrlib.branch import Branch
184
128
    from bzrlib.bzrdir import BzrDir
185
129
    from bzrlib.transport import get_transport
186
130
        
187
 
    br_from = Branch.open_containing(branch)[0]
188
 
    
189
 
    stored_loc = br_from.get_push_location()
190
 
    if location is None:
191
 
        if stored_loc is None:
192
 
            error_dialog(_('Push location is unknown'),
193
 
                                     _('Please specify a location manually.'))
194
 
            return
195
 
        else:
196
 
            location = stored_loc
197
 
 
198
131
    transport = get_transport(location)
199
132
    location_url = transport.base
200
133
 
201
 
    if br_from.get_push_location() is None or remember:
202
 
        br_from.set_push_location(location_url)
203
 
 
204
134
    old_rh = []
205
135
 
206
136
    try:
209
139
    except errors.NotBranchError:
210
140
        # create a branch.
211
141
        transport = transport.clone('..')
212
 
        if not create_prefix:
213
 
            try:
214
 
                relurl = transport.relpath(location_url)
215
 
                transport.mkdir(relurl)
216
 
            except errors.NoSuchFile:
217
 
                error_dialog(_('Non existing parent directory'),
218
 
                             _("The parent directory (%s)\ndoesn't exist.") % location)
 
142
        try:
 
143
            relurl = transport.relpath(location_url)
 
144
            transport.mkdir(relurl)
 
145
        except errors.NoSuchFile:
 
146
            response = question_dialog(_i18n('Non existing parent directory'),
 
147
                         _i18n("The parent directory (%s)\ndoesn't exist. Create?") % location)
 
148
            if response == gtk.RESPONSE_OK:
 
149
                _create_prefix(transport)
 
150
            else:
219
151
                return
220
 
        else:
221
 
            current = transport.base
222
 
            needed = [(transport, transport.relpath(location_url))]
223
 
            while needed:
224
 
                try:
225
 
                    transport, relpath = needed[-1]
226
 
                    transport.mkdir(relpath)
227
 
                    needed.pop()
228
 
                except errors.NoSuchFile:
229
 
                    new_transport = transport.clone('..')
230
 
                    needed.append((new_transport,
231
 
                                   new_transport.relpath(transport.base)))
232
 
                    if new_transport.base == transport.base:
233
 
                        error_dialog(_('Path prefix not created'),
234
 
                                     _("The path leading up to the specified location couldn't\nbe created."))
235
 
                        return
236
152
        dir_to = br_from.bzrdir.clone(location_url,
237
153
            revision_id=br_from.last_revision())
238
154
        br_to = dir_to.open_branch()
243
159
            tree_to = dir_to.open_workingtree()
244
160
        except errors.NotLocalUrl:
245
161
            # FIXME - what to do here? how should we warn the user?
246
 
            #warning('This transport does not update the working '
247
 
            #        'tree of: %s' % (br_to.base,))
248
162
            count = br_to.pull(br_from, overwrite)
249
163
        except errors.NoWorkingTree:
250
164
            count = br_to.pull(br_from, overwrite)