/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/repository.py

  • Committer: Parth Malwankar
  • Date: 2010-02-15 10:46:49 UTC
  • mto: This revision was merged to the branch mainline in revision 5048.
  • Revision ID: parth.malwankar@gmail.com-20100215104649-bo4z8rhqvfibnxhj
updated existing upgrade tests to work.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/repository.py
26
26
    chk_map,
27
27
    config,
28
28
    debug,
 
29
    errors,
29
30
    fetch as _mod_fetch,
30
31
    fifo_cache,
31
32
    generate_ids,
39
40
    lru_cache,
40
41
    osutils,
41
42
    revision as _mod_revision,
42
 
    static_tuple,
43
43
    symbol_versioning,
44
44
    trace,
45
45
    tsort,
 
46
    ui,
46
47
    versionedfile,
47
48
    )
48
49
from bzrlib.bundle import serializer
51
52
from bzrlib.testament import Testament
52
53
""")
53
54
 
54
 
from bzrlib import (
55
 
    errors,
56
 
    registry,
57
 
    ui,
58
 
    )
59
55
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
60
56
from bzrlib.inter import InterObject
61
57
from bzrlib.inventory import (
64
60
    ROOT_ID,
65
61
    entry_factory,
66
62
    )
67
 
from bzrlib.recordcounter import RecordCounter
68
 
from bzrlib.lock import _RelockDebugMixin, LogicalLockResult
 
63
from bzrlib.lock import _RelockDebugMixin
 
64
from bzrlib import registry
69
65
from bzrlib.trace import (
70
66
    log_exception_quietly, note, mutter, mutter_callsite, warning)
71
67
 
74
70
_deprecation_warning_done = False
75
71
 
76
72
 
77
 
class IsInWriteGroupError(errors.InternalBzrError):
78
 
 
79
 
    _fmt = "May not refresh_data of repo %(repo)s while in a write group."
80
 
 
81
 
    def __init__(self, repo):
82
 
        errors.InternalBzrError.__init__(self, repo=repo)
83
 
 
84
 
 
85
73
class CommitBuilder(object):
86
74
    """Provides an interface to build up a commit.
87
75
 
242
230
 
243
231
    def _gen_revision_id(self):
244
232
        """Return new revision-id."""
245
 
        return generate_ids.gen_revision_id(self._committer, self._timestamp)
 
233
        return generate_ids.gen_revision_id(self._config.username(),
 
234
                                            self._timestamp)
246
235
 
247
236
    def _generate_revision_if_needed(self):
248
237
        """Create a revision id if None was supplied.
288
277
 
289
278
        :param tree: The tree which is being committed.
290
279
        """
291
 
        if len(self.parents) == 0:
292
 
            raise errors.RootMissing()
 
280
        # NB: if there are no parents then this method is not called, so no
 
281
        # need to guard on parents having length.
293
282
        entry = entry_factory['directory'](tree.path2id(''), '',
294
283
            None)
295
284
        entry.revision = self._new_revision_id
870
859
        # versioned roots do not change unless the tree found a change.
871
860
 
872
861
 
873
 
class RepositoryWriteLockResult(LogicalLockResult):
874
 
    """The result of write locking a repository.
875
 
 
876
 
    :ivar repository_token: The token obtained from the underlying lock, or
877
 
        None.
878
 
    :ivar unlock: A callable which will unlock the lock.
879
 
    """
880
 
 
881
 
    def __init__(self, unlock, repository_token):
882
 
        LogicalLockResult.__init__(self, unlock)
883
 
        self.repository_token = repository_token
884
 
 
885
 
    def __repr__(self):
886
 
        return "RepositoryWriteLockResult(%s, %s)" % (self.repository_token,
887
 
            self.unlock)
888
 
 
889
 
 
890
862
######################################################################
891
863
# Repositories
892
864
 
893
865
 
894
 
class Repository(_RelockDebugMixin, bzrdir.ControlComponent):
 
866
class Repository(_RelockDebugMixin):
895
867
    """Repository holding history for one or more branches.
896
868
 
897
869
    The repository holds and retrieves historical information including
1045
1017
                " id and insertion revid (%r, %r)"
1046
1018
                % (inv.revision_id, revision_id))
1047
1019
        if inv.root is None:
1048
 
            raise errors.RootMissing()
 
1020
            raise AssertionError()
1049
1021
        return self._add_inventory_checked(revision_id, inv, parents)
1050
1022
 
1051
1023
    def _add_inventory_checked(self, revision_id, inv, parents):
1055
1027
 
1056
1028
        :seealso: add_inventory, for the contract.
1057
1029
        """
1058
 
        inv_lines = self._serializer.write_inventory_to_lines(inv)
 
1030
        inv_lines = self._serialise_inventory_to_lines(inv)
1059
1031
        return self._inventory_add_lines(revision_id, parents,
1060
1032
            inv_lines, check_content=False)
1061
1033
 
1318
1290
 
1319
1291
        :param _format: The format of the repository on disk.
1320
1292
        :param a_bzrdir: The BzrDir of the repository.
 
1293
 
 
1294
        In the future we will have a single api for all stores for
 
1295
        getting file texts, inventories and revisions, then
 
1296
        this construct will accept instances of those things.
1321
1297
        """
1322
 
        # In the future we will have a single api for all stores for
1323
 
        # getting file texts, inventories and revisions, then
1324
 
        # this construct will accept instances of those things.
1325
1298
        super(Repository, self).__init__()
1326
1299
        self._format = _format
1327
1300
        # the following are part of the public API for Repository:
1342
1315
        # rather copying them?
1343
1316
        self._safe_to_return_from_cache = False
1344
1317
 
1345
 
    @property
1346
 
    def user_transport(self):
1347
 
        return self.bzrdir.user_transport
1348
 
 
1349
 
    @property
1350
 
    def control_transport(self):
1351
 
        return self._transport
1352
 
 
1353
1318
    def __repr__(self):
1354
1319
        if self._fallback_repositories:
1355
1320
            return '%s(%r, fallback_repositories=%r)' % (
1403
1368
        data during reads, and allows a 'write_group' to be obtained. Write
1404
1369
        groups must be used for actual data insertion.
1405
1370
 
1406
 
        A token should be passed in if you know that you have locked the object
1407
 
        some other way, and need to synchronise this object's state with that
1408
 
        fact.
1409
 
 
1410
 
        XXX: this docstring is duplicated in many places, e.g. lockable_files.py
1411
 
 
1412
1371
        :param token: if this is already locked, then lock_write will fail
1413
1372
            unless the token matches the existing lock.
1414
1373
        :returns: a token if this instance supports tokens, otherwise None.
1417
1376
        :raises MismatchedToken: if the specified token doesn't match the token
1418
1377
            of the existing lock.
1419
1378
        :seealso: start_write_group.
1420
 
        :return: A RepositoryWriteLockResult.
 
1379
 
 
1380
        A token should be passed in if you know that you have locked the object
 
1381
        some other way, and need to synchronise this object's state with that
 
1382
        fact.
 
1383
 
 
1384
        XXX: this docstring is duplicated in many places, e.g. lockable_files.py
1421
1385
        """
1422
1386
        locked = self.is_locked()
1423
 
        token = self.control_files.lock_write(token=token)
 
1387
        result = self.control_files.lock_write(token=token)
1424
1388
        if not locked:
1425
1389
            self._warn_if_deprecated()
1426
1390
            self._note_lock('w')
1428
1392
                # Writes don't affect fallback repos
1429
1393
                repo.lock_read()
1430
1394
            self._refresh_data()
1431
 
        return RepositoryWriteLockResult(self.unlock, token)
 
1395
        return result
1432
1396
 
1433
1397
    def lock_read(self):
1434
 
        """Lock the repository for read operations.
1435
 
 
1436
 
        :return: An object with an unlock method which will release the lock
1437
 
            obtained.
1438
 
        """
1439
1398
        locked = self.is_locked()
1440
1399
        self.control_files.lock_read()
1441
1400
        if not locked:
1444
1403
            for repo in self._fallback_repositories:
1445
1404
                repo.lock_read()
1446
1405
            self._refresh_data()
1447
 
        return LogicalLockResult(self.unlock)
1448
1406
 
1449
1407
    def get_physical_lock_status(self):
1450
1408
        return self.control_files.get_physical_lock_status()
1510
1468
 
1511
1469
        # now gather global repository information
1512
1470
        # XXX: This is available for many repos regardless of listability.
1513
 
        if self.user_transport.listable():
 
1471
        if self.bzrdir.root_transport.listable():
1514
1472
            # XXX: do we want to __define len__() ?
1515
1473
            # Maybe the versionedfiles object should provide a different
1516
1474
            # method to get the number of keys.
1526
1484
        :param using: If True, list only branches using this repository.
1527
1485
        """
1528
1486
        if using and not self.is_shared():
1529
 
            return self.bzrdir.list_branches()
 
1487
            try:
 
1488
                return [self.bzrdir.open_branch()]
 
1489
            except errors.NotBranchError:
 
1490
                return []
1530
1491
        class Evaluator(object):
1531
1492
 
1532
1493
            def __init__(self):
1541
1502
                    except errors.NoRepositoryPresent:
1542
1503
                        pass
1543
1504
                    else:
1544
 
                        return False, ([], repository)
 
1505
                        return False, (None, repository)
1545
1506
                self.first_call = False
1546
 
                value = (bzrdir.list_branches(), None)
 
1507
                try:
 
1508
                    value = (bzrdir.open_branch(), None)
 
1509
                except errors.NotBranchError:
 
1510
                    value = (None, None)
1547
1511
                return True, value
1548
1512
 
1549
 
        ret = []
1550
 
        for branches, repository in bzrdir.BzrDir.find_bzrdirs(
1551
 
                self.user_transport, evaluate=Evaluator()):
1552
 
            if branches is not None:
1553
 
                ret.extend(branches)
 
1513
        branches = []
 
1514
        for branch, repository in bzrdir.BzrDir.find_bzrdirs(
 
1515
                self.bzrdir.root_transport, evaluate=Evaluator()):
 
1516
            if branch is not None:
 
1517
                branches.append(branch)
1554
1518
            if not using and repository is not None:
1555
 
                ret.extend(repository.find_branches())
1556
 
        return ret
 
1519
                branches.extend(repository.find_branches())
 
1520
        return branches
1557
1521
 
1558
1522
    @needs_read_lock
1559
1523
    def search_missing_revision_ids(self, other, revision_id=None, find_ghosts=True):
1668
1632
        return missing_keys
1669
1633
 
1670
1634
    def refresh_data(self):
1671
 
        """Re-read any data needed to synchronise with disk.
 
1635
        """Re-read any data needed to to synchronise with disk.
1672
1636
 
1673
1637
        This method is intended to be called after another repository instance
1674
1638
        (such as one used by a smart server) has inserted data into the
1675
 
        repository. On all repositories this will work outside of write groups.
1676
 
        Some repository formats (pack and newer for bzrlib native formats)
1677
 
        support refresh_data inside write groups. If called inside a write
1678
 
        group on a repository that does not support refreshing in a write group
1679
 
        IsInWriteGroupError will be raised.
 
1639
        repository. It may not be called during a write group, but may be
 
1640
        called at any other time.
1680
1641
        """
 
1642
        if self.is_in_write_group():
 
1643
            raise errors.InternalBzrError(
 
1644
                "May not refresh_data while in a write group.")
1681
1645
        self._refresh_data()
1682
1646
 
1683
1647
    def resume_write_group(self, tokens):
1722
1686
                "May not fetch while in a write group.")
1723
1687
        # fast path same-url fetch operations
1724
1688
        # TODO: lift out to somewhere common with RemoteRepository
1725
 
        # <https://bugs.launchpad.net/bzr/+bug/401646>
 
1689
        # <https://bugs.edge.launchpad.net/bzr/+bug/401646>
1726
1690
        if (self.has_same_location(source)
1727
1691
            and fetch_spec is None
1728
1692
            and self._has_same_fallbacks(source)):
1937
1901
                rev = self._serializer.read_revision_from_string(text)
1938
1902
                yield (revid, rev)
1939
1903
 
 
1904
    @needs_read_lock
 
1905
    def get_revision_xml(self, revision_id):
 
1906
        # TODO: jam 20070210 This shouldn't be necessary since get_revision
 
1907
        #       would have already do it.
 
1908
        # TODO: jam 20070210 Just use _serializer.write_revision_to_string()
 
1909
        # TODO: this can't just be replaced by:
 
1910
        # return self._serializer.write_revision_to_string(
 
1911
        #     self.get_revision(revision_id))
 
1912
        # as cStringIO preservers the encoding unlike write_revision_to_string
 
1913
        # or some other call down the path.
 
1914
        rev = self.get_revision(revision_id)
 
1915
        rev_tmp = cStringIO.StringIO()
 
1916
        # the current serializer..
 
1917
        self._serializer.write_revision(rev, rev_tmp)
 
1918
        rev_tmp.seek(0)
 
1919
        return rev_tmp.getvalue()
 
1920
 
1940
1921
    def get_deltas_for_revisions(self, revisions, specific_fileids=None):
1941
1922
        """Produce a generator of revision deltas.
1942
1923
 
2455
2436
                result.revision_id, revision_id))
2456
2437
        return result
2457
2438
 
 
2439
    def _serialise_inventory(self, inv):
 
2440
        return self._serializer.write_inventory_to_string(inv)
 
2441
 
 
2442
    def _serialise_inventory_to_lines(self, inv):
 
2443
        return self._serializer.write_inventory_to_lines(inv)
 
2444
 
2458
2445
    def get_serializer_format(self):
2459
2446
        return self._serializer.format_num
2460
2447
 
2524
2511
            else:
2525
2512
                next_id = parents[0]
2526
2513
 
 
2514
    @needs_read_lock
 
2515
    def get_revision_inventory(self, revision_id):
 
2516
        """Return inventory of a past revision."""
 
2517
        # TODO: Unify this with get_inventory()
 
2518
        # bzr 0.0.6 and later imposes the constraint that the inventory_id
 
2519
        # must be the same as its revision, so this is trivial.
 
2520
        if revision_id is None:
 
2521
            # This does not make sense: if there is no revision,
 
2522
            # then it is the current tree inventory surely ?!
 
2523
            # and thus get_root_id() is something that looks at the last
 
2524
            # commit on the branch, and the get_root_id is an inventory check.
 
2525
            raise NotImplementedError
 
2526
            # return Inventory(self.get_root_id())
 
2527
        else:
 
2528
            return self.get_inventory(revision_id)
 
2529
 
2527
2530
    def is_shared(self):
2528
2531
        """Return True if this repository is flagged as a shared repository."""
2529
2532
        raise NotImplementedError(self.is_shared)
2563
2566
            return RevisionTree(self, Inventory(root_id=None),
2564
2567
                                _mod_revision.NULL_REVISION)
2565
2568
        else:
2566
 
            inv = self.get_inventory(revision_id)
 
2569
            inv = self.get_revision_inventory(revision_id)
2567
2570
            return RevisionTree(self, inv, revision_id)
2568
2571
 
2569
2572
    def revision_trees(self, revision_ids):
2622
2625
            keys = tsort.topo_sort(parent_map)
2623
2626
        return [None] + list(keys)
2624
2627
 
2625
 
    def pack(self, hint=None, clean_obsolete_packs=False):
 
2628
    def pack(self, hint=None):
2626
2629
        """Compress the data within the repository.
2627
2630
 
2628
2631
        This operation only makes sense for some repository types. For other
2638
2641
            obtained from the result of commit_write_group(). Out of
2639
2642
            date hints are simply ignored, because concurrent operations
2640
2643
            can obsolete them rapidly.
2641
 
 
2642
 
        :param clean_obsolete_packs: Clean obsolete packs immediately after
2643
 
            the pack operation.
2644
2644
        """
2645
2645
 
2646
2646
    def get_transaction(self):
2671
2671
    def _make_parents_provider(self):
2672
2672
        return self
2673
2673
 
2674
 
    @needs_read_lock
2675
 
    def get_known_graph_ancestry(self, revision_ids):
2676
 
        """Return the known graph for a set of revision ids and their ancestors.
2677
 
        """
2678
 
        st = static_tuple.StaticTuple
2679
 
        revision_keys = [st(r_id).intern() for r_id in revision_ids]
2680
 
        known_graph = self.revisions.get_known_graph_ancestry(revision_keys)
2681
 
        return graph.GraphThunkIdsToKeys(known_graph)
2682
 
 
2683
2674
    def get_graph(self, other_repository=None):
2684
2675
        """Return the graph walker for this repository format"""
2685
2676
        parents_provider = self._make_parents_provider()
3087
3078
    # Is the format experimental ?
3088
3079
    experimental = False
3089
3080
 
3090
 
    def __repr__(self):
3091
 
        return "%s()" % self.__class__.__name__
 
3081
    def __str__(self):
 
3082
        return "<%s>" % self.__class__.__name__
3092
3083
 
3093
3084
    def __eq__(self, other):
3094
3085
        # format objects are generally stateless
3212
3203
        """
3213
3204
        raise NotImplementedError(self.open)
3214
3205
 
3215
 
    def _run_post_repo_init_hooks(self, repository, a_bzrdir, shared):
3216
 
        from bzrlib.bzrdir import BzrDir, RepoInitHookParams
3217
 
        hooks = BzrDir.hooks['post_repo_init']
3218
 
        if not hooks:
3219
 
            return
3220
 
        params = RepoInitHookParams(repository, self, a_bzrdir, shared)
3221
 
        for hook in hooks:
3222
 
            hook(params)
3223
 
 
3224
3206
 
3225
3207
class MetaDirRepositoryFormat(RepositoryFormat):
3226
3208
    """Common base class for the new repositories using the metadir layout."""
3435
3417
        :return: None.
3436
3418
        """
3437
3419
        ui.ui_factory.warn_experimental_format_fetch(self)
3438
 
        from bzrlib.fetch import RepoFetcher
3439
 
        # See <https://launchpad.net/bugs/456077> asking for a warning here
3440
 
        if self.source._format.network_name() != self.target._format.network_name():
3441
 
            ui.ui_factory.show_user_warning('cross_format_fetch',
3442
 
                from_format=self.source._format,
3443
 
                to_format=self.target._format)
3444
 
        f = RepoFetcher(to_repository=self.target,
 
3420
        f = _mod_fetch.RepoFetcher(to_repository=self.target,
3445
3421
                               from_repository=self.source,
3446
3422
                               last_revision=revision_id,
3447
3423
                               fetch_spec=fetch_spec,
4025
4001
        """See InterRepository.fetch()."""
4026
4002
        if fetch_spec is not None:
4027
4003
            raise AssertionError("Not implemented yet...")
 
4004
        # See <https://launchpad.net/bugs/456077> asking for a warning here
 
4005
        #
 
4006
        # nb this is only active for local-local fetches; other things using
 
4007
        # streaming.
 
4008
        ui.ui_factory.warn_cross_format_fetch(self.source._format,
 
4009
            self.target._format)
4028
4010
        ui.ui_factory.warn_experimental_format_fetch(self)
4029
4011
        if (not self.source.supports_rich_root()
4030
4012
            and self.target.supports_rich_root()):
4032
4014
            self._revision_id_to_root_id = {}
4033
4015
        else:
4034
4016
            self._converting_to_rich_root = False
4035
 
        # See <https://launchpad.net/bugs/456077> asking for a warning here
4036
 
        if self.source._format.network_name() != self.target._format.network_name():
4037
 
            ui.ui_factory.show_user_warning('cross_format_fetch',
4038
 
                from_format=self.source._format,
4039
 
                to_format=self.target._format)
4040
4017
        revision_ids = self.target.search_missing_revision_ids(self.source,
4041
4018
            revision_id, find_ghosts=find_ghosts).get_keys()
4042
4019
        if not revision_ids:
4283
4260
                is_resume = False
4284
4261
            try:
4285
4262
                # locked_insert_stream performs a commit|suspend.
4286
 
                return self._locked_insert_stream(stream, src_format,
4287
 
                    is_resume)
 
4263
                return self._locked_insert_stream(stream, src_format, is_resume)
4288
4264
            except:
4289
4265
                self.target_repo.abort_write_group(suppress_errors=True)
4290
4266
                raise
4326
4302
                    self._extract_and_insert_inventories(
4327
4303
                        substream, src_serializer)
4328
4304
            elif substream_type == 'inventory-deltas':
 
4305
                ui.ui_factory.warn_cross_format_fetch(src_format,
 
4306
                    self.target_repo._format)
4329
4307
                self._extract_and_insert_inventory_deltas(
4330
4308
                    substream, src_serializer)
4331
4309
            elif substream_type == 'chk_bytes':
4337
4315
                # required if the serializers are different only in terms of
4338
4316
                # the inventory.
4339
4317
                if src_serializer == to_serializer:
4340
 
                    self.target_repo.revisions.insert_record_stream(substream)
 
4318
                    self.target_repo.revisions.insert_record_stream(
 
4319
                        substream)
4341
4320
                else:
4342
4321
                    self._extract_and_insert_revisions(substream,
4343
4322
                        src_serializer)
4451
4430
        """Create a StreamSource streaming from from_repository."""
4452
4431
        self.from_repository = from_repository
4453
4432
        self.to_format = to_format
4454
 
        self._record_counter = RecordCounter()
4455
4433
 
4456
4434
    def delta_on_metadata(self):
4457
4435
        """Return True if delta's are permitted on metadata streams.