Kiln » TortoiseHg » TortoiseHg
Clone URL:  
/ hggtk
update.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
# # update.py - TortoiseHg's dialog for updating repo # # Copyright (C) 2007 TK Soh <teekaysoh@gmail.com> # import pygtk import os import sys import gtk from dialog import * from mercurial import util, hg, ui from mercurial.node import * from mercurial.i18n import _ from shlib import shell_notify, set_tortoise_icon from hglib import rootpath, toutf, RepoError class UpdateDialog(gtk.Window): """ Dialog to update Mercurial repo """ def __init__(self, rev=None): """ Initialize the Dialog """ gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL) set_tortoise_icon(self, 'menucheckout.ico') self.root = rootpath() self.rev = rev self.notify_func = None u = ui.ui() try: self.repo = hg.repository(u, path=self.root) except RepoError: return None # set dialog title title = "hg update - %s" % toutf(self.root) self.set_title(title) self._create() def set_notify_func(self, func, *args): self.notify_func = func self.notify_args = args def _create(self): self.set_default_size(350, 120) # add toolbar with tooltips self.tbar = gtk.Toolbar() self.tips = gtk.Tooltips() self._btn_update = self._toolbutton( gtk.STOCK_REVERT_TO_SAVED, _('Update'), self._btn_update_clicked, tip=_('Update working directory to selected revision')) tbuttons = [ self._btn_update, ] for btn in tbuttons: self.tbar.insert(btn, -1) vbox = gtk.VBox() self.add(vbox) vbox.pack_start(self.tbar, False, False, 2) # repo parent revisions parentbox = gtk.HBox() lbl = gtk.Label(_('Parent revisions:')) lbl.set_property('width-chars', 18) lbl.set_alignment(0, 0.5) self._parent_revs = gtk.Entry() self._parent_revs.set_sensitive(False) parentbox.pack_start(lbl, False, False) parentbox.pack_start(self._parent_revs, False, False) vbox.pack_start(parentbox, False, False, 2) # revision input revbox = gtk.HBox() lbl = gtk.Label(_('Update to revision:')) lbl.set_property('width-chars', 18) lbl.set_alignment(0, 0.5) # revisions combo box self._revlist = gtk.ListStore(str, str) self._revbox = gtk.ComboBoxEntry(self._revlist, 0) # add extra column to droplist for type of changeset cell = gtk.CellRendererText() self._revbox.pack_start(cell) self._revbox.add_attribute(cell, 'text', 1) self._rev_input = self._revbox.get_child() # setup buttons self._btn_rev_browse = gtk.Button(_('Browse...')) self._btn_rev_browse.connect('clicked', self._btn_rev_clicked) revbox.pack_start(lbl, False, False) revbox.pack_start(self._revbox, False, False) revbox.pack_start(self._btn_rev_browse, False, False, 5) vbox.pack_start(revbox, False, False, 2) self._overwrite = gtk.CheckButton(_('Overwrite local changes')) vbox.pack_end(self._overwrite, False, False, 10) # show them all self._refresh() def _toolbutton(self, stock, label, handler, menu=None, userdata=None, tip=None): if menu: tbutton = gtk.MenuToolButton(stock) tbutton.set_menu(menu) else: tbutton = gtk.ToolButton(stock) tbutton.set_label(label) if tip: tbutton.set_tooltip(self.tips, tip) tbutton.connect('clicked', handler, userdata) return tbutton def _refresh(self): """ update display on dialog with recent repo data """ try: # FIXME: force hg to refresh parents info del self.repo self.repo = hg.repository(ui.ui(), path=self.root) except RepoError: return None # populate parent rev data self._parents = [x.node() for x in self.repo.changectx(None).parents()] self._parent_revs.set_text(", ".join([short(x) for x in self._parents])) # populate revision data heads = self.repo.heads() tip = self.repo.changelog.node(nullrev+len(self.repo.changelog)) self._revlist.clear() for i, node in enumerate(heads): status = "head %d" % (i+1) if node == tip: status += ", tip" self._revlist.append([short(node), "(%s)" %status]) if self.rev is not None: self._revbox.get_child().set_text(str(self.rev)) else: self._revbox.set_active(0) def _btn_rev_clicked(self, button): """ select revision from history dialog """ import histselect rev = histselect.select(self.root) if rev is not None: self._rev_input.set_text(rev) def _btn_update_clicked(self, button, data=None): self._do_update() def _do_update(self): rev = self._rev_input.get_text() overwrite = self._overwrite.get_active() if not rev: error_dialog(self, _('Cannot update'), _('please enter revision to update to')) return response = question_dialog(self, _('Really want to update?'), _('to revision %s') % rev) if response != gtk.RESPONSE_YES: return cmdline = ['hg', 'update', '-R', self.root, '--rev', rev, '--verbose'] if overwrite: cmdline.append('--clean') from hgcmd import CmdDialog dlg = CmdDialog(cmdline) dlg.run() dlg.hide() if self.notify_func: self.notify_func(self.notify_args) self._refresh() shell_notify([self.root]) def run(ui, *pats, **opts): return UpdateDialog()