← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/rio.py

Optimize common case where unique_lcs returns a set of lines all in a row

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/rio.py
1
 
# Copyright (C) 2005 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
1
# Copyright (C) 2005 by Canonical Ltd
 
2
#
 
3
# Distributed under the GNU General Public Licence v2
16
4
 
17
5
# \subsection{\emph{rio} - simple text metaformat}
18
6
# 
183
171
        """Return stanza as a single string"""
184
172
        return ''.join(self.to_lines())
185
173
 
186
 
    def to_unicode(self):
187
 
        """Return stanza as a single Unicode string.
188
 
 
189
 
        This is most useful when adding a Stanza to a parent Stanza
190
 
        """
191
 
        if not self.items:
192
 
            return u''
193
 
 
194
 
        result = []
195
 
        for tag, value in self.items:
196
 
            if value == '':
197
 
                result.append(tag + ': \n')
198
 
            elif '\n' in value:
199
 
                # don't want splitlines behaviour on empty lines
200
 
                val_lines = value.split('\n')
201
 
                result.append(tag + ': ' + val_lines[0] + '\n')
202
 
                for line in val_lines[1:]:
203
 
                    result.append('\t' + line + '\n')
204
 
            else:
205
 
                result.append(tag + ': ' + value + '\n')
206
 
        return u''.join(result)
207
 
 
208
174
    def write(self, to_file):
209
175
        """Write stanza to a file"""
210
176
        to_file.writelines(self.to_lines())
257
223
 
258
224
    The raw lines must be in utf-8 encoding.
259
225
    """
260
 
    unicode_iter = (line.decode('utf-8') for line in line_iter)
261
 
    return read_stanza_unicode(unicode_iter)
262
 
 
263
 
 
264
 
def read_stanza_unicode(unicode_iter):
265
 
    """Read a Stanza from a list of lines or a file.
266
 
 
267
 
    The lines should already be in unicode form. This returns a single
268
 
    stanza that was read. If there is a blank line at the end of the Stanza,
269
 
    it is consumed. It is not an error for there to be no blank line at
270
 
    the end of the iterable. If there is a blank line at the beginning,
271
 
    this is treated as an empty Stanza and None is returned.
272
 
 
273
 
    Only the stanza lines and the trailing blank (if any) are consumed
274
 
    from the unicode_iter
275
 
 
276
 
    :param unicode_iter: A iterable, yeilding Unicode strings. See read_stanza
277
 
        if you have a utf-8 encoded string.
278
 
    :return: A Stanza object if there are any lines in the file.
279
 
        None otherwise
280
 
    """
 
226
    items = []
281
227
    stanza = Stanza()
282
228
    tag = None
283
229
    accum_value = None
284
 
    
285
 
    # TODO: jam 20060922 This code should raise real errors rather than
286
 
    #       using 'assert' to process user input, or raising ValueError
287
 
    #       rather than a more specific error.
288
 
 
289
 
    for line in unicode_iter:
290
 
        if line is None or line == '':
 
230
    for line in line_iter:
 
231
        if line == None or line == '':
291
232
            break       # end of file
292
233
        if line == '\n':
293
234
            break       # end of stanza
294
 
        assert line.endswith('\n')
 
235
        line = line.decode('utf-8')
 
236
        assert line[-1] == '\n'
295
237
        real_l = line
296
238
        if line[0] == '\t': # continues previous value
297
239
            if tag is None:
303
245
            try:
304
246
                colon_index = line.index(': ')
305
247
            except ValueError:
306
 
                raise ValueError('tag/value separator not found in line %r'
307
 
                                 % real_l)
 
248
                raise ValueError('tag/value separator not found in line %r' % real_l)
308
249
            tag = str(line[:colon_index])
309
250
            assert valid_tag(tag), \
310
251
                    "invalid rio tag %r" % tag
311
252
            accum_value = line[colon_index+2:-1]
312
 
 
313
253
    if tag is not None: # add last tag-value
314
254
        stanza.add(tag, accum_value)
315
255
        return stanza