← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/rio.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/rio.py
32
32
 
33
33
import re
34
34
 
 
35
from bzrlib import osutils
35
36
from bzrlib.iterablefile import IterableFile
36
37
 
37
38
# XXX: some redundancy is allowing to write stanzas in isolation as well as
129
130
                            % (value, type(value)))
130
131
        self.items.append((tag, value))
131
132
 
 
133
    @classmethod
 
134
    def from_pairs(cls, pairs):
 
135
        ret = cls()
 
136
        ret.items = pairs
 
137
        return ret
 
138
 
132
139
    def __contains__(self, find_tag):
133
140
        """True if there is any field in this stanza with the given tag."""
134
141
        for tag, value in self.items:
191
198
 
192
199
        result = []
193
200
        for tag, value in self.items:
194
 
            if value == '':
195
 
                result.append(tag + ': \n')
196
 
            elif '\n' in value:
 
201
            if value == u'':
 
202
                result.append(tag + u': \n')
 
203
            elif u'\n' in value:
197
204
                # don't want splitlines behaviour on empty lines
198
 
                val_lines = value.split('\n')
199
 
                result.append(tag + ': ' + val_lines[0] + '\n')
 
205
                val_lines = value.split(u'\n')
 
206
                result.append(tag + u': ' + val_lines[0] + u'\n')
200
207
                for line in val_lines[1:]:
201
 
                    result.append('\t' + line + '\n')
 
208
                    result.append(u'\t' + line + u'\n')
202
209
            else:
203
 
                result.append(tag + ': ' + value + '\n')
 
210
                result.append(tag + u': ' + value + u'\n')
204
211
        return u''.join(result)
205
212
 
206
213
    def write(self, to_file):
236
243
            d[tag] = value
237
244
        return d
238
245
 
239
 
_tag_re = re.compile(r'^[-a-zA-Z0-9_]+$')
 
246
 
240
247
def valid_tag(tag):
241
 
    return bool(_tag_re.match(tag))
 
248
    return _valid_tag(tag)
242
249
 
243
250
 
244
251
def read_stanza(line_iter):
254
261
 
255
262
    The raw lines must be in utf-8 encoding.
256
263
    """
257
 
    unicode_iter = (line.decode('utf-8') for line in line_iter)
258
 
    return read_stanza_unicode(unicode_iter)
 
264
    return _read_stanza_utf8(line_iter)
259
265
 
260
266
 
261
267
def read_stanza_unicode(unicode_iter):
275
281
    :return: A Stanza object if there are any lines in the file.
276
282
        None otherwise
277
283
    """
278
 
    stanza = Stanza()
279
 
    tag = None
280
 
    accum_value = None
281
 
 
282
 
    # TODO: jam 20060922 This code should raise real errors rather than
283
 
    #       using 'assert' to process user input, or raising ValueError
284
 
    #       rather than a more specific error.
285
 
 
286
 
    for line in unicode_iter:
287
 
        if line is None or line == '':
288
 
            break       # end of file
289
 
        if line == '\n':
290
 
            break       # end of stanza
291
 
        real_l = line
292
 
        if line[0] == '\t': # continues previous value
293
 
            if tag is None:
294
 
                raise ValueError('invalid continuation line %r' % real_l)
295
 
            accum_value += '\n' + line[1:-1]
296
 
        else: # new tag:value line
297
 
            if tag is not None:
298
 
                stanza.add(tag, accum_value)
299
 
            try:
300
 
                colon_index = line.index(': ')
301
 
            except ValueError:
302
 
                raise ValueError('tag/value separator not found in line %r'
303
 
                                 % real_l)
304
 
            tag = str(line[:colon_index])
305
 
            if not valid_tag(tag):
306
 
                raise ValueError("invalid rio tag %r" % (tag,))
307
 
            accum_value = line[colon_index+2:-1]
308
 
 
309
 
    if tag is not None: # add last tag-value
310
 
        stanza.add(tag, accum_value)
311
 
        return stanza
312
 
    else:     # didn't see any content
313
 
        return None
 
284
    return _read_stanza_unicode(unicode_iter)
314
285
 
315
286
 
316
287
def to_patch_lines(stanza, max_width=72):
399
370
    :return: a Stanza
400
371
    """
401
372
    return read_stanza(_patch_stanza_iter(line_iter))
 
373
 
 
374
 
 
375
try:
 
376
    from bzrlib._rio_pyx import (
 
377
        _read_stanza_utf8,
 
378
        _read_stanza_unicode,
 
379
        _valid_tag,
 
380
        )
 
381
except ImportError, e:
 
382
    osutils.failed_to_load_extension(e)
 
383
    from bzrlib._rio_py import (
 
384
       _read_stanza_utf8,
 
385
       _read_stanza_unicode,
 
386
       _valid_tag,
 
387
       )