← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/check.py

  • Committer: Martin Pool
  • Date: 2005-06-27 01:26:11 UTC
  • Revision ID: mbp@sourcefrog.net-20050627012611-4effb7007553fde1
- tweak rsync upload script

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/check.py
16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
17
 
18
18
 
19
 
def _update_store_entry(obj, obj_id, branch, store_name, store):
20
 
    """This is just a meta-function, which handles both revision entries
21
 
    and inventory entries.
22
 
    """
23
 
    from bzrlib.trace import mutter
24
 
    import tempfile, os, errno
25
 
    from osutils import rename
26
 
    obj_tmp = tempfile.TemporaryFile()
27
 
    obj.write_xml(obj_tmp)
28
 
    obj_tmp.seek(0)
29
 
 
30
 
    tmpfd, tmp_path = tempfile.mkstemp(prefix=obj_id, suffix='.gz',
31
 
        dir=branch.controlfilename(store_name))
32
 
    os.close(tmpfd)
33
 
    try:
34
 
        orig_obj_path = branch.controlfilename([store_name, obj_id+'.gz'])
35
 
        # Remove the old entry out of the way
36
 
        rename(orig_obj_path, tmp_path)
37
 
        try:
38
 
            # TODO: We may need to handle the case where the old
39
 
            # entry was not compressed (and thus did not end with .gz)
40
 
 
41
 
            store.add(obj_tmp, obj_id) # Add the new one
42
 
            os.remove(tmp_path) # Remove the old name
43
 
            mutter('    Updated %s entry {%s}' % (store_name, obj_id))
44
 
        except:
45
 
            # On any exception, restore the old entry
46
 
            rename(tmp_path, orig_obj_path)
47
 
            raise
48
 
    finally:
49
 
        if os.path.exists(tmp_path):
50
 
            # Unfortunately, the next command might throw
51
 
            # an exception, which will mask a previous exception.
52
 
            os.remove(tmp_path)
53
 
        obj_tmp.close()
54
 
 
55
 
def _update_revision_entry(rev, branch):
56
 
    """After updating the values in a revision, make sure to
57
 
    write out the data, but try to do it in an atomic manner.
58
 
 
59
 
    :param rev:    The Revision object to store
60
 
    :param branch: The Branch object where this Revision is to be stored.
61
 
    """
62
 
    _update_store_entry(rev, rev.revision_id, branch,
63
 
            'revision-store', branch.revision_store)
64
 
 
65
 
def _update_inventory_entry(inv, inv_id, branch):
66
 
    """When an inventory has been modified (such as by adding a unique tree root)
67
 
    this atomically re-generates the file.
68
 
 
69
 
    :param inv:     The Inventory
70
 
    :param inv_id:  The inventory id for this inventory
71
 
    :param branch:  The Branch where this entry will be stored.
72
 
    """
73
 
    _update_store_entry(inv, inv_id, branch,
74
 
            'inventory-store', branch.inventory_store)
 
19
 
75
20
 
76
21
def check(branch):
77
22
    """Run consistency checks on a branch.
78
23
 
79
 
    TODO: Also check non-mainline revisions mentioned as parents.
80
 
 
81
 
    TODO: Check for extra files in the control directory.
 
24
    TODO: Also check non-mailine revisions mentioned as parents.
82
25
    """
83
26
    from bzrlib.trace import mutter
84
27
    from bzrlib.errors import BzrCheckError
85
28
    from bzrlib.osutils import fingerprint_file
86
29
    from bzrlib.progress import ProgressBar
87
 
    from bzrlib.inventory import ROOT_ID
88
 
    from bzrlib.branch import gen_root_id
89
30
 
90
31
    branch.lock_read()
91
32
 
99
40
        history = branch.revision_history()
100
41
        revno = 0
101
42
        revcount = len(history)
102
 
        mismatch_inv_id = []
103
43
 
104
44
        # for all texts checked, text_id -> sha1
105
45
        checked_texts = {}
143
83
                                    "by {%s}"
144
84
                                    % (rev_id, last_rev_id))
145
85
 
146
 
            if rev.inventory_id != rev_id:
147
 
                mismatch_inv_id.append(rev_id)
148
 
 
149
86
            ## TODO: Check all the required fields are present on the revision.
150
87
 
151
88
            if rev.inventory_sha1:
224
161
    if missing_revision_sha_cnt:
225
162
        print '%d parent links are missing revision_sha1' % missing_revision_sha_cnt
226
163
 
227
 
    # stub this out for now because the main bzr branch has references
228
 
    # to revisions that aren't present in the store -- mbp 20050804
229
 
#    if (missing_inventory_sha_cnt
230
 
#        or missing_revision_sha_cnt):
231
 
#        print '  (use "bzr upgrade" to fix them)'
232
 
 
233
 
    if mismatch_inv_id:
234
 
        print '%d revisions have mismatched inventory ids:' % len(mismatch_inv_id)
235
 
        for rev_id in mismatch_inv_id:
236
 
            print '  ', rev_id
 
164
    if (missing_inventory_sha_cnt
 
165
        or missing_revision_sha_cnt):
 
166
        print '  (use "bzr upgrade" to fix them)'