← 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-07-07 10:22:02 UTC
  • Revision ID: mbp@sourcefrog.net-20050707102201-2d2a13a25098b101
- rearrange and clear up merged weave

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):
84
26
    TODO: Check for extra files in the control directory.
85
27
    """
86
28
    from bzrlib.trace import mutter
87
 
    from bzrlib.errors import BzrCheckError, NoSuchRevision
 
29
    from bzrlib.errors import BzrCheckError
88
30
    from bzrlib.osutils import fingerprint_file
89
 
    from bzrlib.inventory import ROOT_ID
90
 
    from bzrlib.branch import gen_root_id
 
31
    from bzrlib.progress import ProgressBar
91
32
 
92
33
    branch.lock_read()
93
34
 
94
35
    try:
 
36
        pb = ProgressBar(show_spinner=True)
95
37
        last_rev_id = None
96
38
 
97
39
        missing_inventory_sha_cnt = 0
98
40
        missing_revision_sha_cnt = 0
99
 
        missing_revision_cnt = 0
100
41
 
101
42
        history = branch.revision_history()
102
43
        revno = 0
106
47
        # for all texts checked, text_id -> sha1
107
48
        checked_texts = {}
108
49
 
109
 
        progress = bzrlib.ui.ui_factory.progress_bar()
110
 
 
111
50
        for rev_id in history:
112
51
            revno += 1
113
 
            progress.update('checking revision', revno, revcount)
114
 
            # mutter('    revision {%s}' % rev_id)
 
52
            pb.update('checking revision', revno, revcount)
 
53
            mutter('    revision {%s}' % rev_id)
115
54
            rev = branch.get_revision(rev_id)
116
55
            if rev.revision_id != rev_id:
117
56
                raise BzrCheckError('wrong internal revision id in revision {%s}'
136
75
                        missing_revision_sha_cnt += 1
137
76
                        continue
138
77
                    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
 
                        
 
78
                    actual_sha = branch.get_revision_sha1(prid)
148
79
                    if prr.revision_sha1 != actual_sha:
149
80
                        raise BzrCheckError("mismatched revision sha1 for "
150
81
                                            "parent {%s} of {%s}: %s vs %s"
185
116
            for file_id in inv:
186
117
                i += 1
187
118
                if i & 31 == 0:
188
 
                    progress.tick()
 
119
                    pb.tick()
189
120
 
190
121
                ie = inv[file_id]
191
122
 
214
145
                        raise BzrCheckError('directory {%s} has text in revision {%s}'
215
146
                                % (file_id, rev_id))
216
147
 
217
 
            progress.tick()
 
148
            pb.tick()
218
149
            for path, ie in inv.iter_entries():
219
150
                if path in seen_names:
220
151
                    raise BzrCheckError('duplicated path %s '
226
157
    finally:
227
158
        branch.unlock()
228
159
 
229
 
    progress.clear()
 
160
    pb.clear()
230
161
 
231
 
    note('checked %d revisions, %d file texts' % (revcount, len(checked_texts)))
 
162
    print 'checked %d revisions, %d file texts' % (revcount, len(checked_texts))
232
163
    
233
164
    if missing_inventory_sha_cnt:
234
 
        note('%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt)
 
165
        print '%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt
235
166
 
236
167
    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)'
 
168
        print '%d parent links are missing revision_sha1' % missing_revision_sha_cnt
 
169
 
 
170
    if (missing_inventory_sha_cnt
 
171
        or missing_revision_sha_cnt):
 
172
        print '  (use "bzr upgrade" to fix them)'
250
173
 
251
174
    if mismatch_inv_id:
252
 
        warning('%d revisions have mismatched inventory ids:' % len(mismatch_inv_id))
 
175
        print '%d revisions have mismatched inventory ids:' % len(mismatch_inv_id)
253
176
        for rev_id in mismatch_inv_id:
254
 
            warning('  %s', rev_id)
 
177
            print '  ', rev_id