« back to all changes in this revision
Viewing changes to shelf.py
- browse files at revision 0.1.21
- compare with another revision
- compare with revision 676.1.1
- download diff
- download tarball
- view history from revision 0.1.21
- Committer: Michael Ellerman
- Date: 2005-10-19 10:17:43 UTC
- mto: (0.3.1 shelf-dev) (325.1.2 bzrtools)
- mto: This revision was merged to the branch mainline in revision 246.
- Revision ID: michael@ellerman.id.au-20051019101743-8a1e9d9464db1567
Remove HunkSelector from shelf.py
- files modified:
- shelf.py
added
removed
shelf.py
98
98
if pipe.close() is not None:
99
99
raise Exception("Failed running bzr")
100
100
return lines
101
102
103
class HunkSelector:
104
class Option:
105
def __init__(self, char, action, help, default=False):
106
self.char = char
107
self.action = action
108
self.default = default
109
self.help = help
110
111
standard_options = [
112
Option('n', 'shelve', 'shelve this change for the moment.',
113
default=True),
114
Option('y', 'keep', 'keep this change in your tree.'),
115
Option('d', 'done', 'done, skip to the end.'),
116
Option('i', 'invert', 'invert the current selection of all hunks.'),
117
Option('s', 'status', 'show status of hunks.'),
118
Option('q', 'quit', 'quit')
119
]
120
121
end_options = [
122
Option('y', 'continue', 'proceed to shelve selected changes.',
123
default=True),
124
Option('r', 'restart', 'restart the hunk selection loop.'),
125
Option('s', 'status', 'show status of hunks.'),
126
Option('i', 'invert', 'invert the current selection of all hunks.'),
127
Option('q', 'quit', 'quit')
128
]
129
130
def __init__(self, patches):
131
self.patches = patches
132
self.total_hunks = 0
133
for patch in patches:
134
for hunk in patch.hunks:
135
# everything's shelved by default
136
hunk.selected = True
137
self.total_hunks += 1
138
139
def __get_option(self, char):
140
for opt in self.standard_options:
141
if opt.char == char:
142
return opt
143
raise Exception('Option "%s" not found!' % char)
144
145
def __select_loop(self):
146
j = 0
147
for patch in self.patches:
148
i = 0
149
lasti = -1
150
while i < len(patch.hunks):
151
hunk = patch.hunks[i]
152
if lasti != i:
153
print patch.get_header(), hunk
154
j += 1
155
lasti = i
156
157
prompt = 'Keep this change? (%d of %d)' \
158
% (j, self.total_hunks)
159
160
if hunk.selected:
161
self.__get_option('n').default = True
162
self.__get_option('y').default = False
163
else:
164
self.__get_option('n').default = False
165
self.__get_option('y').default = True
166
167
action = self.__ask_user(prompt, self.standard_options)
168
169
if action == 'keep':
170
hunk.selected = False
171
elif action == 'shelve':
172
hunk.selected = True
173
elif action == 'done':
174
return True
175
elif action == 'invert':
176
self.__invert_selection()
177
self.__show_status()
178
continue
179
elif action == 'status':
180
self.__show_status()
181
continue
182
elif action == 'quit':
183
return False
184
185
i += 1
186
return True
187
188
def select(self):
189
if self.total_hunks == 0:
190
return []
191
192
done = False
193
while not done:
194
if not self.__select_loop():
195
return []
196
197
while True:
198
self.__show_status()
199
prompt = "Shelve these changes, or restart?"
200
action = self.__ask_user(prompt, self.end_options)
201
202
if action == 'continue':
203
done = True
204
break
205
elif action == 'quit':
206
return []
207
elif action == 'status':
208
self.__show_status()
209
elif action == 'invert':
210
self.__invert_selection()
211
elif action == 'restart':
212
break
213
214
215
for patch in self.patches:
216
tmp = []
217
for hunk in patch.hunks:
218
if hunk.selected:
219
tmp.append(hunk)
220
patch.hunks = tmp
221
222
tmp = []
223
for patch in self.patches:
224
if len(patch.hunks):
225
tmp.append(patch)
226
self.patches = tmp
227
228
return self.patches
229
230
def __invert_selection(self):
231
for patch in self.patches:
232
for hunk in patch.hunks:
233
if hunk.__dict__.has_key('selected'):
234
hunk.selected = not hunk.selected
235
else:
236
hunk.selected = True
237
238
def __show_status(self):
239
print '\nStatus:'
240
for patch in self.patches:
241
print ' %s' % patch.oldname
242
shelve = 0
243
keep = 0
244
for hunk in patch.hunks:
245
if hunk.selected:
246
shelve += 1
247
else:
248
keep += 1
249
250
print ' %d hunks to be shelved' % shelve
251
print ' %d hunks to be kept' % keep
252
print
253
254
def __getchar(self):
255
fd = sys.stdin.fileno()
256
settings = termios.tcgetattr(fd)
257
try:
258
tty.setraw(fd)
259
ch = sys.stdin.read(1)
260
finally:
261
termios.tcsetattr(fd, termios.TCSADRAIN, settings)
262
return ch
263
264
def __ask_user(self, prompt, options):
265
while True:
266
sys.stdout.write(prompt)
267
sys.stdout.write(' [')
268
for opt in options:
269
if opt.default:
270
default = opt
271
sys.stdout.write(opt.char)
272
sys.stdout.write('?] (%s): ' % default.char)
273
274
response = self.__getchar()
275
276
# default, which we see as newline, is 'n'
277
if response in ['\n', '\r', '\r\n']:
278
response = default.char
279
280
print response # because echo is off
281
282
for opt in options:
283
if opt.char == response:
284
return opt.action
285
286
for opt in options:
287
print ' %s - %s' % (opt.char, opt.help)
Loggerhead is a web-based interface for Breezy
Version: 2.0.1