← 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-22 06:37:43 UTC
  • Revision ID: mbp@sourcefrog.net-20050622063743-e395f04c4db8977f
- move old blackbox code from testbzr into bzrlib.selftest.blackbox

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/check.py
15
15
# along with this program; if not, write to the Free Software
16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
17
 
18
 
import bzrlib.ui
19
 
from bzrlib.trace import note, warning
20
 
 
21
 
def _update_store_entry(obj, obj_id, branch, store_name, store):
22
 
    """This is just a meta-function, which handles both revision entries
23
 
    and inventory entries.
24
 
    """
25
 
    from bzrlib.trace import mutter
26
 
    import tempfile, os, errno
27
 
    from osutils import rename
28
 
    obj_tmp = tempfile.TemporaryFile()
29
 
    obj.write_xml(obj_tmp)
30
 
    obj_tmp.seek(0)
31
 
 
32
 
    tmpfd, tmp_path = tempfile.mkstemp(prefix=obj_id, suffix='.gz',
33
 
        dir=branch.controlfilename(store_name))
34
 
    os.close(tmpfd)
35
 
    try:
36
 
        orig_obj_path = branch.controlfilename([store_name, obj_id+'.gz'])
37
 
        # Remove the old entry out of the way
38
 
        rename(orig_obj_path, tmp_path)
39
 
        try:
40
 
            # TODO: We may need to handle the case where the old
41
 
            # entry was not compressed (and thus did not end with .gz)
42
 
 
43
 
            store.add(obj_tmp, obj_id) # Add the new one
44
 
            os.remove(tmp_path) # Remove the old name
45
 
            mutter('    Updated %s entry {%s}' % (store_name, obj_id))
46
 
        except:
47
 
            # On any exception, restore the old entry
48
 
            rename(tmp_path, orig_obj_path)
49
 
            raise
50
 
    finally:
51
 
        if os.path.exists(tmp_path):
52
 
            # Unfortunately, the next command might throw
53
 
            # an exception, which will mask a previous exception.
54
 
            os.remove(tmp_path)
55
 
        obj_tmp.close()
56
 
 
57
 
def _update_revision_entry(rev, branch):
58
 
    """After updating the values in a revision, make sure to
59
 
    write out the data, but try to do it in an atomic manner.
60
 
 
61
 
    :param rev:    The Revision object to store
62
 
    :param branch: The Branch object where this Revision is to be stored.
63
 
    """
64
 
    _update_store_entry(rev, rev.revision_id, branch,
65
 
            'revision-store', branch.revision_store)
66
 
 
67
 
def _update_inventory_entry(inv, inv_id, branch):
68
 
    """When an inventory has been modified (such as by adding a unique tree root)
69
 
    this atomically re-generates the file.
70
 
 
71
 
    :param inv:     The Inventory
72
 
    :param inv_id:  The inventory id for this inventory
73
 
    :param branch:  The Branch where this entry will be stored.
74
 
    """
75
 
    _update_store_entry(inv, inv_id, branch,
76
 
            'inventory-store', branch.inventory_store)
 
18
 
77
19
 
78
20
 
79
21
def check(branch):
80
22
    """Run consistency checks on a branch.
81
23
 
82
 
    TODO: Also check non-mainline revisions mentioned as parents.
83
 
 
84
 
    TODO: Check for extra files in the control directory.
 
24
    TODO: Also check non-mailine revisions mentioned as parents.
85
25
    """
86
26
    from bzrlib.trace import mutter
87
 
    from bzrlib.errors import BzrCheckError, NoSuchRevision
 
27
    from bzrlib.errors import BzrCheckError
88
28
    from bzrlib.osutils import fingerprint_file
89
 
    from bzrlib.inventory import ROOT_ID
90
 
    from bzrlib.branch import gen_root_id
 
29
    from bzrlib.progress import ProgressBar
91
30
 
92
31
    branch.lock_read()
93
32
 
94
33
    try:
 
34
        pb = ProgressBar(show_spinner=True)
95
35
        last_rev_id = None
96
36
 
97
37
        missing_inventory_sha_cnt = 0
98
38
        missing_revision_sha_cnt = 0
99
 
        missing_revision_cnt = 0
100
39
 
101
40
        history = branch.revision_history()
102
41
        revno = 0
103
42
        revcount = len(history)
104
 
        mismatch_inv_id = []
105
43
 
106
44
        # for all texts checked, text_id -> sha1
107
45
        checked_texts = {}
108
46
 
109
 
        progress = bzrlib.ui.ui_factory.progress_bar()
110
 
 
111
47
        for rev_id in history:
112
48
            revno += 1
113
 
            progress.update('checking revision', revno, revcount)
114
 
            # mutter('    revision {%s}' % rev_id)
 
49
            pb.update('checking revision', revno, revcount)
 
50
            mutter('    revision {%s}' % rev_id)
115
51
            rev = branch.get_revision(rev_id)
116
52
            if rev.revision_id != rev_id:
117
53
                raise BzrCheckError('wrong internal revision id in revision {%s}'
136
72
                        missing_revision_sha_cnt += 1
137
73
                        continue
138
74
                    prid = prr.revision_id
139
 
                    
140
 
                    try:
141
 
                        actual_sha = branch.get_revision_sha1(prid)
142
 
                    except NoSuchRevision:
143
 
                        missing_revision_cnt += 1
144
 
                        mutter("parent {%s} of {%s} not present in store",
145
 
                               prid, rev_id)
146
 
                        continue
147
 
                        
 
75
                    actual_sha = branch.get_revision_sha1(prid)
148
76
                    if prr.revision_sha1 != actual_sha:
149
77
                        raise BzrCheckError("mismatched revision sha1 for "
150
78
                                            "parent {%s} of {%s}: %s vs %s"
155
83
                                    "by {%s}"
156
84
                                    % (rev_id, last_rev_id))
157
85
 
158
 
            if rev.inventory_id != rev_id:
159
 
                mismatch_inv_id.append(rev_id)
160
 
 
161
86
            ## TODO: Check all the required fields are present on the revision.
162
87
 
163
88
            if rev.inventory_sha1:
185
110
            for file_id in inv:
186
111
                i += 1
187
112
                if i & 31 == 0:
188
 
                    progress.tick()
 
113
                    pb.tick()
189
114
 
190
115
                ie = inv[file_id]
191
116
 
214
139
                        raise BzrCheckError('directory {%s} has text in revision {%s}'
215
140
                                % (file_id, rev_id))
216
141
 
217
 
            progress.tick()
 
142
            pb.tick()
218
143
            for path, ie in inv.iter_entries():
219
144
                if path in seen_names:
220
145
                    raise BzrCheckError('duplicated path %s '
226
151
    finally:
227
152
        branch.unlock()
228
153
 
229
 
    progress.clear()
 
154
    pb.clear()
230
155
 
231
 
    note('checked %d revisions, %d file texts' % (revcount, len(checked_texts)))
 
156
    print 'checked %d revisions, %d file texts' % (revcount, len(checked_texts))
232
157
    
233
158
    if missing_inventory_sha_cnt:
234
 
        note('%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt)
 
159
        print '%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt
235
160
 
236
161
    if missing_revision_sha_cnt:
237
 
        note('%d parent links are missing revision_sha1' % missing_revision_sha_cnt)
238
 
 
239
 
    if missing_revision_cnt:
240
 
        note('%d revisions are mentioned but not present' % missing_revision_cnt)
241
 
 
242
 
    if missing_revision_cnt:
243
 
        print '%d revisions are mentioned but not present' % missing_revision_cnt
244
 
 
245
 
    # stub this out for now because the main bzr branch has references
246
 
    # to revisions that aren't present in the store -- mbp 20050804
247
 
#    if (missing_inventory_sha_cnt
248
 
#        or missing_revision_sha_cnt):
249
 
#        print '  (use "bzr upgrade" to fix them)'
250
 
 
251
 
    if mismatch_inv_id:
252
 
        warning('%d revisions have mismatched inventory ids:' % len(mismatch_inv_id))
253
 
        for rev_id in mismatch_inv_id:
254
 
            warning('  %s', rev_id)
 
162
        print '%d parent links are missing revision_sha1' % missing_revision_sha_cnt
 
163
 
 
164
    if (missing_inventory_sha_cnt
 
165
        or missing_revision_sha_cnt):
 
166
        print '  (use "bzr upgrade" to fix them)'