~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/knitrepo.py

  • Committer: Martin Packman
  • Date: 2011-12-19 10:37:57 UTC
  • mto: This revision was merged to the branch mainline in revision 6394.
  • Revision ID: martin.packman@canonical.com-20111219103757-b85as9n9pb7e6qvn
Add tests for deprecated unicode wrapper functions in win32utils

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
from bzrlib.lazy_import import lazy_import
18
18
lazy_import(globals(), """
 
19
import itertools
 
20
 
19
21
from bzrlib import (
20
22
    bzrdir,
21
23
    errors,
32
34
    xml7,
33
35
    )
34
36
""")
35
 
from bzrlib import (
36
 
    symbol_versioning,
37
 
    )
38
37
from bzrlib.decorators import needs_read_lock, needs_write_lock
39
38
from bzrlib.repository import (
40
 
    CommitBuilder,
41
 
    MetaDirRepository,
42
 
    MetaDirRepositoryFormat,
43
 
    RepositoryFormat,
44
 
    RootCommitBuilder,
45
 
    )
 
39
    InterRepository,
 
40
    IsInWriteGroupError,
 
41
    RepositoryFormatMetaDir,
 
42
    )
 
43
from bzrlib.vf_repository import (
 
44
    InterSameDataRepository,
 
45
    MetaDirVersionedFileRepository,
 
46
    MetaDirVersionedFileRepositoryFormat,
 
47
    VersionedFileCommitBuilder,
 
48
    VersionedFileRootCommitBuilder,
 
49
    )
 
50
from bzrlib import symbol_versioning
46
51
 
47
52
 
48
53
class _KnitParentsProvider(object):
102
107
        return result
103
108
 
104
109
 
105
 
class KnitRepository(MetaDirRepository):
 
110
class KnitRepository(MetaDirVersionedFileRepository):
106
111
    """Knit format repository."""
107
112
 
108
113
    # These attributes are inherited from the Repository base class. Setting
114
119
 
115
120
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
116
121
        _serializer):
117
 
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
 
122
        super(KnitRepository, self).__init__(_format, a_bzrdir, control_files)
118
123
        self._commit_builder_class = _commit_builder_class
119
124
        self._serializer = _serializer
120
125
        self._reconcile_fixes_text_parents = True
176
181
        result.get_parent_map([('A',)])
177
182
        return result
178
183
 
179
 
    def fileid_involved_between_revs(self, from_revid, to_revid):
180
 
        """Find file_id(s) which are involved in the changes between revisions.
181
 
 
182
 
        This determines the set of revisions which are involved, and then
183
 
        finds all file ids affected by those revisions.
184
 
        """
185
 
        vf = self._get_revision_vf()
186
 
        from_set = set(vf.get_ancestry(from_revid))
187
 
        to_set = set(vf.get_ancestry(to_revid))
188
 
        changed = to_set.difference(from_set)
189
 
        return self._fileid_involved_by_set(changed)
190
 
 
191
 
    def fileid_involved(self, last_revid=None):
192
 
        """Find all file_ids modified in the ancestry of last_revid.
193
 
 
194
 
        :param last_revid: If None, last_revision() will be used.
195
 
        """
196
 
        if not last_revid:
197
 
            changed = set(self.all_revision_ids())
198
 
        else:
199
 
            changed = set(self.get_ancestry(last_revid))
200
 
        if None in changed:
201
 
            changed.remove(None)
202
 
        return self._fileid_involved_by_set(changed)
203
 
 
204
184
    @needs_read_lock
205
185
    def get_revision(self, revision_id):
206
186
        """Return the Revision object for a named revision"""
210
190
    def _refresh_data(self):
211
191
        if not self.is_locked():
212
192
            return
 
193
        if self.is_in_write_group():
 
194
            raise IsInWriteGroupError(self)
213
195
        # Create a new transaction to force all knits to see the scope change.
214
196
        # This is safe because we're outside a write group.
215
197
        self.control_files._finish_transaction()
229
211
    def _make_parents_provider(self):
230
212
        return _KnitsParentsProvider(self.revisions)
231
213
 
232
 
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
233
 
        """Find revisions with different parent lists in the revision object
234
 
        and in the index graph.
235
 
 
236
 
        :param revisions_iterator: None, or an iterator of (revid,
237
 
            Revision-or-None). This iterator controls the revisions checked.
238
 
        :returns: an iterator yielding tuples of (revison-id, parents-in-index,
239
 
            parents-in-revision).
240
 
        """
241
 
        if not self.is_locked():
242
 
            raise AssertionError()
243
 
        vf = self.revisions
244
 
        if revisions_iterator is None:
245
 
            revisions_iterator = self._iter_revisions(None)
246
 
        for revid, revision in revisions_iterator:
247
 
            if revision is None:
248
 
                pass
249
 
            parent_map = vf.get_parent_map([(revid,)])
250
 
            parents_according_to_index = tuple(parent[-1] for parent in
251
 
                parent_map[(revid,)])
252
 
            parents_according_to_revision = tuple(revision.parent_ids)
253
 
            if parents_according_to_index != parents_according_to_revision:
254
 
                yield (revid, parents_according_to_index,
255
 
                    parents_according_to_revision)
256
 
 
257
 
    def _check_for_inconsistent_revision_parents(self):
258
 
        inconsistencies = list(self._find_inconsistent_revision_parents())
259
 
        if inconsistencies:
260
 
            raise errors.BzrCheckError(
261
 
                "Revision knit has inconsistent parents.")
262
 
 
263
 
    def revision_graph_can_have_wrong_parents(self):
264
 
        # The revision.kndx could potentially claim a revision has a different
265
 
        # parent to the revision text.
266
 
        return True
267
 
 
268
 
 
269
 
class RepositoryFormatKnit(MetaDirRepositoryFormat):
 
214
 
 
215
class RepositoryFormatKnit(MetaDirVersionedFileRepositoryFormat):
270
216
    """Bzr repository knit format (generalized).
271
217
 
272
218
    This repository format has:
301
247
    _fetch_order = 'topological'
302
248
    _fetch_uses_deltas = True
303
249
    fast_deltas = False
 
250
    supports_funky_characters = True
 
251
    # The revision.kndx could potentially claim a revision has a different
 
252
    # parent to the revision text.
 
253
    revision_graph_can_have_wrong_parents = True
304
254
 
305
255
    def _get_inventories(self, repo_transport, repo, name='inventory'):
306
256
        mapper = versionedfile.ConstantMapper(name)
371
321
                                    than normal. I.e. during 'upgrade'.
372
322
        """
373
323
        if not _found:
374
 
            format = RepositoryFormat.find_format(a_bzrdir)
 
324
            format = RepositoryFormatMetaDir.find_format(a_bzrdir)
375
325
        if _override_transport is not None:
376
326
            repo_transport = _override_transport
377
327
        else:
409
359
    """
410
360
 
411
361
    repository_class = KnitRepository
412
 
    _commit_builder_class = CommitBuilder
 
362
    _commit_builder_class = VersionedFileCommitBuilder
413
363
    @property
414
364
    def _serializer(self):
415
365
        return xml5.serializer_v5
417
367
    def __ne__(self, other):
418
368
        return self.__class__ is not other.__class__
419
369
 
420
 
    def get_format_string(self):
 
370
    @classmethod
 
371
    def get_format_string(cls):
421
372
        """See RepositoryFormat.get_format_string()."""
422
373
        return "Bazaar-NG Knit Repository Format 1"
423
374
 
443
394
    """
444
395
 
445
396
    repository_class = KnitRepository
446
 
    _commit_builder_class = RootCommitBuilder
 
397
    _commit_builder_class = VersionedFileRootCommitBuilder
447
398
    rich_root_data = True
448
399
    experimental = True
449
400
    supports_tree_reference = True
459
410
 
460
411
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
461
412
 
462
 
    def get_format_string(self):
 
413
    @classmethod
 
414
    def get_format_string(cls):
463
415
        """See RepositoryFormat.get_format_string()."""
464
416
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
465
417
 
485
437
    """
486
438
 
487
439
    repository_class = KnitRepository
488
 
    _commit_builder_class = RootCommitBuilder
 
440
    _commit_builder_class = VersionedFileRootCommitBuilder
489
441
    rich_root_data = True
490
442
    supports_tree_reference = False
491
443
    @property
500
452
 
501
453
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
502
454
 
503
 
    def get_format_string(self):
 
455
    @classmethod
 
456
    def get_format_string(cls):
504
457
        """See RepositoryFormat.get_format_string()."""
505
458
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'
506
459
 
507
460
    def get_format_description(self):
508
461
        """See RepositoryFormat.get_format_description()."""
509
462
        return "Knit repository format 4"
 
463
 
 
464
 
 
465
class InterKnitRepo(InterSameDataRepository):
 
466
    """Optimised code paths between Knit based repositories."""
 
467
 
 
468
    @classmethod
 
469
    def _get_repo_format_to_test(self):
 
470
        return RepositoryFormatKnit1()
 
471
 
 
472
    @staticmethod
 
473
    def is_compatible(source, target):
 
474
        """Be compatible with known Knit formats.
 
475
 
 
476
        We don't test for the stores being of specific types because that
 
477
        could lead to confusing results, and there is no need to be
 
478
        overly general.
 
479
        """
 
480
        try:
 
481
            are_knits = (isinstance(source._format, RepositoryFormatKnit) and
 
482
                isinstance(target._format, RepositoryFormatKnit))
 
483
        except AttributeError:
 
484
            return False
 
485
        return are_knits and InterRepository._same_model(source, target)
 
486
 
 
487
    @needs_read_lock
 
488
    def search_missing_revision_ids(self,
 
489
            revision_id=symbol_versioning.DEPRECATED_PARAMETER,
 
490
            find_ghosts=True, revision_ids=None, if_present_ids=None,
 
491
            limit=None):
 
492
        """See InterRepository.search_missing_revision_ids()."""
 
493
        if symbol_versioning.deprecated_passed(revision_id):
 
494
            symbol_versioning.warn(
 
495
                'search_missing_revision_ids(revision_id=...) was '
 
496
                'deprecated in 2.4.  Use revision_ids=[...] instead.',
 
497
                DeprecationWarning, stacklevel=2)
 
498
            if revision_ids is not None:
 
499
                raise AssertionError(
 
500
                    'revision_ids is mutually exclusive with revision_id')
 
501
            if revision_id is not None:
 
502
                revision_ids = [revision_id]
 
503
        del revision_id
 
504
        source_ids_set = self._present_source_revisions_for(
 
505
            revision_ids, if_present_ids)
 
506
        # source_ids is the worst possible case we may need to pull.
 
507
        # now we want to filter source_ids against what we actually
 
508
        # have in target, but don't try to check for existence where we know
 
509
        # we do not have a revision as that would be pointless.
 
510
        target_ids = set(self.target.all_revision_ids())
 
511
        possibly_present_revisions = target_ids.intersection(source_ids_set)
 
512
        actually_present_revisions = set(
 
513
            self.target._eliminate_revisions_not_present(possibly_present_revisions))
 
514
        required_revisions = source_ids_set.difference(actually_present_revisions)
 
515
        if revision_ids is not None:
 
516
            # we used get_ancestry to determine source_ids then we are assured all
 
517
            # revisions referenced are present as they are installed in topological order.
 
518
            # and the tip revision was validated by get_ancestry.
 
519
            result_set = required_revisions
 
520
        else:
 
521
            # if we just grabbed the possibly available ids, then
 
522
            # we only have an estimate of whats available and need to validate
 
523
            # that against the revision records.
 
524
            result_set = set(
 
525
                self.source._eliminate_revisions_not_present(required_revisions))
 
526
        if limit is not None:
 
527
            topo_ordered = self.source.get_graph().iter_topo_order(result_set)
 
528
            result_set = set(itertools.islice(topo_ordered, limit))
 
529
        return self.source.revision_ids_to_search_result(result_set)
 
530
 
 
531
 
 
532
InterRepository.register_optimiser(InterKnitRepo)