← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to tools/convertinv.py

  • Committer: Martin Pool
  • Date: 2005-07-07 10:22:02 UTC
  • Revision ID: mbp@sourcefrog.net-20050707102201-2d2a13a25098b101
- rearrange and clear up merged weave

Show diffs side-by-side

added added

removed removed

Lines of Context:
tools/convertinv.py
20
20
 
21
21
 
22
22
import bzrlib.branch
23
 
from bzrlib.weave import Weave
24
 
from bzrlib.weavefile import write_weave
25
 
from bzrlib.progress import ProgressBar
 
23
from weave import Weave
 
24
from weavefile import write_weave
26
25
import tempfile
27
 
import hotshot, hotshot.stats
28
 
import sys
 
26
import hotshot
29
27
 
30
28
def convert():
31
29
    WEAVE_NAME = "inventory.weave"
32
30
 
33
 
    pb = ProgressBar()
34
 
 
35
31
    wf = Weave()
36
32
 
37
33
    b = bzrlib.branch.find_branch('.')
38
34
 
 
35
    print 'converting...'
 
36
 
39
37
    parents = set()
40
38
    revno = 1
41
 
    rev_history = b.revision_history()
42
 
    for rev_id in rev_history:
43
 
        pb.update('converting inventory', revno, len(rev_history))
 
39
    for rev_id in b.revision_history():
 
40
        print revno
44
41
        inv_xml = b.inventory_store[rev_id].readlines()
45
42
        weave_id = wf.add(parents, inv_xml)
46
 
        parents = set([weave_id])       # always just one parent
 
43
        parents.add(weave_id)
47
44
        revno += 1
48
45
 
49
 
    pb.update('write weave', None, None)
50
46
    write_weave(wf, file(WEAVE_NAME, 'wb'))
51
47
 
52
 
    pb.clear()
53
 
 
54
 
 
55
 
def profile_convert(): 
56
 
    prof_f = tempfile.NamedTemporaryFile()
57
 
 
58
 
    prof = hotshot.Profile(prof_f.name)
59
 
 
60
 
    prof.runcall(convert) 
61
 
    prof.close()
62
 
 
63
 
    stats = hotshot.stats.load(prof_f.name)
64
 
    #stats.strip_dirs()
65
 
    stats.sort_stats('time')
66
 
    ## XXX: Might like to write to stderr or the trace file instead but
67
 
    ## print_stats seems hardcoded to stdout
68
 
    stats.print_stats(20)
 
48
 
 
49
 
 
50
 
 
51
prof_f = tempfile.NamedTemporaryFile()
 
52
 
 
53
prof = hotshot.Profile(prof_f.name)
 
54
 
 
55
prof.runcall(convert) 
 
56
prof.close()
 
57
 
 
58
import hotshot.stats
 
59
stats = hotshot.stats.load(prof_f.name)
 
60
#stats.strip_dirs()
 
61
stats.sort_stats('time')
 
62
## XXX: Might like to write to stderr or the trace file instead but
 
63
## print_stats seems hardcoded to stdout
 
64
stats.print_stats(20)
69
65
            
70
66
 
71
 
if '-p' in sys.argv[1:]:
72
 
    profile_convert()
73
 
else:
74
 
    convert()
75