← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to shelf.py

  • Committer: Aaron Bentley
  • Date: 2006-05-17 22:02:34 UTC
  • Revision ID: abentley@panoramicfeedback.com-20060517220234-51a588052d24b7d0
Start work on import plugin

Show diffs side-by-side

added added

removed removed

Lines of Context:
shelf.py
7
7
from errors import CommandError, PatchFailed
8
8
from hunk_selector import ShelveHunkSelector, UnshelveHunkSelector
9
9
from patchsource import PatchSource, FilePatchSource
10
 
from bzrlib.osutils import rename
11
10
 
12
11
class Shelf(object):
13
12
    MESSAGE_PREFIX = "# Shelved patch: "
57
56
 
58
57
    def delete(self, patch):
59
58
        path = self.__path_from_user(patch)
60
 
        rename(path, '%s~' % path)
 
59
        os.rename(path, '%s~' % path)
61
60
 
62
 
    def display(self, patch=None):
63
 
        if patch is None:
64
 
            path = self.last_patch()
65
 
        else:
66
 
            path = self.__path_from_user(patch)
 
61
    def display(self, patch):
 
62
        path = self.__path_from_user(patch)
67
63
        sys.stdout.write(open(path).read())
68
64
 
69
65
    def list(self):
82
78
    def __path_from_user(self, patch_id):
83
79
        try:
84
80
            patch_index = int(patch_id)
85
 
        except (TypeError, ValueError):
 
81
        except TypeError:
86
82
            raise CommandError("Invalid patch name '%s'" % patch_id)
87
83
 
88
84
        path = self.__path(patch_index)
134
130
            return None
135
131
        return patch[len(self.MESSAGE_PREFIX):patch.index('\n')]
136
132
 
137
 
    def unshelve(self, patch_source, patch_name=None, all=False, force=False):
 
133
    def unshelve(self, patch_source, all=False, force=False):
138
134
        self._check_upgrade()
139
135
 
 
136
        patch_name = self.last_patch()
 
137
 
140
138
        if patch_name is None:
141
 
            patch_path = self.last_patch()
142
 
        else:
143
 
            patch_path = self.__path_from_user(patch_name)
144
 
 
145
 
        if patch_path is None:
146
139
            raise CommandError("No patch found on shelf %s" % self.name)
147
140
 
148
 
        patches = FilePatchSource(patch_path).readpatches()
 
141
        patches = FilePatchSource(patch_name).readpatches()
149
142
        if all:
150
143
            to_unshelve = patches
151
144
            to_remain = []
155
148
        if len(to_unshelve) == 0:
156
149
            raise CommandError('Nothing to unshelve')
157
150
 
158
 
        message = self.get_patch_message(patch_path)
 
151
        message = self.get_patch_message(patch_name)
159
152
        if message is None:
160
153
            message = "No message saved with patch."
161
154
        self.log('Unshelving from %s/%s: "%s"\n' % \
162
 
                (self.name, os.path.basename(patch_path), message))
 
155
                (self.name, os.path.basename(patch_name), message))
163
156
 
164
157
        try:
165
158
            self._run_patch(to_unshelve, dry_run=True)
181
174
                    "longer applies cleanly to the working tree!")
182
175
 
183
176
        # Backup the shelved patch
184
 
        rename(patch_path, '%s~' % patch_path)
 
177
        os.rename(patch_name, '%s~' % patch_name)
185
178
 
186
179
        if len(to_remain) > 0:
187
 
            f = open(patch_path, 'w')
 
180
            f = open(patch_name, 'w')
188
181
            for patch in to_remain:
189
182
                f.write(str(patch))
190
183
            f.close()
206
199
            timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
207
200
            message = "Changes shelved on %s" % timestamp
208
201
 
209
 
        patch_path = self.next_patch()
 
202
        patch_name = self.next_patch()
210
203
        self.log('Shelving to %s/%s: "%s"\n' % \
211
 
                (self.name, os.path.basename(patch_path), message))
 
204
                (self.name, os.path.basename(patch_name), message))
212
205
 
213
 
        f = open(patch_path, 'a')
 
206
        f = open(patch_name, 'a')
214
207
 
215
208
        assert '\n' not in message
216
209
        f.write("%s%s\n" % (self.MESSAGE_PREFIX, message))
235
228
 
236
229
    def _run_patch(self, patches, strip=0, reverse=False, dry_run=False):
237
230
        args = ['patch', '-d', self.base, '-s', '-p%d' % strip, '-f']
238
 
 
239
 
        if sys.platform == "win32":
240
 
            args.append('--binary')
241
 
 
242
231
        if reverse:
243
232
            args.append('-R')
244
233
        if dry_run:
315
304
            new_file.close()
316
305
            self.log('Copied %s to %s/%s\n' % (os.path.basename(patch),
317
306
                self.name, os.path.basename(new_path)))
318
 
            rename(patch, patch + '~')
 
307
            os.rename(patch, patch + '~')