1
# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
31
import bzrlib.errors as errors
34
""" Display Push dialog and perform the needed actions. """
35
def __init__(self, gladefile, comm):
36
""" Initialize the Push dialog. """
37
self.gladefile = gladefile
38
self.glade = gtk.glade.XML(self.gladefile, 'window_push')
40
# Communication object
43
self.window = self.glade.get_widget('window_push')
45
# Dictionary for signal_autoconnect
46
dic = { "on_button_push_push_clicked": self.push,
47
"on_button_push_cancel_clicked": self.close,
48
"on_button_push_test_clicked": self.test,
49
"on_radiobutton_push_stored_toggled": self.stored_toggled,
50
"on_radiobutton_push_specific_toggled": self.specific_toggled, }
52
# Connect the signals to the handlers
53
self.glade.signal_autoconnect(dic)
55
# Get some useful widgets
56
self.radio_stored = self.glade.get_widget('radiobutton_push_stored')
57
self.radio_specific = self.glade.get_widget('radiobutton_push_specific')
58
self.entry_stored = self.glade.get_widget('entry_push_stored')
59
self.entry_location = self.glade.get_widget('entry_push_location')
60
self.check_remember = self.glade.get_widget('checkbutton_push_remember')
61
self.check_overwrite = self.glade.get_widget('checkbutton_push_overwrite')
62
self.check_create = self.glade.get_widget('checkbutton_push_create')
63
self.label_test = self.glade.get_widget('label_push_test')
64
self.image_test = self.glade.get_widget('image_push_test')
67
self.entry_location.set_sensitive(0)
68
self.check_remember.set_sensitive(0)
69
self.check_create.set_sensitive(0)
72
self.notbranch = False
74
from bzrlib.branch import Branch
76
branch = Branch.open_containing(self.comm.get_path())[0]
78
self.entry_stored.set_text(branch.get_push_location())
79
except errors.NotBranchError:
84
""" Display the Push dialog. """
86
error_dialog(_('Directory is not a branch'),
87
_('You can perform this action only in a branch.'))
91
self.width, self.height = self.window.get_size()
93
def stored_toggled(self, widget):
94
if widget.get_active():
95
self.entry_stored.set_sensitive(1)
96
self.entry_location.set_sensitive(0)
97
self.check_remember.set_sensitive(0)
98
self.check_create.set_sensitive(0)
100
self.entry_stored.set_sensitive(0)
101
self.entry_location.set_sensitive(1)
102
self.check_remember.set_sensitive(1)
103
self.check_create.set_sensitive(1)
105
def specific_toggled(self, widget):
106
if widget.get_active():
107
self.entry_stored.set_sensitive(0)
108
self.entry_location.set_sensitive(1)
109
self.check_remember.set_sensitive(1)
110
self.check_create.set_sensitive(1)
112
self.entry_stored.set_sensitive(1)
113
self.entry_location.set_sensitive(0)
114
self.check_remember.set_sensitive(0)
115
self.check_create.set_sensitive(0)
117
def push(self, widget):
119
self.comm.set_busy(self.window)
120
if self.radio_stored.get_active():
122
revs = do_push(self.comm.get_path(),
123
overwrite=self.check_overwrite.get_active())
124
except errors.NotBranchError:
125
error_dialog(_('Directory is not a branch'),
126
_('You can perform this action only in a branch.'))
128
except errors.DivergedBranches:
129
error_dialog(_('Branches have been diverged'),
130
_('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
134
elif self.radio_specific.get_active():
135
location = self.entry_location.get_text()
137
error_dialog(_('No location specified'),
138
_('Please specify a location or use the default.'))
142
revs = do_push(self.comm.get_path(), location,
143
self.check_remember.get_active(),
144
self.check_overwrite.get_active(),
145
self.check_create.get_active())
146
except errors.NotBranchError:
147
error_dialog(_('Directory is not a branch'),
148
_('You can perform this action only in a branch.'))
149
self.comm.set_busy(self.window, False)
151
except errors.DivergedBranches:
152
error_dialog(_('Branches have been diverged'),
153
_('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
154
self.comm.set_busy(self.window, False)
159
# This should really never happen
163
info_dialog(_('Push successful'),
164
_('%d revision(s) pushed.') % revs)
166
def test(self, widget):
167
""" Test if write access possible. """
169
_urlRE = re.compile(r'^(?P<proto>[^:/\\]+)://(?P<path>.*)$')
171
if self.radio_stored.get_active():
172
url = self.entry_stored.get_text()
173
elif self.radio_specific.get_active():
174
url = self.entry_location.get_text()
176
m = _urlRE.match(url)
178
proto = m.groupdict()['proto']
179
if (proto == 'sftp') or (proto == 'file') or (proto == 'ftp'):
180
# have write acces (most probably)
181
self.image_test.set_from_stock(gtk.STOCK_YES, 4)
182
self.label_test.set_markup(_('<b>Write access is probably available</b>'))
185
self.image_test.set_from_stock(gtk.STOCK_NO, 4)
186
self.label_test.set_markup(_('<b>No write access</b>'))
189
self.image_test.set_from_stock(gtk.STOCK_DIALOG_QUESTION, 4)
190
self.label_test.set_markup(_('<b>Could not determine</b>'))
192
def close(self, widget=None):
193
self.window.destroy()
195
def do_push(branch, location=None, remember=False, overwrite=False,
196
create_prefix=False):
197
""" Update a mirror of a branch.
199
:param branch: the source branch
201
:param location: the location of the branch that you'd like to update
203
:param remember: if set, the location will be stored
205
:param overwrite: overwrite target location if it diverged
207
:param create_prefix: create the path leading up to the branch if it doesn't exist
209
:return: number of revisions pushed
211
from bzrlib.branch import Branch
212
from bzrlib.bzrdir import BzrDir
213
from bzrlib.transport import get_transport
215
br_from = Branch.open_containing(branch)[0]
217
stored_loc = br_from.get_push_location()
219
if stored_loc is None:
220
error_dialog(_('Push location is unknown'),
221
_('Please specify a location manually.'))
224
location = stored_loc
226
transport = get_transport(location)
227
location_url = transport.base
229
if br_from.get_push_location() is None or remember:
230
br_from.set_push_location(location_url)
235
dir_to = BzrDir.open(location_url)
236
br_to = dir_to.open_branch()
237
except errors.NotBranchError:
239
transport = transport.clone('..')
240
if not create_prefix:
242
relurl = transport.relpath(location_url)
243
transport.mkdir(relurl)
244
except errors.NoSuchFile:
245
error_dialog(_('Non existing parent directory'),
246
_("The parent directory (%s)\ndoesn't exist.") % location)
249
current = transport.base
250
needed = [(transport, transport.relpath(location_url))]
253
transport, relpath = needed[-1]
254
transport.mkdir(relpath)
256
except errors.NoSuchFile:
257
new_transport = transport.clone('..')
258
needed.append((new_transport,
259
new_transport.relpath(transport.base)))
260
if new_transport.base == transport.base:
261
error_dialog(_('Path prefix not created'),
262
_("The path leading up to the specified location couldn't\nbe created."))
264
dir_to = br_from.bzrdir.clone(location_url,
265
revision_id=br_from.last_revision())
266
br_to = dir_to.open_branch()
267
count = len(br_to.revision_history())
269
old_rh = br_to.revision_history()
272
tree_to = dir_to.open_workingtree()
273
except errors.NotLocalUrl:
274
# FIXME - what to do here? how should we warn the user?
275
#warning('This transport does not update the working '
276
# 'tree of: %s' % (br_to.base,))
277
count = br_to.pull(br_from, overwrite)
278
except errors.NoWorkingTree:
279
count = br_to.pull(br_from, overwrite)
281
count = tree_to.pull(br_from, overwrite)
282
except errors.DivergedBranches: