← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to reweave_inventory.py

  • Committer: Aaron Bentley
  • Date: 2005-11-08 17:14:39 UTC
  • Revision ID: aaron.bentley@utoronto.ca-20051108171439-6835d23881be10b8
Added force-reweave-inventory from Daniel Silverstone

Show diffs side-by-side

added added

removed removed

Lines of Context:
reweave_inventory.py
 
1
# (C) 2005 Canonical
 
2
 
 
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.
 
7
 
 
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.
 
12
 
 
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
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""Plugin to force-reweave the inventory of a branch.
 
18
 
 
19
This makes sure that the inventory weave's DAG of ancestry is correct so that
 
20
attempts to fetch the branch over http, or certain merge operations cope
 
21
correctly.
 
22
 
 
23
This is most likely needed if you have used fetch-ghosts from bzrlib to
 
24
resolve ghosts after a baz (or otherwise) import and you get bizarre behaviour
 
25
when either exporting over http or when merging from other translated branches.
 
26
"""
 
27
 
 
28
from bzrlib.commands import Command
 
29
from bzrlib.weavefile import write_weave_v5 as w5
 
30
 
 
31
import bzrlib.branch
 
32
import bzrlib.progress
 
33
 
 
34
import os
 
35
 
 
36
class cmd_force_reweave_inventory(Command):
 
37
    """Force-reweave the inventory in this branch.
 
38
 
 
39
    The branch URL *MUST* be on the local filesystem."""
 
40
    takes_args = ['branch_dir?']
 
41
 
 
42
    def run(self, branch_dir="."):
 
43
        branch = bzrlib.branch.Branch.open(branch_dir)
 
44
        branch.lock_write()
 
45
        try:
 
46
            self._do_reweave(branch, branch_dir)
 
47
        finally:
 
48
            branch.unlock()
 
49
 
 
50
    def _do_reweave(self, branch, branch_dir):
 
51
        inventory_weave = branch.get_inventory_weave()
 
52
 
 
53
        new_weave = bzrlib.weave.Weave()
 
54
 
 
55
        pending = list(inventory_weave.iter_names())
 
56
        total = len(pending)
 
57
        progress = bzrlib.progress.ProgressBar()
 
58
 
 
59
        while pending:
 
60
            index = pending.pop(0)
 
61
            done = total - len(pending)
 
62
            progress.update('re-weaving', done, total)
 
63
            rev = branch.get_revision(index)
 
64
            parents = []
 
65
            for parent in rev.parent_ids:
 
66
                if parent in inventory_weave:
 
67
                    parents.append(parent)
 
68
            unavailable = [p for p in parents if p not in new_weave]
 
69
            if len(unavailable) == 0:
 
70
                new_weave.add(index, parents, inventory_weave.get(index))
 
71
            else:
 
72
                pending.append(index)
 
73
 
 
74
                
 
75
        progress.update('Writing weave')
 
76
 
 
77
        f = open(os.path.join(branch_dir, ".bzr", "inventory.weave"), "w")
 
78
        w5(new_weave, f)
 
79
        f.close()
 
80
 
 
81
        progress.clear()