14
16
# along with this program; if not, write to the Free Software
15
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""bzr upgrade logic."""
19
# change upgrade from .bzr to create a '.bzr-new', then do a bait and switch.
22
from bzrlib.bzrdir import ConvertBzrDir4To5, ConvertBzrDir5To6, BzrDir, BzrDirFormat4, BzrDirFormat5
23
import bzrlib.errors as errors
24
from bzrlib.transport import get_transport
25
import bzrlib.ui as ui
28
class Convert(object):
30
def __init__(self, url, format):
32
self.bzrdir = BzrDir.open_unsupported(url)
33
if self.bzrdir.root_transport.is_readonly():
34
raise errors.UpgradeReadonly
35
self.transport = self.bzrdir.root_transport
36
self.pb = ui.ui_factory.nested_progress_bar()
44
branch = self.bzrdir.open_branch()
45
if branch.bzrdir.root_transport.base != \
46
self.bzrdir.root_transport.base:
47
self.pb.note("This is a checkout. The branch (%s) needs to be "
48
"upgraded separately.",
49
branch.bzrdir.root_transport.base)
50
except errors.NotBranchError:
52
if not self.bzrdir.needs_format_conversion(self.format):
53
raise errors.UpToDateFormat(self.bzrdir._format)
54
if not self.bzrdir.can_convert_format():
55
raise errors.BzrError("cannot upgrade from branch format %s" %
57
self.pb.note('starting upgrade of %s', self.transport.base)
58
self._backup_control_dir()
59
while self.bzrdir.needs_format_conversion(self.format):
60
converter = self.bzrdir._format.get_converter(self.format)
61
self.bzrdir = converter.convert(self.bzrdir, self.pb)
62
self.pb.note("finished")
64
def _backup_control_dir(self):
65
self.pb.note('making backup of tree history')
66
self.transport.copy_tree('.bzr', '.bzr.backup')
67
self.pb.note('%s.bzr has been backed up to %s.bzr.backup',
70
self.pb.note('if conversion fails, you can move this directory back to .bzr')
71
self.pb.note('if it succeeds, you can remove this directory if you wish')
73
def upgrade(url, format=None):
74
"""Upgrade to format, or the default bzrdir format if not supplied."""
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
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 = bzrlib.branch.find_branch('.')
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:]:
added
removed
tools/history2weaves.py
