14
16
# along with this program; if not, write to the Free Software
15
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
from gi.repository import Gtk
25
21
from errors import show_bzr_error
27
from bzrlib.config import LocationConfig
28
import bzrlib.errors as errors
30
from dialog import error_dialog, info_dialog, question_dialog
32
from history import UrlHistory
34
class PushDialog(gtk.Dialog):
35
""" New implementation of the Push dialog. """
36
def __init__(self, branch, parent=None):
37
""" Initialize the Push dialog. """
38
gtk.Dialog.__init__(self, title="Push - Olive",
41
buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
27
from bzrlib.controldir import ControlDir
28
from bzrlib.push import PushResult
29
from bzrlib.transport import get_transport
31
from bzrlib.plugins.gtk.dialog import (
34
from bzrlib.plugins.gtk.history import UrlHistory
35
from bzrlib.plugins.gtk.i18n import _i18n
36
from bzrlib.plugins.gtk.ui import ProgressPanel
39
class PushDialog(Gtk.Dialog):
40
"""New implementation of the Push dialog."""
42
def __init__(self, repository=None, revid=None, branch=None, parent=None):
43
"""Initialize the Push dialog. """
44
super(PushDialog, self).__init__(
45
title="Push", parent=parent, flags=0, border_width=6,
46
buttons=(Gtk.STOCK_CLOSE, Gtk.ResponseType.CLOSE))
47
if repository is None:
48
repository = branch.repository
44
49
self.branch = branch
52
self.repository = repository
54
revid = branch.last_revision()
46
57
# Create the widgets
47
self._label_location = gtk.Label(_("Location:"))
48
self._label_test = gtk.Label(_("(click the Test button to check write access)"))
49
self._check_remember = gtk.CheckButton(_("_Remember as default location"),
51
self._check_prefix = gtk.CheckButton(_("Create the path _leading up to the location"),
53
self._check_overwrite = gtk.CheckButton(_("_Overwrite target if diverged"),
55
self._combo = gtk.ComboBoxEntry()
56
self._button_test = gtk.Button(_("_Test"), use_underline=True)
57
self._button_push = gtk.Button(_("_Push"), use_underline=True)
58
self._hbox_location = gtk.HBox()
59
self._hbox_test = gtk.HBox()
60
self._image_test = gtk.Image()
58
self._label_location = Gtk.Label(label=_i18n("Location:"))
59
self._combo = Gtk.ComboBox.new_with_entry()
60
self._button_push = Gtk.Button(_i18n("_Push"), use_underline=True)
61
self._hbox_location = Gtk.Box(Gtk.Orientation.HORIZONTAL, 6)
62
self._push_message = Gtk.Label(xalign=0)
63
self._progress_widget = ProgressPanel()
63
self._button_test.connect('clicked', self._on_test_clicked)
66
ui.ui_factory.set_progress_bar_widget(self._progress_widget)
67
self.connect('close', self._on_close_clicked)
64
68
self._button_push.connect('clicked', self._on_push_clicked)
67
self._image_test.set_from_stock(gtk.STOCK_DIALOG_INFO, gtk.ICON_SIZE_BUTTON)
68
self._label_location.set_alignment(0, 0.5)
69
self._label_test.set_alignment(0, 0.5)
70
self._hbox_location.set_spacing(3)
71
self._hbox_test.set_spacing(3)
72
self.vbox.set_spacing(3)
71
content_area = self.get_content_area()
72
content_area.set_spacing(6)
75
self._hbox_location.pack_start(self._label_location, False, False)
76
self._hbox_location.pack_start(self._combo, True, True)
77
self._hbox_test.pack_start(self._image_test, False, False)
78
self._hbox_test.pack_start(self._label_test, True, True)
79
self.vbox.pack_start(self._hbox_location)
80
self.vbox.pack_start(self._check_remember)
81
self.vbox.pack_start(self._check_prefix)
82
self.vbox.pack_start(self._check_overwrite)
83
self.vbox.pack_start(self._hbox_test)
84
self.action_area.pack_start(self._button_test)
85
self.action_area.pack_end(self._button_push)
75
self._hbox_location.pack_start(self._label_location, False, False, 0)
76
self._hbox_location.pack_start(self._combo, False, False, 0)
77
content_area.pack_start(self._hbox_location, True, True, 0)
78
content_area.pack_start(self._progress_widget, True, True, 0)
79
content_area.pack_start(self._push_message, True, True, 0)
80
self.get_action_area().pack_end(self._button_push, True, True, 0)
83
content_area.show_all()
84
self._progress_widget.hide()
85
self._push_message.hide()
90
87
# Build location history
91
88
self._history = UrlHistory(self.branch.get_config(), 'push_history')
92
89
self._build_history()
94
91
def _build_history(self):
95
""" Build up the location history. """
96
self._combo_model = gtk.ListStore(str)
92
"""Build up the location history. """
93
self._combo_model = Gtk.ListStore(str)
97
94
for item in self._history.get_entries():
98
self._combo_model.append([ item ])
95
self._combo_model.append([item])
99
96
self._combo.set_model(self._combo_model)
100
self._combo.set_text_column(0)
102
location = self.branch.get_push_location()
104
self._combo.get_child().set_text(location)
106
def _on_test_clicked(self, widget):
107
""" Test button clicked handler. """
109
_urlRE = re.compile(r'^(?P<proto>[^:/\\]+)://(?P<path>.*)$')
111
url = self._combo.get_child().get_text()
113
m = _urlRE.match(url)
115
proto = m.groupdict()['proto']
116
# FIXME: This should ask the transport or branch rather than
117
# guessing using regular expressions. JRV 20070714
118
if proto in ('sftp', 'file', 'ftp'):
119
# have write access (most probably)
120
self._image_test.set_from_stock(gtk.STOCK_YES, 4)
121
self._label_test.set_markup(_('<b>Write access is probably available</b>'))
124
self._image_test.set_from_stock(gtk.STOCK_NO, 4)
125
self._label_test.set_markup(_('<b>No write access</b>'))
128
self._image_test.set_from_stock(gtk.STOCK_DIALOG_QUESTION, 4)
129
self._label_test.set_markup(_('<b>Could not determine</b>'))
97
self._combo.set_entry_text_column(0)
99
if self.branch is not None:
100
location = self.branch.get_push_location()
101
if location is not None:
102
self._combo.get_child().set_text(location)
104
def _on_close_clicked(self, widget):
105
"""Close dialog handler."""
106
ui.ui_factory.set_progress_bar_widget(None)
132
109
def _on_push_clicked(self, widget):
133
""" Push button clicked handler. """
110
"""Push button clicked handler. """
111
self._push_message.hide()
112
self._progress_widget.tick()
134
113
location = self._combo.get_child().get_text()
137
revs = do_push(self.branch,
139
overwrite=self._check_overwrite.get_active(),
140
remember=self._check_remember.get_active(),
141
create_prefix=self._check_prefix.get_active())
116
message = do_push(self.branch, location, overwrite=False)
142
117
except errors.DivergedBranches:
143
response = question_dialog(_('Branches have been diverged'),
144
_('You cannot push if branches have diverged.\nOverwrite?'))
145
if response == gtk.RESPONSE_OK:
146
revs = do_push(self.branch, overwrite=True)
118
response = question_dialog(
119
_i18n('Branches have been diverged'),
120
_i18n('You cannot push if branches have diverged.\n'
122
if response == Gtk.ResponseType.YES:
123
message = do_push(self.branch, location, overwrite=True)
149
126
self._history.add_entry(location)
150
info_dialog(_('Push successful'),
151
_("%d revision(s) pushed.") % revs)
153
self.response(gtk.RESPONSE_OK)
155
def do_push(branch, location=None, remember=False, overwrite=False,
156
create_prefix=False):
157
""" Update a mirror of a branch.
159
:param branch: the source branch
127
if (self.branch is not None
128
and self.branch.get_push_location() is None):
129
self.branch.set_push_location(location)
131
self._progress_widget.finished()
132
self._push_message.props.label = message
133
self._push_message.show()
136
def do_push(br_from, location, overwrite):
137
"""Update a mirror of a branch.
139
:param br_from: the source branch
161
140
:param location: the location of the branch that you'd like to update
163
:param remember: if set, the location will be stored
165
141
:param overwrite: overwrite target location if it diverged
167
:param create_prefix: create the path leading up to the branch if it doesn't exist
169
142
:return: number of revisions pushed
171
from bzrlib.bzrdir import BzrDir
172
from bzrlib.transport import get_transport
176
stored_loc = br_from.get_push_location()
178
if stored_loc is None:
179
error_dialog(_('Push location is unknown'),
180
_('Please specify a location manually.'))
183
location = stored_loc
185
transport = get_transport(location)
186
location_url = transport.base
188
if br_from.get_push_location() is None or remember:
189
br_from.set_push_location(location_url)
145
to_transport = get_transport(location)
194
dir_to = BzrDir.open(location_url)
195
br_to = dir_to.open_branch()
147
dir_to = ControlDir.open_from_transport(to_transport)
196
148
except errors.NotBranchError:
198
transport = transport.clone('..')
199
if not create_prefix:
201
relurl = transport.relpath(location_url)
202
transport.mkdir(relurl)
203
except errors.NoSuchFile:
204
error_dialog(_('Non existing parent directory'),
205
_("The parent directory (%s)\ndoesn't exist.") % location)
208
current = transport.base
209
needed = [(transport, transport.relpath(location_url))]
212
transport, relpath = needed[-1]
213
transport.mkdir(relpath)
215
except errors.NoSuchFile:
216
new_transport = transport.clone('..')
217
needed.append((new_transport,
218
new_transport.relpath(transport.base)))
219
if new_transport.base == transport.base:
220
error_dialog(_('Path prefix not created'),
221
_("The path leading up to the specified location couldn't\nbe created."))
223
dir_to = br_from.bzrdir.clone(location_url,
224
revision_id=br_from.last_revision())
225
br_to = dir_to.open_branch()
226
count = len(br_to.revision_history())
153
br_to = br_from.create_clone_on_transport(
154
to_transport, revision_id=revision_id)
155
except errors.NoSuchFile:
156
response = question_dialog(
157
_i18n('Non existing parent directory'),
158
_i18n("The parent directory (%s)\ndoesn't exist. Create?") %
160
if response == Gtk.ResponseType.OK:
161
br_to = br_from.create_clone_on_transport(
162
to_transport, revision_id=revision_id, create_prefix=True)
164
return _i18n("Push aborted.")
165
push_result = create_push_result(br_from, br_to)
228
old_rh = br_to.revision_history()
230
tree_to = dir_to.open_workingtree()
231
except errors.NotLocalUrl:
232
# FIXME - what to do here? how should we warn the user?
233
#warning('This transport does not update the working '
234
# 'tree of: %s' % (br_to.base,))
235
count = br_to.pull(br_from, overwrite)
236
except errors.NoWorkingTree:
237
count = br_to.pull(br_from, overwrite)
239
count = tree_to.pull(br_from, overwrite)
167
push_result = dir_to.push_branch(br_from, revision_id, overwrite)
168
message = create_push_message(br_from, push_result)
172
def create_push_message(br_from, push_result):
173
"""Return a mesage explaining what happened during the push."""
175
rev_count = br_from.revno() - push_result.old_revno
176
messages.append(_i18n("%d revision(s) pushed.") % rev_count)
177
if push_result.stacked_on is not None:
178
messages.append(_i18n("Stacked on %s.") % push_result.stacked_on)
179
if push_result.workingtree_updated is False:
180
messages.append(_i18n(
181
"\nThe working tree was not updated:"
182
"\nSee 'bzr help working-trees' for more information."))
183
return '\n'.join(messages)
186
def create_push_result(br_from, br_to):
187
"""Return a PushResult like one created by ControlDir.push_branch()."""
188
push_result = PushResult()
189
push_result.source_branch = br_from
190
push_result.target_branch = br_to
191
push_result.branch_push_result = None
192
push_result.master_branch = None
193
push_result.old_revno = 0
194
push_result.old_revid = br_to.last_revision()
195
push_result.workingtree_updated = None # Not applicable to this case.
197
push_result.stacked_on = br_to.get_stacked_on_url()
198
except (errors.UnstackableBranchFormat,
199
errors.UnstackableRepositoryFormat,
201
push_result.stacked_on = None