~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repository.py

  • Committer: Patch Queue Manager
  • Date: 2016-02-01 19:13:13 UTC
  • mfrom: (6614.2.2 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20160201191313-wdfvmfff1djde6oq
(vila) Release 2.7.0 (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
 
18
 
17
19
from bzrlib.lazy_import import lazy_import
18
20
lazy_import(globals(), """
19
21
import itertools
20
22
import time
21
23
 
22
24
from bzrlib import (
23
 
    bzrdir,
24
25
    config,
25
26
    controldir,
26
27
    debug,
39
40
""")
40
41
 
41
42
from bzrlib import (
 
43
    bzrdir,
42
44
    errors,
43
45
    registry,
44
46
    symbol_versioning,
78
80
    # being committed to
79
81
    updates_branch = False
80
82
 
81
 
    def __init__(self, repository, parents, config, timestamp=None,
 
83
    def __init__(self, repository, parents, config_stack, timestamp=None,
82
84
                 timezone=None, committer=None, revprops=None,
83
85
                 revision_id=None, lossy=False):
84
86
        """Initiate a CommitBuilder.
93
95
        :param lossy: Whether to discard data that can not be natively
94
96
            represented, when pushing to a foreign VCS 
95
97
        """
96
 
        self._config = config
 
98
        self._config_stack = config_stack
97
99
        self._lossy = lossy
98
100
 
99
101
        if committer is None:
100
 
            self._committer = self._config.username()
 
102
            self._committer = self._config_stack.get('email')
101
103
        elif not isinstance(committer, unicode):
102
104
            self._committer = committer.decode() # throw if non-ascii
103
105
        else:
675
677
    def _resume_write_group(self, tokens):
676
678
        raise errors.UnsuspendableWriteGroup(self)
677
679
 
678
 
    def fetch(self, source, revision_id=None, find_ghosts=False,
679
 
            fetch_spec=None):
 
680
    def fetch(self, source, revision_id=None, find_ghosts=False):
680
681
        """Fetch the content required to construct revision_id from source.
681
682
 
682
 
        If revision_id is None and fetch_spec is None, then all content is
683
 
        copied.
 
683
        If revision_id is None, then all content is copied.
684
684
 
685
685
        fetch() may not be used when the repository is in a write group -
686
686
        either finish the current write group before using fetch, or use
692
692
        :param revision_id: If specified, all the content needed for this
693
693
            revision ID will be copied to the target.  Fetch will determine for
694
694
            itself which content needs to be copied.
695
 
        :param fetch_spec: If specified, a SearchResult or
696
 
            PendingAncestryResult that describes which revisions to copy.  This
697
 
            allows copying multiple heads at once.  Mutually exclusive with
698
 
            revision_id.
699
695
        """
700
 
        if fetch_spec is not None and revision_id is not None:
701
 
            raise AssertionError(
702
 
                "fetch_spec and revision_id are mutually exclusive.")
703
696
        if self.is_in_write_group():
704
697
            raise errors.InternalBzrError(
705
698
                "May not fetch while in a write group.")
707
700
        # TODO: lift out to somewhere common with RemoteRepository
708
701
        # <https://bugs.launchpad.net/bzr/+bug/401646>
709
702
        if (self.has_same_location(source)
710
 
            and fetch_spec is None
711
703
            and self._has_same_fallbacks(source)):
712
704
            # check that last_revision is in 'from' and then return a
713
705
            # no-operation.
716
708
                self.get_revision(revision_id)
717
709
            return 0, []
718
710
        inter = InterRepository.get(source, self)
719
 
        return inter.fetch(revision_id=revision_id,
720
 
            find_ghosts=find_ghosts, fetch_spec=fetch_spec)
 
711
        return inter.fetch(revision_id=revision_id, find_ghosts=find_ghosts)
721
712
 
722
713
    def create_bundle(self, target, base, fileobj, format=None):
723
714
        return serializer.write_bundle(self, target, base, fileobj, format)
724
715
 
725
 
    def get_commit_builder(self, branch, parents, config, timestamp=None,
 
716
    def get_commit_builder(self, branch, parents, config_stack, timestamp=None,
726
717
                           timezone=None, committer=None, revprops=None,
727
718
                           revision_id=None, lossy=False):
728
719
        """Obtain a CommitBuilder for this repository.
729
720
 
730
721
        :param branch: Branch to commit to.
731
722
        :param parents: Revision ids of the parents of the new revision.
732
 
        :param config: Configuration to use.
 
723
        :param config_stack: Configuration stack to use.
733
724
        :param timestamp: Optional timestamp recorded for commit.
734
725
        :param timezone: Optional timezone for timestamp.
735
726
        :param committer: Optional committer to set for commit.
990
981
            raise AssertionError('_iter_for_revno returned too much history')
991
982
        return (True, partial_history[-1])
992
983
 
993
 
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
994
 
    def iter_reverse_revision_history(self, revision_id):
995
 
        """Iterate backwards through revision ids in the lefthand history
996
 
 
997
 
        :param revision_id: The revision id to start with.  All its lefthand
998
 
            ancestors will be traversed.
999
 
        """
1000
 
        graph = self.get_graph()
1001
 
        stop_revisions = (None, _mod_revision.NULL_REVISION)
1002
 
        return graph.iter_lefthand_ancestry(revision_id, stop_revisions)
1003
 
 
1004
984
    def is_shared(self):
1005
985
        """Return True if this repository is flagged as a shared repository."""
1006
986
        raise NotImplementedError(self.is_shared)
1043
1023
        """
1044
1024
        raise NotImplementedError(self.revision_trees)
1045
1025
 
1046
 
    @needs_read_lock
1047
 
    @symbol_versioning.deprecated_method(
1048
 
        symbol_versioning.deprecated_in((2, 4, 0)))
1049
 
    def get_ancestry(self, revision_id, topo_sorted=True):
1050
 
        """Return a list of revision-ids integrated by a revision.
1051
 
 
1052
 
        The first element of the list is always None, indicating the origin
1053
 
        revision.  This might change when we have history horizons, or
1054
 
        perhaps we should have a new API.
1055
 
 
1056
 
        This is topologically sorted.
1057
 
        """
1058
 
        if 'evil' in debug.debug_flags:
1059
 
            mutter_callsite(2, "get_ancestry is linear with history.")
1060
 
        if _mod_revision.is_null(revision_id):
1061
 
            return [None]
1062
 
        if not self.has_revision(revision_id):
1063
 
            raise errors.NoSuchRevision(self, revision_id)
1064
 
        graph = self.get_graph()
1065
 
        keys = set()
1066
 
        search = graph._make_breadth_first_searcher([revision_id])
1067
 
        while True:
1068
 
            try:
1069
 
                found, ghosts = search.next_with_ghosts()
1070
 
            except StopIteration:
1071
 
                break
1072
 
            keys.update(found)
1073
 
        if _mod_revision.NULL_REVISION in keys:
1074
 
            keys.remove(_mod_revision.NULL_REVISION)
1075
 
        if topo_sorted:
1076
 
            parent_map = graph.get_parent_map(keys)
1077
 
            keys = tsort.topo_sort(parent_map)
1078
 
        return [None] + list(keys)
1079
 
 
1080
1026
    def pack(self, hint=None, clean_obsolete_packs=False):
1081
1027
        """Compress the data within the repository.
1082
1028
 
1158
1104
                [parents_provider, other_repository._make_parents_provider()])
1159
1105
        return graph.Graph(parents_provider)
1160
1106
 
1161
 
    def revision_ids_to_search_result(self, result_set):
1162
 
        """Convert a set of revision ids to a graph SearchResult."""
1163
 
        result_parents = set()
1164
 
        for parents in self.get_graph().get_parent_map(
1165
 
            result_set).itervalues():
1166
 
            result_parents.update(parents)
1167
 
        included_keys = result_set.intersection(result_parents)
1168
 
        start_keys = result_set.difference(included_keys)
1169
 
        exclude_keys = result_parents.difference(result_set)
1170
 
        result = graph.SearchResult(start_keys, exclude_keys,
1171
 
            len(result_set), result_set)
1172
 
        return result
1173
 
 
1174
1107
    @needs_write_lock
1175
1108
    def set_make_working_trees(self, new_value):
1176
1109
        """Set the policy flag for making working trees when creating branches.
1196
1129
    @needs_read_lock
1197
1130
    def verify_revision_signature(self, revision_id, gpg_strategy):
1198
1131
        """Verify the signature on a revision.
1199
 
        
 
1132
 
1200
1133
        :param revision_id: the revision to verify
1201
1134
        :gpg_strategy: the GPGStrategy object to used
1202
 
        
 
1135
 
1203
1136
        :return: gpg.SIGNATURE_VALID or a failed SIGNATURE_ value
1204
1137
        """
1205
1138
        if not self.has_signature_for_revision_id(revision_id):
1211
1144
 
1212
1145
        return gpg_strategy.verify(signature, plaintext)
1213
1146
 
 
1147
    @needs_read_lock
 
1148
    def verify_revision_signatures(self, revision_ids, gpg_strategy):
 
1149
        """Verify revision signatures for a number of revisions.
 
1150
 
 
1151
        :param revision_id: the revision to verify
 
1152
        :gpg_strategy: the GPGStrategy object to used
 
1153
        :return: Iterator over tuples with revision id, result and keys
 
1154
        """
 
1155
        for revid in revision_ids:
 
1156
            (result, key) = self.verify_revision_signature(revid, gpg_strategy)
 
1157
            yield revid, result, key
 
1158
 
1214
1159
    def has_signature_for_revision_id(self, revision_id):
1215
1160
        """Query for a revision signature for revision_id in the repository."""
1216
1161
        raise NotImplementedError(self.has_signature_for_revision_id)
1246
1191
            return
1247
1192
        try:
1248
1193
            if branch is None:
1249
 
                conf = config.GlobalConfig()
 
1194
                conf = config.GlobalStack()
1250
1195
            else:
1251
 
                conf = branch.get_config()
1252
 
            if conf.suppress_warning('format_deprecation'):
 
1196
                conf = branch.get_config_stack()
 
1197
            if 'format_deprecation' in conf.get('suppress_warnings'):
1253
1198
                return
1254
1199
            warning("Format %s for %s is deprecated -"
1255
1200
                    " please use 'bzr upgrade' to get better performance"
1315
1260
        """Returns the policy for making working trees on new branches."""
1316
1261
        return not self._transport.has('no-working-trees')
1317
1262
 
 
1263
    @needs_write_lock
 
1264
    def update_feature_flags(self, updated_flags):
 
1265
        """Update the feature flags for this branch.
 
1266
 
 
1267
        :param updated_flags: Dictionary mapping feature names to necessities
 
1268
            A necessity can be None to indicate the feature should be removed
 
1269
        """
 
1270
        self._format._update_feature_flags(updated_flags)
 
1271
        self.control_transport.put_bytes('format', self._format.as_string())
 
1272
 
1318
1273
 
1319
1274
class RepositoryFormatRegistry(controldir.ControlComponentFormatRegistry):
1320
1275
    """Repository format registry."""
1432
1387
    def __ne__(self, other):
1433
1388
        return not self == other
1434
1389
 
1435
 
    @classmethod
1436
 
    def find_format(klass, a_bzrdir):
1437
 
        """Return the format for the repository object in a_bzrdir.
1438
 
 
1439
 
        This is used by bzr native formats that have a "format" file in
1440
 
        the repository.  Other methods may be used by different types of
1441
 
        control directory.
1442
 
        """
1443
 
        try:
1444
 
            transport = a_bzrdir.get_repository_transport(None)
1445
 
            format_string = transport.get_bytes("format")
1446
 
            return format_registry.get(format_string)
1447
 
        except errors.NoSuchFile:
1448
 
            raise errors.NoRepositoryPresent(a_bzrdir)
1449
 
        except KeyError:
1450
 
            raise errors.UnknownFormatError(format=format_string,
1451
 
                                            kind='repository')
1452
 
 
1453
 
    @classmethod
1454
 
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
1455
 
    def register_format(klass, format):
1456
 
        format_registry.register(format)
1457
 
 
1458
 
    @classmethod
1459
 
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
1460
 
    def unregister_format(klass, format):
1461
 
        format_registry.remove(format)
1462
 
 
1463
 
    @classmethod
1464
 
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
1465
 
    def get_default_format(klass):
1466
 
        """Return the current default format."""
1467
 
        return format_registry.get_default()
1468
 
 
1469
 
    def get_format_string(self):
1470
 
        """Return the ASCII format string that identifies this format.
1471
 
 
1472
 
        Note that in pre format ?? repositories the format string is
1473
 
        not permitted nor written to disk.
1474
 
        """
1475
 
        raise NotImplementedError(self.get_format_string)
1476
 
 
1477
1390
    def get_format_description(self):
1478
1391
        """Return the short description for this format."""
1479
1392
        raise NotImplementedError(self.get_format_description)
1545
1458
            hook(params)
1546
1459
 
1547
1460
 
1548
 
class MetaDirRepositoryFormat(RepositoryFormat):
 
1461
class RepositoryFormatMetaDir(bzrdir.BzrFormat, RepositoryFormat):
1549
1462
    """Common base class for the new repositories using the metadir layout."""
1550
1463
 
1551
1464
    rich_root_data = False
1561
1474
        return matching
1562
1475
 
1563
1476
    def __init__(self):
1564
 
        super(MetaDirRepositoryFormat, self).__init__()
 
1477
        RepositoryFormat.__init__(self)
 
1478
        bzrdir.BzrFormat.__init__(self)
1565
1479
 
1566
1480
    def _create_control_files(self, a_bzrdir):
1567
1481
        """Create the required files and the initial control_files object."""
1591
1505
        finally:
1592
1506
            control_files.unlock()
1593
1507
 
1594
 
    def network_name(self):
1595
 
        """Metadir formats have matching disk and network format strings."""
1596
 
        return self.get_format_string()
 
1508
    @classmethod
 
1509
    def find_format(klass, a_bzrdir):
 
1510
        """Return the format for the repository object in a_bzrdir.
 
1511
 
 
1512
        This is used by bzr native formats that have a "format" file in
 
1513
        the repository.  Other methods may be used by different types of
 
1514
        control directory.
 
1515
        """
 
1516
        try:
 
1517
            transport = a_bzrdir.get_repository_transport(None)
 
1518
            format_string = transport.get_bytes("format")
 
1519
        except errors.NoSuchFile:
 
1520
            raise errors.NoRepositoryPresent(a_bzrdir)
 
1521
        return klass._find_format(format_registry, 'repository', format_string)
 
1522
 
 
1523
    def check_support_status(self, allow_unsupported, recommend_upgrade=True,
 
1524
            basedir=None):
 
1525
        RepositoryFormat.check_support_status(self,
 
1526
            allow_unsupported=allow_unsupported, recommend_upgrade=recommend_upgrade,
 
1527
            basedir=basedir)
 
1528
        bzrdir.BzrFormat.check_support_status(self, allow_unsupported=allow_unsupported,
 
1529
            recommend_upgrade=recommend_upgrade, basedir=basedir)
1597
1530
 
1598
1531
 
1599
1532
# formats which have no format string are not discoverable or independently
1716
1649
        self.target.fetch(self.source, revision_id=revision_id)
1717
1650
 
1718
1651
    @needs_write_lock
1719
 
    def fetch(self, revision_id=None, find_ghosts=False,
1720
 
            fetch_spec=None):
 
1652
    def fetch(self, revision_id=None, find_ghosts=False):
1721
1653
        """Fetch the content required to construct revision_id.
1722
1654
 
1723
1655
        The content is copied from self.source to self.target.