3
# Copyright (C) 2005 Canonical Ltd
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.
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.
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
19
"""Experiment in converting existing bzr branches to weaves."""
21
# To make this properly useful
23
# 1. assign text version ids, and put those text versions into
24
# the inventory as they're converted.
26
# 2. keep track of the previous version of each file, rather than
27
# just using the last one imported
29
# 3. assign entry versions when files are added, renamed or moved.
31
# 4. when merged-in versions are observed, walk down through them
32
# to discover everything, then commit bottom-up
34
# 5. track ancestry as things are merged in, and commit that in each
37
# Perhaps it's best to first walk the whole graph and make a plan for
38
# what should be imported in what order? Need a kind of topological
39
# sort of all revisions. (Or do we, can we just before doing a revision
40
# see that all its parents have either been converted or abandoned?)
50
import hotshot, hotshot.stats
54
from bzrlib.branch import Branch, find_branch
55
from bzrlib.revfile import Revfile
56
from bzrlib.weave import Weave
57
from bzrlib.weavefile import read_weave, write_weave
58
from bzrlib.progress import ProgressBar
59
from bzrlib.atomicfile import AtomicFile
65
bzrlib.trace.enable_default_logging()
73
# holds in-memory weaves for all files
76
b = Branch('.', relax_version_check=True)
79
rev_history = b.revision_history()
84
for rev_id in rev_history:
85
pb.update('converting revision', revno, len(rev_history))
87
inv_xml = b.get_inventory_xml(rev_id).readlines()
89
new_idx = inv_weave.add(rev_id, inv_parents, inv_xml)
90
inv_parents = [new_idx]
92
tree = b.revision_tree(rev_id)
95
# for each file in the inventory, put it into its own revfile
100
if last_text_sha.get(file_id) == ie.text_sha1:
103
last_text_sha[file_id] = ie.text_sha1
105
# new text (though possibly already stored); need to store it
106
text_lines = tree.get_file(file_id).readlines()
108
# if the file's created for the first time in this
109
# revision then make a new weave; else find the old one
110
if file_id not in text_weaves:
111
text_weaves[file_id] = Weave()
113
w = text_weaves[file_id]
115
# base the new text version off whatever was last
116
# (actually it'd be better to track this, to allow for
117
# files that are deleted and then reappear)
124
w.add(rev_id, parents, text_lines)
130
print '%6d revisions and inventories' % revno
131
print '%6d texts' % text_count
134
# TODO: commit them all atomically at the end, not one by one
135
write_atomic_weave(inv_weave, 'weaves/inventory.weave')
136
for file_id, file_weave in text_weaves.items():
137
pb.update('writing weave', i, len(text_weaves))
138
write_atomic_weave(file_weave, 'weaves/%s.weave' % file_id)
144
def write_atomic_weave(weave, filename):
145
inv_wf = AtomicFile(filename)
147
write_weave(weave, inv_wf)
155
def profile_convert():
156
prof_f = tempfile.NamedTemporaryFile()
158
prof = hotshot.Profile(prof_f.name)
160
prof.runcall(convert)
163
stats = hotshot.stats.load(prof_f.name)
165
stats.sort_stats('time')
166
# XXX: Might like to write to stderr or the trace file instead but
167
# print_stats seems hardcoded to stdout
168
stats.print_stats(20)
171
if '-p' in sys.argv[1:]: