← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/check.py

  • Committer: aaron.bentley at utoronto
  • Date: 2005-09-04 02:59:56 UTC
  • mfrom: (1172)
  • mto: (1185.3.4)
  • mto: This revision was merged to the branch mainline in revision 1178.
  • Revision ID: aaron.bentley@utoronto.ca-20050904025956-776ba4f07de97700
Merged mpool's latest changes (~0.0.7)

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
 
 
 
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)
19
77
 
20
78
 
21
79
def check(branch):
22
80
    """Run consistency checks on a branch.
23
81
 
24
 
    TODO: Also check non-mailine revisions mentioned as parents.
 
82
    TODO: Also check non-mainline revisions mentioned as parents.
 
83
 
 
84
    TODO: Check for extra files in the control directory.
25
85
    """
26
86
    from bzrlib.trace import mutter
27
 
    from bzrlib.errors import BzrCheckError
 
87
    from bzrlib.errors import BzrCheckError, NoSuchRevision
28
88
    from bzrlib.osutils import fingerprint_file
29
 
    from bzrlib.progress import ProgressBar
 
89
    from bzrlib.inventory import ROOT_ID
 
90
    from bzrlib.branch import gen_root_id
30
91
 
31
92
    branch.lock_read()
32
93
 
33
94
    try:
34
 
        pb = ProgressBar(show_spinner=True)
35
95
        last_rev_id = None
36
96
 
37
97
        missing_inventory_sha_cnt = 0
38
98
        missing_revision_sha_cnt = 0
 
99
        missing_revision_cnt = 0
39
100
 
40
101
        history = branch.revision_history()
41
102
        revno = 0
42
103
        revcount = len(history)
 
104
        mismatch_inv_id = []
43
105
 
44
106
        # for all texts checked, text_id -> sha1
45
107
        checked_texts = {}
46
108
 
 
109
        progress = bzrlib.ui.ui_factory.progress_bar()
 
110
 
47
111
        for rev_id in history:
48
112
            revno += 1
49
 
            pb.update('checking revision', revno, revcount)
50
 
            mutter('    revision {%s}' % rev_id)
 
113
            progress.update('checking revision', revno, revcount)
 
114
            # mutter('    revision {%s}' % rev_id)
51
115
            rev = branch.get_revision(rev_id)
52
116
            if rev.revision_id != rev_id:
53
117
                raise BzrCheckError('wrong internal revision id in revision {%s}'
72
136
                        missing_revision_sha_cnt += 1
73
137
                        continue
74
138
                    prid = prr.revision_id
75
 
                    actual_sha = branch.get_revision_sha1(prid)
 
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
                        
76
148
                    if prr.revision_sha1 != actual_sha:
77
149
                        raise BzrCheckError("mismatched revision sha1 for "
78
150
                                            "parent {%s} of {%s}: %s vs %s"
83
155
                                    "by {%s}"
84
156
                                    % (rev_id, last_rev_id))
85
157
 
 
158
            if rev.inventory_id != rev_id:
 
159
                mismatch_inv_id.append(rev_id)
 
160
 
86
161
            ## TODO: Check all the required fields are present on the revision.
87
162
 
88
163
            if rev.inventory_sha1:
110
185
            for file_id in inv:
111
186
                i += 1
112
187
                if i & 31 == 0:
113
 
                    pb.tick()
 
188
                    progress.tick()
114
189
 
115
190
                ie = inv[file_id]
116
191
 
139
214
                        raise BzrCheckError('directory {%s} has text in revision {%s}'
140
215
                                % (file_id, rev_id))
141
216
 
142
 
            pb.tick()
 
217
            progress.tick()
143
218
            for path, ie in inv.iter_entries():
144
219
                if path in seen_names:
145
220
                    raise BzrCheckError('duplicated path %s '
151
226
    finally:
152
227
        branch.unlock()
153
228
 
154
 
    pb.clear()
 
229
    progress.clear()
155
230
 
156
 
    print 'checked %d revisions, %d file texts' % (revcount, len(checked_texts))
 
231
    note('checked %d revisions, %d file texts' % (revcount, len(checked_texts)))
157
232
    
158
233
    if missing_inventory_sha_cnt:
159
 
        print '%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt
 
234
        note('%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt)
160
235
 
161
236
    if missing_revision_sha_cnt:
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)'
 
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)