class CheckoutDialog(gtk.Dialog):
""" New implementation of the Checkout dialog. """
def __init__(self, path=None, parent=None, remote_path=None):
""" Initialize the Checkout dialog. """
gtk.Dialog.__init__(self, title="Checkout - Olive",
parent=parent,
flags=0,
buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
# Get arguments
self.path = path
# Create the widgets
self._button_checkout = gtk.Button(_i18n("Check_out"), use_underline=True)
self._button_revision = gtk.Button('')
self._image_browse = gtk.Image()
self._filechooser = gtk.FileChooserButton(_i18n("Please select a folder"))
self._combo = gtk.ComboBoxEntry()
self._label_location = gtk.Label(_i18n("Branch location:"))
self._label_destination = gtk.Label(_i18n("Destination:"))
self._label_nick = gtk.Label(_i18n("Branch nick:"))
self._label_revision = gtk.Label(_i18n("Revision:"))
self._hbox_revision = gtk.HBox()
self._entry_revision = gtk.Entry()
self._entry_nick = gtk.Entry()
self._check_lightweight = gtk.CheckButton(_i18n("_Lightweight checkout"),
use_underline=True)
# Set callbacks
self._button_checkout.connect('clicked', self._on_checkout_clicked)
self._button_revision.connect('clicked', self._on_revision_clicked)
self._combo.child.connect('focus-out-event', self._on_combo_changed)
# Create the table and pack the widgets into it
self._table = gtk.Table(rows=3, columns=2)
self._table.attach(self._label_location, 0, 1, 0, 1)
self._table.attach(self._label_destination, 0, 1, 1, 2)
self._table.attach(self._label_nick, 0, 1, 2, 3)
self._table.attach(self._label_revision, 0, 1, 3, 4)
self._table.attach(self._combo, 1, 2, 0, 1)
self._table.attach(self._filechooser, 1, 2, 1, 2)
self._table.attach(self._entry_nick, 1, 2, 2, 3)
self._table.attach(self._hbox_revision, 1, 2, 3, 4)
self._table.attach(self._check_lightweight, 1, 2, 4, 5)
# Set properties
self._image_browse.set_from_stock(gtk.STOCK_OPEN, gtk.ICON_SIZE_BUTTON)
self._button_revision.set_image(self._image_browse)
self._button_revision.set_sensitive(False)
self._filechooser.set_action(gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER)
self._label_location.set_alignment(0, 0.5)
self._label_destination.set_alignment(0, 0.5)
self._label_nick.set_alignment(0, 0.5)
self._label_revision.set_alignment(0, 0.5)
self._table.set_row_spacings(3)
self._table.set_col_spacings(3)
self.vbox.set_spacing(3)
if self.path is not None:
self._filechooser.set_filename(self.path)
if remote_path is not None:
self._combo.child.set_text(remote_path)
# Pack some widgets
self._hbox_revision.pack_start(self._entry_revision, True, True)
self._hbox_revision.pack_start(self._button_revision, False, False)
self.vbox.add(self._table)
self.action_area.pack_end(self._button_checkout)
# Show the dialog
self.vbox.show_all()
# Build checkout history
self._history = UrlHistory(GlobalConfig(), 'branch_history')
self._build_history()
def _build_history(self):
""" Build up the checkout history. """
self._combo_model = gtk.ListStore(str)
for item in self._history.get_entries():
self._combo_model.append([ item ])
self._combo.set_model(self._combo_model)
self._combo.set_text_column(0)
def _get_last_revno(self):
""" Get the revno of the last revision (if any). """
location = self._combo.get_child().get_text()
try:
br = Branch.open(location)
except:
return None
else:
return br.revno()
def _on_revision_clicked(self, button):
""" Browse for revision button clicked handler. """
from revbrowser import RevisionBrowser
location = self._combo.get_child().get_text()
try:
br = Branch.open(location)
except:
return
else:
revb = RevisionBrowser(br, self)
response = revb.run()
if response != gtk.RESPONSE_NONE:
revb.hide()
if response == gtk.RESPONSE_OK:
if revb.selected_revno is not None:
self._entry_revision.set_text(revb.selected_revno)
revb.destroy()
@show_bzr_error
def _on_checkout_clicked(self, button):
""" Checkout button clicked handler. """
location = self._combo.get_child().get_text()
if location is '':
error_dialog(_i18n('Missing branch location'),
_i18n('You must specify a branch location.'))
return
destination = self._filechooser.get_filename()
try:
revno = int(self._entry_revision.get_text())
except:
revno = None
nick = self._entry_nick.get_text()
if nick is '':
nick = os.path.basename(location.rstrip("/\\"))
br_from = Branch.open(location)
revision_id = br_from.get_rev_id(revno)
lightweight = self._check_lightweight.get_active()
to_location = destination + os.sep + nick
os.mkdir(to_location)
br_from.create_checkout(to_location, revision_id, lightweight)
self._history.add_entry(location)
self.response(gtk.RESPONSE_OK)
def _on_combo_changed(self, widget, event):
""" We try to get the last revision if focus lost. """
rev = self._get_last_revno()
if rev is None:
self._entry_revision.set_text(_i18n('N/A'))
self._button_revision.set_sensitive(False)
else:
self._entry_revision.set_text(str(rev))
self._button_revision.set_sensitive(True)
if self._entry_nick.get_text() == '':
self._entry_nick.set_text(os.path.basename(self._combo.get_child().get_text().rstrip("/\\")))
class PushDialog(gtk.Dialog):
""" New implementation of the Push dialog. """
def __init__(self, repository, revid, branch=None, parent=None):
""" Initialize the Push dialog. """
gtk.Dialog.__init__(self, title="Push - Olive",
parent=parent,
flags=0,
buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
# Get arguments
self.repository = repository
self.revid = revid
self.branch = branch
# Create the widgets
self._label_location = gtk.Label(_i18n("Location:"))
self._combo = gtk.ComboBoxEntry()
self._button_push = gtk.Button(_i18n("_Push"), use_underline=True)
self._hbox_location = gtk.HBox()
# Set callbacks
self._button_push.connect('clicked', self._on_push_clicked)
# Set properties
self._label_location.set_alignment(0, 0.5)
self._hbox_location.set_spacing(3)
self.vbox.set_spacing(3)
# Pack widgets
self._hbox_location.pack_start(self._label_location, False, False)
self._hbox_location.pack_start(self._combo, True, True)
self.vbox.pack_start(self._hbox_location)
self.action_area.pack_end(self._button_push)
# Show the dialog
self.vbox.show_all()
# Build location history
self._history = UrlHistory(self.branch.get_config(), 'push_history')
self._build_history()
def _build_history(self):
""" Build up the location history. """
self._combo_model = gtk.ListStore(str)
for item in self._history.get_entries():
self._combo_model.append([ item ])
self._combo.set_model(self._combo_model)
self._combo.set_text_column(0)
if self.branch is not None:
location = self.branch.get_push_location()
if location is not None:
self._combo.get_child().set_text(location)
@show_bzr_error
def _on_push_clicked(self, widget):
""" Push button clicked handler. """
location = self._combo.get_child().get_text()
revs = 0
try:
revs = do_push(self.branch, location=location, overwrite=False)
except errors.DivergedBranches:
response = question_dialog(_i18n('Branches have been diverged'),
_i18n('You cannot push if branches have diverged.\nOverwrite?'))
if response == gtk.RESPONSE_YES:
revs = do_push(self.branch, location=location, overwrite=True)
if self.branch is not None and self.branch.get_push_location() is None:
self.branch.set_push_location(location)
self._history.add_entry(location)
info_dialog(_i18n('Push successful'),
_i18n("%d revision(s) pushed.") % revs)
self.response(gtk.RESPONSE_OK)