« back to all changes in this revision
Viewing changes to shelf.py
- browse files at revision 87
- compare with another revision
- download diff
- download tarball
- view history from revision 87
- Committer: Aaron Bentley
- Date: 2005-06-15 15:53:52 UTC
- Revision ID: abentley@panoramicfeedback.com-20050615155352-afb1ccd8c20839c7
updated docs
added
removed
shelf.py
1
1
#!/usr/bin/python
2
2
3
from patches import parse_patches
3
import patches
4
4
import os
5
5
import sys
6
6
import string
7
import tty, termios
8
import glob
9
7
from bzrlib.commands import Command
10
8
11
def main(args):
12
name = os.path.basename(args.pop(0))
13
14
if name not in ['shelve', 'unshelve']:
15
raise Exception("Unknown command name '%s'" % name)
16
17
if len(args) > 0:
18
if args[0] == '--bzr-usage':
19
print '\n'
20
return 0
21
elif args[0] == '--bzr-help':
22
print 'Shelve a patch, you can get it back later with unshelve.'
23
return 0
24
else:
25
raise Exception("Don't understand args %s" % args)
26
27
if eval(name + "()"):
28
return 0
29
30
return 1
31
32
def tree_root():
33
return run_bzr('root')[0].strip()
34
35
def shelf_suffix(index):
36
if index == 0:
37
return ""
9
def parse_args(args):
10
if len(args) == 2 and args[1] == '--bzr-usage':
11
print '\n'
12
return True
13
elif len(args) == 2 and args[1] == '--bzr-help':
14
print 'Shelve a patch, you can get it back later with unshelve.'
15
return True
16
elif len(args) == 1:
17
pass
38
18
else:
39
return "-%d" % index
40
41
def next_shelf():
42
def name_sequence():
43
i = 0
44
while True:
45
yield shelf_suffix(i)
46
i = i + 1
47
48
stem = os.path.join(tree_root(), '.bzr-shelf')
49
for end in name_sequence():
50
name = stem + end
51
if not os.path.exists(name):
52
return name
53
54
def last_shelf():
55
stem = os.path.join(tree_root(), '.bzr-shelf')
56
shelves = glob.glob(stem)
57
shelves.extend(glob.glob(stem + '-*'))
58
def shelf_index(name):
59
if name == stem:
60
return 0
61
return int(name[len(stem)+1:])
62
shelvenums = [shelf_index(f) for f in shelves]
63
shelvenums.sort()
64
65
if len(shelvenums) == 0:
66
return None
67
return stem + shelf_suffix(shelvenums[-1])
68
69
def get_shelf_message(shelf):
70
prefix = "# shelf: "
71
if not shelf.startswith(prefix):
72
return None
73
return shelf[len(prefix):shelf.index('\n')]
19
raise Exception("Don't understand args %s" % args)
20
21
return False
74
22
75
23
def unshelve():
76
shelf = last_shelf()
24
root = run_bzr('root')[0].strip()
25
shelf = os.path.join(root, '.bzr-shelf')
77
26
78
if shelf is None:
79
raise Exception("No shelf found in '%s'" % tree_root())
27
if not os.path.exists(shelf):
28
raise Exception("No shelf found in '%s'" % shelf)
80
29
81
30
patch = open(shelf, 'r').read()
82
31
83
print >>sys.stderr, "Reapplying shelved patches",
84
message = get_shelf_message(patch)
85
if message is not None:
86
print >>sys.stderr, ' "%s"' % message
87
else:
88
print >>sys.stderr, ""
89
pipe = os.popen('patch -d %s -s -p0' % tree_root(), 'w')
32
print >>sys.stderr, "Reapplying shelved patches"
33
pipe = os.popen('patch -d %s -s -p0' % root, 'w')
90
34
pipe.write(patch)
91
35
pipe.flush()
92
36
99
43
100
44
return True
101
45
102
class QuitException(Exception):
103
pass
104
105
def shelve(message = None, revision = None, file_list = None):
106
cmd = ['diff']
107
if revision is not None:
108
cmd.extend(['--revision', str(revision[0])])
109
if file_list is not None:
110
cmd.extend(file_list)
111
patches = parse_patches(run_bzr(cmd))
112
try:
113
patches = HunkSelector(patches).select()
114
except QuitException:
115
return False
116
117
if len(patches) == 0:
46
def shelve():
47
diff_lines = run_bzr('diff')
48
patch_list = patches.parse_patches(diff_lines.__iter__())
49
to_shelve = prompt_user(patch_list)
50
51
if len(to_shelve) == 0:
118
52
print >>sys.stderr, 'Nothing to shelve'
119
53
return True
120
54
121
shelf = next_shelf()
55
root = run_bzr('root')[0].strip()
56
shelf = os.path.join(root, '.bzr-shelf')
122
57
print >>sys.stderr, "Saving shelved patches to", shelf
123
58
shelf = open(shelf, 'a')
124
if message is not None:
125
assert '\n' not in message
126
shelf.write("# shelf: %s\n" % message)
127
for patch in patches:
59
60
for patch in to_shelve:
128
61
shelf.write(str(patch))
129
62
130
63
shelf.flush()
132
65
shelf.close()
133
66
134
67
print >>sys.stderr, "Reverting shelved patches"
135
pipe = os.popen('patch -d %s -sR -p0' % tree_root(), 'w')
136
for patch in patches:
68
pipe = os.popen('patch -d %s -sR -p0' % root, 'w')
69
for patch in to_shelve:
137
70
pipe.write(str(patch))
138
71
pipe.flush()
139
72
154
87
raise Exception("Failed running bzr")
155
88
return lines
156
89
90
def prompt_user(patch_list):
91
total = 0
92
for patch in patch_list:
93
total += len(patch.hunks)
94
95
out = sys.stdout
96
inp = sys.stdin
97
98
to_shelve = []
99
i = 1
100
for patch in patch_list:
101
if patch.oldname != patch.newname:
102
name = "%s -> %s" % (patch.oldname, patch.newname)
103
else:
104
name = patch.oldname
105
106
hunks = patch.hunks[:]
107
for hunk in hunks:
108
print name
109
print hunk
110
while True:
111
out.write('Shelve this change? (%d of %d) [yn] ' % (i, total))
112
line = inp.readline().strip().lower()
113
if line == 'y' or line == 'yes':
114
break
115
elif line == 'n' or line == 'no':
116
patch.hunks.remove(hunk)
117
break
118
119
i += 1
120
121
if len(patch.hunks) > 0:
122
to_shelve.append(patch)
123
124
return to_shelve
125
157
126
class cmd_shelve(Command):
158
127
"""Temporarily remove some changes from the current tree.
159
128
Use 'unshelve' to restore these changes.
160
161
If filenames are specified, only changes to those files will be unshelved.
162
If a revision is specified, all changes since that revision will may be
163
unshelved.
164
129
"""
165
takes_args = ['file*']
166
takes_options = ['message', 'revision']
167
def run(self, file_list=None, message=None, revision=None):
168
return shelve(message=message, file_list=file_list, revision=revision)
130
131
def run(self):
132
return shelve()
169
133
170
134
class cmd_unshelve(Command):
171
135
"""Restore previously-shelved changes to the current tree.
173
137
"""
174
138
def run(self):
175
139
return unshelve()
176
177
class HunkSelector:
178
class Option:
179
def __init__(self, char, action, help, default=False):
180
self.char = char
181
self.action = action
182
self.default = default
183
self.help = help
184
185
standard_options = [
186
Option('n', 'shelve', 'shelve this change for the moment.',
187
default=True),
188
Option('y', 'keep', 'keep this change in your tree.'),
189
Option('d', 'done', 'done, skip to the end.'),
190
Option('i', 'invert', 'invert the current selection of all hunks.'),
191
Option('s', 'status', 'show status of hunks.'),
192
Option('q', 'quit', 'quit')
193
]
194
195
end_options = [
196
Option('y', 'continue', 'proceed to shelve selected changes.',
197
default=True),
198
Option('r', 'restart', 'restart the hunk selection loop.'),
199
Option('s', 'status', 'show status of hunks.'),
200
Option('i', 'invert', 'invert the current selection of all hunks.'),
201
Option('q', 'quit', 'quit')
202
]
203
204
def __init__(self, patches):
205
self.patches = patches
206
self.total_hunks = 0
207
for patch in patches:
208
for hunk in patch.hunks:
209
# everything's shelved by default
210
hunk.selected = True
211
self.total_hunks += 1
212
213
def __get_option(self, char):
214
for opt in self.standard_options:
215
if opt.char == char:
216
return opt
217
raise Exception('Option "%s" not found!' % char)
218
219
def __select_loop(self):
220
j = 0
221
for patch in self.patches:
222
i = 0
223
lasti = -1
224
while i < len(patch.hunks):
225
hunk = patch.hunks[i]
226
if lasti != i:
227
print patch.get_header(), hunk
228
j += 1
229
lasti = i
230
231
prompt = 'Keep this change? (%d of %d)' \
232
% (j, self.total_hunks)
233
234
if hunk.selected:
235
self.__get_option('n').default = True
236
self.__get_option('y').default = False
237
else:
238
self.__get_option('n').default = False
239
self.__get_option('y').default = True
240
241
action = self.__ask_user(prompt, self.standard_options)
242
243
if action == 'keep':
244
hunk.selected = False
245
elif action == 'shelve':
246
hunk.selected = True
247
elif action == 'done':
248
return True
249
elif action == 'invert':
250
self.__invert_selection()
251
self.__show_status()
252
continue
253
elif action == 'status':
254
self.__show_status()
255
continue
256
elif action == 'quit':
257
return False
258
259
i += 1
260
return True
261
262
def select(self):
263
if self.total_hunks == 0:
264
return []
265
266
done = False
267
while not done:
268
if not self.__select_loop():
269
return []
270
271
while True:
272
self.__show_status()
273
prompt = "Shelve these changes, or restart?"
274
action = self.__ask_user(prompt, self.end_options)
275
276
if action == 'continue':
277
done = True
278
break
279
elif action == 'quit':
280
return []
281
elif action == 'status':
282
self.__show_status()
283
elif action == 'invert':
284
self.__invert_selection()
285
elif action == 'restart':
286
break
287
288
289
for patch in self.patches:
290
tmp = []
291
for hunk in patch.hunks:
292
if hunk.selected:
293
tmp.append(hunk)
294
patch.hunks = tmp
295
296
tmp = []
297
for patch in self.patches:
298
if len(patch.hunks):
299
tmp.append(patch)
300
self.patches = tmp
301
302
return self.patches
303
304
def __invert_selection(self):
305
for patch in self.patches:
306
for hunk in patch.hunks:
307
if hunk.__dict__.has_key('selected'):
308
hunk.selected = not hunk.selected
309
else:
310
hunk.selected = True
311
312
def __show_status(self):
313
print '\nStatus:'
314
for patch in self.patches:
315
print ' %s' % patch.oldname
316
shelve = 0
317
keep = 0
318
for hunk in patch.hunks:
319
if hunk.selected:
320
shelve += 1
321
else:
322
keep += 1
323
324
print ' %d hunks to be shelved' % shelve
325
print ' %d hunks to be kept' % keep
326
print
327
328
def __getchar(self):
329
fd = sys.stdin.fileno()
330
settings = termios.tcgetattr(fd)
331
try:
332
tty.setraw(fd)
333
ch = sys.stdin.read(1)
334
finally:
335
termios.tcsetattr(fd, termios.TCSADRAIN, settings)
336
return ch
337
338
def __ask_user(self, prompt, options):
339
while True:
340
sys.stdout.write(prompt)
341
sys.stdout.write(' [')
342
for opt in options:
343
if opt.default:
344
default = opt
345
sys.stdout.write(opt.char)
346
sys.stdout.write('?] (%s): ' % default.char)
347
348
response = self.__getchar()
349
350
# default, which we see as newline, is 'n'
351
if response in ['\n', '\r', '\r\n']:
352
response = default.char
353
354
print response # because echo is off
355
356
for opt in options:
357
if opt.char == response:
358
return opt.action
359
360
for opt in options:
361
print ' %s - %s' % (opt.char, opt.help)
Loggerhead is a web-based interface for Breezy
Version: 2.0.1