~bzr-pqm/bzr/bzr.dev

2052.3.1 by John Arbash Meinel
Add tests to cleanup the copyright of all source files
1
# Copyright (C) 2005 Canonical Ltd
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
2
#
236 by mbp at sourcefrog
- Experiments in inventory performance
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.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
7
#
236 by mbp at sourcefrog
- Experiments in inventory performance
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.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
12
#
236 by mbp at sourcefrog
- Experiments in inventory performance
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
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
236 by mbp at sourcefrog
- Experiments in inventory performance
16
17
1185.1.41 by Robert Collins
massive patch from Alexander Belchenko - many PEP8 fixes, removes unused function uuid
18
from bzrlib.errors import BzrError
19
from bzrlib.inventory import InventoryEntry, Inventory
236 by mbp at sourcefrog
- Experiments in inventory performance
20
21
22
START_MARK = "# bzr inventory format 3\n"
23
END_MARK = "# end of inventory\n"
24
25
26
def escape(s):
27
    """Very simple URL-like escaping.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
28
236 by mbp at sourcefrog
- Experiments in inventory performance
29
    (Why not just use backslashes?  Because then we couldn't parse
30
    lines just by splitting on spaces.)"""
31
    return (s.replace('\\', r'\x5c')
32
            .replace(' ', r'\x20')
33
            .replace('\t', r'\x09')
34
            .replace('\n', r'\x0a'))
35
36
37
38
def unescape(s):
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
39
    if s.find(' ') != -1:
40
        raise AssertionError()
236 by mbp at sourcefrog
- Experiments in inventory performance
41
    s = (s.replace(r'\x20', ' ')
42
         .replace(r'\x09', '\t')
43
         .replace(r'\x0a', '\n')
44
         .replace(r'\x5c', '\\'))
45
46
    # TODO: What if there's anything else?
47
48
    return s
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
49
50
236 by mbp at sourcefrog
- Experiments in inventory performance
51
52
53
def write_text_inventory(inv, outf):
54
    """Write out inv in a simple trad-unix text format."""
55
    outf.write(START_MARK)
1731.1.39 by Aaron Bentley
Reject removing is_root
56
    for path, ie in inv.iter_entries():
57
        if inv.is_root(ie.file_id):
58
            continue
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
59
236 by mbp at sourcefrog
- Experiments in inventory performance
60
        outf.write(ie.file_id + ' ')
61
        outf.write(escape(ie.name) + ' ')
62
        outf.write(ie.kind + ' ')
63
        outf.write(ie.parent_id + ' ')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
64
236 by mbp at sourcefrog
- Experiments in inventory performance
65
        if ie.kind == 'file':
66
            outf.write(ie.text_id)
67
            outf.write(' ' + ie.text_sha1)
68
            outf.write(' ' + str(ie.text_size))
69
        outf.write("\n")
70
    outf.write(END_MARK)
71
72
73
def read_text_inventory(tf):
74
    """Return an inventory read in from tf"""
75
    if tf.readline() != START_MARK:
76
        raise BzrError("missing start mark")
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
77
236 by mbp at sourcefrog
- Experiments in inventory performance
78
    inv = Inventory()
79
80
    for l in tf:
81
        fields = l.split(' ')
82
        if fields[0] == '#':
83
            break
84
        ie = {'file_id': fields[0],
85
              'name': unescape(fields[1]),
86
              'kind': fields[2],
87
              'parent_id': fields[3]}
88
        ##inv.add(ie)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
89
236 by mbp at sourcefrog
- Experiments in inventory performance
90
    if l != END_MARK:
91
        raise BzrError("missing end mark")
92
    return inv