← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/patches.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2006-08-17 07:52:09 UTC
  • mfrom: (1910.3.4 trivial)
  • Revision ID: pqm@pqm.ubuntu.com-20060817075209-e85a1f9e05ff8b87
(andrew) Trivial fixes to NotImplemented errors.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/patches.py
1
 
# Copyright (C) 2004 - 2006 Aaron Bentley, Canonical Ltd
 
1
# Copyright (C) 2004 - 2006 Aaron Bentley
2
2
# <aaron.bentley@utoronto.ca>
3
3
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
 
import re
 
4
#    This program is free software; you can redistribute it and/or modify
 
5
#    it under the terms of the GNU General Public License as published by
 
6
#    the Free Software Foundation; either version 2 of the License, or
 
7
#    (at your option) any later version.
 
8
#
 
9
#    This program is distributed in the hope that it will be useful,
 
10
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
#    GNU General Public License for more details.
 
13
#
 
14
#    You should have received a copy of the GNU General Public License
 
15
#    along with this program; if not, write to the Free Software
 
16
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
17
 
19
18
 
20
19
class PatchSyntax(Exception):
95
94
 
96
95
 
97
96
def hunk_from_header(line):
98
 
    matches = re.match(r'\@\@ ([^@]*) \@\@( (.*))?\n', line)
99
 
    if matches is None:
100
 
        raise MalformedHunkHeader("Does not match format.", line)
 
97
    if not line.startswith("@@") or not line.endswith("@@\n") \
 
98
        or not len(line) > 4:
 
99
        raise MalformedHunkHeader("Does not start and end with @@.", line)
101
100
    try:
102
 
        (orig, mod) = matches.group(1).split(" ")
103
 
    except (ValueError, IndexError), e:
 
101
        (orig, mod) = line[3:-4].split(" ")
 
102
    except Exception, e:
104
103
        raise MalformedHunkHeader(str(e), line)
105
104
    if not orig.startswith('-') or not mod.startswith('+'):
106
105
        raise MalformedHunkHeader("Positions don't start with + or -.", line)
107
106
    try:
108
107
        (orig_pos, orig_range) = parse_range(orig[1:])
109
108
        (mod_pos, mod_range) = parse_range(mod[1:])
110
 
    except (ValueError, IndexError), e:
 
109
    except Exception, e:
111
110
        raise MalformedHunkHeader(str(e), line)
112
111
    if mod_range < 0 or orig_range < 0:
113
112
        raise MalformedHunkHeader("Hunk range is negative", line)
114
 
    tail = matches.group(3)
115
 
    return Hunk(orig_pos, orig_range, mod_pos, mod_range, tail)
 
113
    return Hunk(orig_pos, orig_range, mod_pos, mod_range)
116
114
 
117
115
 
118
116
class HunkLine:
172
170
 
173
171
 
174
172
class Hunk:
175
 
    def __init__(self, orig_pos, orig_range, mod_pos, mod_range, tail=None):
 
173
    def __init__(self, orig_pos, orig_range, mod_pos, mod_range):
176
174
        self.orig_pos = orig_pos
177
175
        self.orig_range = orig_range
178
176
        self.mod_pos = mod_pos
179
177
        self.mod_range = mod_range
180
 
        self.tail = tail
181
178
        self.lines = []
182
179
 
183
180
    def get_header(self):
184
 
        if self.tail is None:
185
 
            tail_str = ''
186
 
        else:
187
 
            tail_str = ' ' + self.tail
188
 
        return "@@ -%s +%s @@%s\n" % (self.range_str(self.orig_pos,
189
 
                                                     self.orig_range),
190
 
                                      self.range_str(self.mod_pos,
191
 
                                                     self.mod_range),
192
 
                                      tail_str)
 
181
        return "@@ -%s +%s @@\n" % (self.range_str(self.orig_pos, 
 
182
                                                   self.orig_range),
 
183
                                    self.range_str(self.mod_pos, 
 
184
                                                   self.mod_range))
193
185
 
194
186
    def range_str(self, pos, range):
195
187
        """Return a file range, special-casing for 1-line files.
324
316
 
325
317
def iter_file_patch(iter_lines):
326
318
    saved_lines = []
327
 
    orig_range = 0
328
319
    for line in iter_lines:
329
320
        if line.startswith('=== ') or line.startswith('*** '):
330
321
            continue
331
322
        if line.startswith('#'):
332
323
            continue
333
 
        elif orig_range > 0:
334
 
            if line.startswith('-') or line.startswith(' '):
335
 
                orig_range -= 1
336
324
        elif line.startswith('--- '):
337
325
            if len(saved_lines) > 0:
338
326
                yield saved_lines
339
327
            saved_lines = []
340
 
        elif line.startswith('@@'):
341
 
            hunk = hunk_from_header(line)
342
 
            orig_range = hunk.orig_range
343
328
        saved_lines.append(line)
344
329
    if len(saved_lines) > 0:
345
330
        yield saved_lines