~bzr-pqm/bzr/bzr.dev

0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
1
#! /usr/bin/python
2
3
# Copyright (C) 2005 Canonical Ltd
4
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
19
"""Experiment in converting existing bzr branches to weaves."""
20
21
22
import sys
23
import bzrlib.branch
942 by Martin Pool
- update convertinv to work with current weave code
24
from bzrlib.weave import Weave
25
from bzrlib.weavefile import write_weave
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
26
27
import hotshot
28
import tempfile
29
30
31
def convert():
32
    WEAVE_NAME = "test.weave"
33
34
    wf = Weave()
35
0.1.92 by Martin Pool
Use proper relative path when converting a file from bzr
36
    toconvert = sys.argv[1]
37
    
38
    b = bzrlib.branch.find_branch(toconvert)
39
    rp = b.relpath(toconvert)
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
40
41
    print 'converting...'
42
0.1.92 by Martin Pool
Use proper relative path when converting a file from bzr
43
    fid = b.read_working_inventory().path2id(rp)
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
44
0.1.81 by Martin Pool
Avoid re-encoding versions which have not changed
45
    last_lines = None
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
46
    parents = set()
0.1.81 by Martin Pool
Avoid re-encoding versions which have not changed
47
    revno = 0
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
48
    for rev_id in b.revision_history():
0.1.81 by Martin Pool
Avoid re-encoding versions which have not changed
49
        revno += 1
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
50
        print revno
51
        tree = b.revision_tree(rev_id)
52
        inv = tree.inventory
53
54
        if fid not in tree:
55
            print '  (not present)'
56
            continue
57
58
        text = tree.get_file(fid).readlines()
0.1.81 by Martin Pool
Avoid re-encoding versions which have not changed
59
60
        if text == last_lines:
61
            continue
62
        last_lines = text
63
        
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
64
        weave_id = wf.add(parents, text)
942 by Martin Pool
- update convertinv to work with current weave code
65
        parents = [weave_id]
0.1.80 by Martin Pool
Add test code to convert from old storage to a weave
66
67
        print '  %4d lines' % len(text)
68
69
    write_weave(wf, file(WEAVE_NAME, 'wb'))
70
71
72
prof_f = tempfile.NamedTemporaryFile()
73
74
prof = hotshot.Profile(prof_f.name)
75
76
prof.runcall(convert) 
77
prof.close()
78
79
import hotshot.stats
80
stats = hotshot.stats.load(prof_f.name)
81
#stats.strip_dirs()
82
stats.sort_stats('time')
83
## XXX: Might like to write to stderr or the trace file instead but
84
## print_stats seems hardcoded to stdout
85
stats.print_stats(20)
86
            
87