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, dialog):
36
""" Initialize the Push dialog. """
37
self.gladefile = gladefile
38
self.glade = gtk.glade.XML(self.gladefile, 'window_push')
40
# Communication object
45
self.window = self.glade.get_widget('window_push')
47
# Dictionary for signal_autoconnect
48
dic = { "on_button_push_push_clicked": self.push,
49
"on_button_push_cancel_clicked": self.close,
50
"on_button_push_test_clicked": self.test,
51
"on_radiobutton_push_stored_toggled": self.stored_toggled,
52
"on_radiobutton_push_specific_toggled": self.specific_toggled, }
54
# Connect the signals to the handlers
55
self.glade.signal_autoconnect(dic)
57
# Get some useful widgets
58
self.radio_stored = self.glade.get_widget('radiobutton_push_stored')
59
self.radio_specific = self.glade.get_widget('radiobutton_push_specific')
60
self.entry_stored = self.glade.get_widget('entry_push_stored')
61
self.entry_location = self.glade.get_widget('entry_push_location')
62
self.check_remember = self.glade.get_widget('checkbutton_push_remember')
63
self.check_overwrite = self.glade.get_widget('checkbutton_push_overwrite')
64
self.check_create = self.glade.get_widget('checkbutton_push_create')
65
self.label_test = self.glade.get_widget('label_push_test')
66
self.image_test = self.glade.get_widget('image_push_test')
69
self.entry_location.set_sensitive(0)
70
self.check_remember.set_sensitive(0)
71
self.check_create.set_sensitive(0)
74
self.notbranch = False
76
from bzrlib.branch import Branch
78
branch = Branch.open_containing(self.comm.get_path())[0]
80
self.entry_stored.set_text(branch.get_push_location())
81
except errors.NotBranchError:
86
""" Display the Push dialog. """
88
self.dialog.error_dialog(_('Directory is not a branch'),
89
_('You can perform this action only in a branch.'))
93
self.width, self.height = self.window.get_size()
95
def stored_toggled(self, widget):
96
if widget.get_active():
97
self.entry_stored.set_sensitive(1)
98
self.entry_location.set_sensitive(0)
99
self.check_remember.set_sensitive(0)
100
self.check_create.set_sensitive(0)
102
self.entry_stored.set_sensitive(0)
103
self.entry_location.set_sensitive(1)
104
self.check_remember.set_sensitive(1)
105
self.check_create.set_sensitive(1)
107
def specific_toggled(self, widget):
108
if widget.get_active():
109
self.entry_stored.set_sensitive(0)
110
self.entry_location.set_sensitive(1)
111
self.check_remember.set_sensitive(1)
112
self.check_create.set_sensitive(1)
114
self.entry_stored.set_sensitive(1)
115
self.entry_location.set_sensitive(0)
116
self.check_remember.set_sensitive(0)
117
self.check_create.set_sensitive(0)
119
def push(self, widget):
121
self.comm.set_busy(self.window)
122
if self.radio_stored.get_active():
124
revs = do_push(self.comm.get_path(),
125
overwrite=self.check_overwrite.get_active())
126
except errors.NotBranchError:
127
self.dialog.error_dialog(_('Directory is not a branch'),
128
_('You can perform this action only in a branch.'))
130
except errors.DivergedBranches:
131
self.dialog.error_dialog(_('Branches have been diverged'),
132
_('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
136
elif self.radio_specific.get_active():
137
location = self.entry_location.get_text()
139
self.dialog.error_dialog(_('No location specified'),
140
_('Please specify a location or use the default.'))
144
revs = do_push(self.comm.get_path(), location,
145
self.check_remember.get_active(),
146
self.check_overwrite.get_active(),
147
self.check_create.get_active())
148
except errors.NotBranchError:
149
self.dialog.error_dialog(_('Directory is not a branch'),
150
_('You can perform this action only in a branch.'))
151
self.comm.set_busy(self.window, False)
153
except errors.DivergedBranches:
154
self.dialog.error_dialog(_('Branches have been diverged'),
155
_('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
156
self.comm.set_busy(self.window, False)
161
# This should really never happen
165
self.dialog.info_dialog(_('Push successful'),
166
_('%d revision(s) pushed.') % revs)
168
def test(self, widget):
169
""" Test if write access possible. """
171
_urlRE = re.compile(r'^(?P<proto>[^:/\\]+)://(?P<path>.*)$')
173
if self.radio_stored.get_active():
174
url = self.entry_stored.get_text()
175
elif self.radio_specific.get_active():
176
url = self.entry_location.get_text()
178
m = _urlRE.match(url)
180
proto = m.groupdict()['proto']
181
if (proto == 'sftp') or (proto == 'file') or (proto == 'ftp'):
182
# have write acces (most probably)
183
self.image_test.set_from_stock(gtk.STOCK_YES, 4)
184
self.label_test.set_markup(_('<b>Write access is probably available</b>'))
187
self.image_test.set_from_stock(gtk.STOCK_NO, 4)
188
self.label_test.set_markup(_('<b>No write access</b>'))
191
self.image_test.set_from_stock(gtk.STOCK_DIALOG_QUESTION, 4)
192
self.label_test.set_markup(_('<b>Could not determine</b>'))
194
def close(self, widget=None):
195
self.window.destroy()
197
def do_push(branch, location=None, remember=False, overwrite=False,
198
create_prefix=False):
199
""" Update a mirror of a branch.
201
:param branch: the source branch
203
:param location: the location of the branch that you'd like to update
205
:param remember: if set, the location will be stored
207
:param overwrite: overwrite target location if it diverged
209
:param create_prefix: create the path leading up to the branch if it doesn't exist
211
:return: number of revisions pushed
213
from bzrlib.branch import Branch
214
from bzrlib.bzrdir import BzrDir
215
from bzrlib.transport import get_transport
217
br_from = Branch.open_containing(branch)[0]
219
stored_loc = br_from.get_push_location()
221
if stored_loc is None:
222
self.dialog.error_dialog(_('Push location is unknown'),
223
_('Please specify a location manually.'))
226
location = stored_loc
228
transport = get_transport(location)
229
location_url = transport.base
231
if br_from.get_push_location() is None or remember:
232
br_from.set_push_location(location_url)
237
dir_to = BzrDir.open(location_url)
238
br_to = dir_to.open_branch()
239
except errors.NotBranchError:
241
transport = transport.clone('..')
242
if not create_prefix:
244
relurl = transport.relpath(location_url)
245
transport.mkdir(relurl)
246
except errors.NoSuchFile:
247
self.dialog.error_dialog(_('Non existing parent directory'),
248
_("The parent directory (%s)\ndoesn't exist.") % location)
251
current = transport.base
252
needed = [(transport, transport.relpath(location_url))]
255
transport, relpath = needed[-1]
256
transport.mkdir(relpath)
258
except errors.NoSuchFile:
259
new_transport = transport.clone('..')
260
needed.append((new_transport,
261
new_transport.relpath(transport.base)))
262
if new_transport.base == transport.base:
263
self.dialog.error_dialog(_('Path prefix not created'),
264
_("The path leading up to the specified location couldn't\nbe created."))
266
dir_to = br_from.bzrdir.clone(location_url,
267
revision_id=br_from.last_revision())
268
br_to = dir_to.open_branch()
269
count = len(br_to.revision_history())
271
old_rh = br_to.revision_history()
274
tree_to = dir_to.open_workingtree()
275
except errors.NotLocalUrl:
276
# FIXME - what to do here? how should we warn the user?
277
#warning('This transport does not update the working '
278
# 'tree of: %s' % (br_to.base,))
279
count = br_to.pull(br_from, overwrite)
280
except errors.NoWorkingTree:
281
count = br_to.pull(br_from, overwrite)
283
count = tree_to.pull(br_from, overwrite)
284
except errors.DivergedBranches: