/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/revision.py

  • Committer: Martin Pool
  • Date: 2008-05-08 04:12:06 UTC
  • mto: This revision was merged to the branch mainline in revision 3415.
  • Revision ID: mbp@sourcefrog.net-20080508041206-tkrr8ucmcyrlzkum
Some review cleanups for assertion removal

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
# TODO: Some kind of command-line display of revision properties: 
 
18
# perhaps show them in log -v and allow them as options to the commit command.
 
19
 
 
20
 
 
21
from bzrlib import (
 
22
    errors,
 
23
    symbol_versioning,
 
24
    )
 
25
from bzrlib.deprecated_graph import (
 
26
    all_descendants,
 
27
    Graph,
 
28
    node_distances,
 
29
    select_farthest,
 
30
    )
 
31
from bzrlib.osutils import contains_whitespace
 
32
from bzrlib.progress import DummyProgress
 
33
from bzrlib.symbol_versioning import (deprecated_function,
 
34
        )
 
35
 
 
36
NULL_REVISION="null:"
 
37
CURRENT_REVISION="current:"
 
38
 
 
39
 
 
40
class Revision(object):
 
41
    """Single revision on a branch.
 
42
 
 
43
    Revisions may know their revision_hash, but only once they've been
 
44
    written out.  This is not stored because you cannot write the hash
 
45
    into the file it describes.
 
46
 
 
47
    After bzr 0.0.5 revisions are allowed to have multiple parents.
 
48
 
 
49
    parent_ids
 
50
        List of parent revision_ids
 
51
 
 
52
    properties
 
53
        Dictionary of revision properties.  These are attached to the
 
54
        revision as extra metadata.  The name must be a single 
 
55
        word; the value can be an arbitrary string.
 
56
    """
 
57
    
 
58
    def __init__(self, revision_id, properties=None, **args):
 
59
        self.revision_id = revision_id
 
60
        self.properties = properties or {}
 
61
        self._check_properties()
 
62
        self.parent_ids = []
 
63
        self.parent_sha1s = []
 
64
        """Not used anymore - legacy from for 4."""
 
65
        self.__dict__.update(args)
 
66
 
 
67
    def __repr__(self):
 
68
        return "<Revision id %s>" % self.revision_id
 
69
 
 
70
    def __eq__(self, other):
 
71
        if not isinstance(other, Revision):
 
72
            return False
 
73
        # FIXME: rbc 20050930 parent_ids are not being compared
 
74
        return (
 
75
                self.inventory_sha1 == other.inventory_sha1
 
76
                and self.revision_id == other.revision_id
 
77
                and self.timestamp == other.timestamp
 
78
                and self.message == other.message
 
79
                and self.timezone == other.timezone
 
80
                and self.committer == other.committer
 
81
                and self.properties == other.properties)
 
82
 
 
83
    def __ne__(self, other):
 
84
        return not self.__eq__(other)
 
85
 
 
86
    def _check_properties(self):
 
87
        """Verify that all revision properties are OK."""
 
88
        for name, value in self.properties.iteritems():
 
89
            if not isinstance(name, basestring) or contains_whitespace(name):
 
90
                raise ValueError("invalid property name %r" % name)
 
91
            if not isinstance(value, basestring):
 
92
                raise ValueError("invalid property value %r for %r" % 
 
93
                                 (name, value))
 
94
 
 
95
    def get_history(self, repository):
 
96
        """Return the canonical line-of-history for this revision.
 
97
 
 
98
        If ghosts are present this may differ in result from a ghost-free
 
99
        repository.
 
100
        """
 
101
        current_revision = self
 
102
        reversed_result = []
 
103
        while current_revision is not None:
 
104
            reversed_result.append(current_revision.revision_id)
 
105
            if not len (current_revision.parent_ids):
 
106
                reversed_result.append(None)
 
107
                current_revision = None
 
108
            else:
 
109
                next_revision_id = current_revision.parent_ids[0]
 
110
                current_revision = repository.get_revision(next_revision_id)
 
111
        reversed_result.reverse()
 
112
        return reversed_result
 
113
 
 
114
    def get_summary(self):
 
115
        """Get the first line of the log message for this revision.
 
116
        """
 
117
        return self.message.lstrip().split('\n', 1)[0]
 
118
 
 
119
    def get_apparent_author(self):
 
120
        """Return the apparent author of this revision.
 
121
 
 
122
        If the revision properties contain the author name,
 
123
        return it. Otherwise return the committer name.
 
124
        """
 
125
        return self.properties.get('author', self.committer)
 
126
 
 
127
 
 
128
@deprecated_function(symbol_versioning.one_zero)
 
129
def is_ancestor(revision_id, candidate_id, branch):
 
130
    """Return true if candidate_id is an ancestor of revision_id.
 
131
 
 
132
    A false negative will be returned if any intermediate descendent of
 
133
    candidate_id is not present in any of the revision_sources.
 
134
    
 
135
    revisions_source is an object supporting a get_revision operation that
 
136
    behaves like Branch's.
 
137
 
 
138
    This function is deprecated, it is better for callers to directly use
 
139
    Graph.is_ancestor() (just watch out that the parameter order is switched)
 
140
    """
 
141
    return branch.repository.get_graph().is_ancestor(candidate_id, revision_id)
 
142
 
 
143
 
 
144
def iter_ancestors(revision_id, revision_source, only_present=False):
 
145
    ancestors = (revision_id,)
 
146
    distance = 0
 
147
    while len(ancestors) > 0:
 
148
        new_ancestors = []
 
149
        for ancestor in ancestors:
 
150
            if not only_present:
 
151
                yield ancestor, distance
 
152
            try:
 
153
                revision = revision_source.get_revision(ancestor)
 
154
            except errors.NoSuchRevision, e:
 
155
                if e.revision == revision_id:
 
156
                    raise 
 
157
                else:
 
158
                    continue
 
159
            if only_present:
 
160
                yield ancestor, distance
 
161
            new_ancestors.extend(revision.parent_ids)
 
162
        ancestors = new_ancestors
 
163
        distance += 1
 
164
 
 
165
 
 
166
def find_present_ancestors(revision_id, revision_source):
 
167
    """Return the ancestors of a revision present in a branch.
 
168
 
 
169
    It's possible that a branch won't have the complete ancestry of
 
170
    one of its revisions.  
 
171
 
 
172
    """
 
173
    found_ancestors = {}
 
174
    anc_iter = enumerate(iter_ancestors(revision_id, revision_source,
 
175
                         only_present=True))
 
176
    for anc_order, (anc_id, anc_distance) in anc_iter:
 
177
        if anc_id not in found_ancestors:
 
178
            found_ancestors[anc_id] = (anc_order, anc_distance)
 
179
    return found_ancestors
 
180
    
 
181
 
 
182
def __get_closest(intersection):
 
183
    intersection.sort()
 
184
    matches = [] 
 
185
    for entry in intersection:
 
186
        if entry[0] == intersection[0][0]:
 
187
            matches.append(entry[2])
 
188
    return matches
 
189
 
 
190
 
 
191
@deprecated_function(symbol_versioning.one_four)
 
192
def revision_graph(revision, revision_source):
 
193
    """Produce a graph of the ancestry of the specified revision.
 
194
    
 
195
    :return: root, ancestors map, descendants map
 
196
    """
 
197
    revision_source.lock_read()
 
198
    try:
 
199
        return _revision_graph(revision, revision_source)
 
200
    finally:
 
201
        revision_source.unlock()
 
202
 
 
203
 
 
204
def _revision_graph(revision, revision_source):
 
205
    """See revision_graph."""
 
206
    from bzrlib.tsort import topo_sort
 
207
    graph = revision_source.get_revision_graph(revision)
 
208
    # mark all no-parent revisions as being NULL_REVISION parentage.
 
209
    for node, parents in graph.items():
 
210
        if len(parents) == 0:
 
211
            graph[node] = [NULL_REVISION]
 
212
    # add NULL_REVISION to the graph
 
213
    graph[NULL_REVISION] = []
 
214
 
 
215
    # pick a root. If there are multiple roots
 
216
    # this could pick a random one.
 
217
    topo_order = topo_sort(graph.items())
 
218
    root = topo_order[0]
 
219
 
 
220
    ancestors = {}
 
221
    descendants = {}
 
222
 
 
223
    # map the descendants of the graph.
 
224
    # and setup our set based return graph.
 
225
    for node in graph.keys():
 
226
        descendants[node] = {}
 
227
    for node, parents in graph.items():
 
228
        for parent in parents:
 
229
            descendants[parent][node] = 1
 
230
        ancestors[node] = set(parents)
 
231
    return root, ancestors, descendants
 
232
 
 
233
 
 
234
@deprecated_function(symbol_versioning.one_three)
 
235
def combined_graph(revision_a, revision_b, revision_source):
 
236
    """Produce a combined ancestry graph.
 
237
    Return graph root, ancestors map, descendants map, set of common nodes"""
 
238
    root, ancestors, descendants = revision_graph(
 
239
        revision_a, revision_source)
 
240
    root_b, ancestors_b, descendants_b = revision_graph(
 
241
        revision_b, revision_source)
 
242
    if root != root_b:
 
243
        raise errors.NoCommonRoot(revision_a, revision_b)
 
244
    common = set()
 
245
    for node, node_anc in ancestors_b.iteritems():
 
246
        if node in ancestors:
 
247
            common.add(node)
 
248
        else:
 
249
            ancestors[node] = set()
 
250
        ancestors[node].update(node_anc)
 
251
    for node, node_dec in descendants_b.iteritems():
 
252
        if node not in descendants:
 
253
            descendants[node] = {}
 
254
        descendants[node].update(node_dec)
 
255
    return root, ancestors, descendants, common
 
256
 
 
257
 
 
258
@deprecated_function(symbol_versioning.one_three)
 
259
def common_ancestor(revision_a, revision_b, revision_source, 
 
260
                    pb=DummyProgress()):
 
261
    if None in (revision_a, revision_b):
 
262
        return None
 
263
    if NULL_REVISION in (revision_a, revision_b):
 
264
        return NULL_REVISION
 
265
    # trivial optimisation
 
266
    if revision_a == revision_b:
 
267
        return revision_a
 
268
    try:
 
269
        try:
 
270
            pb.update('Picking ancestor', 1, 3)
 
271
            graph = revision_source.get_revision_graph_with_ghosts(
 
272
                [revision_a, revision_b])
 
273
            # Shortcut the case where one of the tips is already included in
 
274
            # the other graphs ancestry.
 
275
            ancestry_a = graph.get_ancestry(revision_a, topo_sorted=False)
 
276
            if revision_b in ancestry_a:
 
277
                return revision_b
 
278
            ancestry_b = graph.get_ancestry(revision_b, topo_sorted=False)
 
279
            if revision_a in ancestry_b:
 
280
                return revision_a
 
281
            # convert to a NULL_REVISION based graph.
 
282
            ancestors = graph.get_ancestors()
 
283
            descendants = graph.get_descendants()
 
284
            common = set(ancestry_a)
 
285
            common.intersection_update(ancestry_b)
 
286
            descendants[NULL_REVISION] = {}
 
287
            ancestors[NULL_REVISION] = []
 
288
            for root in graph.roots:
 
289
                descendants[NULL_REVISION][root] = 1
 
290
                ancestors[root].append(NULL_REVISION)
 
291
            for ghost in graph.ghosts:
 
292
                # ghosts act as roots for the purpose of finding 
 
293
                # the longest paths from the root: any ghost *might*
 
294
                # be directly attached to the root, so we treat them
 
295
                # as being such.
 
296
                # ghost now descends from NULL
 
297
                descendants[NULL_REVISION][ghost] = 1
 
298
                # that is it has an ancestor of NULL
 
299
                ancestors[ghost] = [NULL_REVISION]
 
300
                # ghost is common if any of ghosts descendants are common:
 
301
                for ghost_descendant in descendants[ghost]:
 
302
                    if ghost_descendant in common:
 
303
                        common.add(ghost)
 
304
                
 
305
            root = NULL_REVISION
 
306
            common.add(NULL_REVISION)
 
307
        except errors.NoCommonRoot:
 
308
            raise errors.NoCommonAncestor(revision_a, revision_b)
 
309
            
 
310
        pb.update('Picking ancestor', 2, 3)
 
311
        distances = node_distances (descendants, ancestors, root)
 
312
        pb.update('Picking ancestor', 3, 2)
 
313
        farthest = select_farthest(distances, common)
 
314
        if farthest is None or farthest == NULL_REVISION:
 
315
            raise errors.NoCommonAncestor(revision_a, revision_b)
 
316
    finally:
 
317
        pb.clear()
 
318
    return farthest
 
319
 
 
320
 
 
321
class MultipleRevisionSources(object):
 
322
    """Proxy that looks in multiple branches for revisions."""
 
323
 
 
324
    @symbol_versioning.deprecated_method(symbol_versioning.one_three)
 
325
    def __init__(self, *args):
 
326
        object.__init__(self)
 
327
        self._revision_sources = args
 
328
 
 
329
    def revision_parents(self, revision_id):
 
330
        for source in self._revision_sources:
 
331
            try:
 
332
                return source.revision_parents(revision_id)
 
333
            except (errors.WeaveRevisionNotPresent, errors.NoSuchRevision), e:
 
334
                pass
 
335
        raise e
 
336
 
 
337
    def get_revision(self, revision_id):
 
338
        for source in self._revision_sources:
 
339
            try:
 
340
                return source.get_revision(revision_id)
 
341
            except errors.NoSuchRevision, e:
 
342
                pass
 
343
        raise e
 
344
 
 
345
    def get_revision_graph(self, revision_id):
 
346
        # we could probe incrementally until the pending
 
347
        # ghosts list stop growing, but its cheaper for now
 
348
        # to just ask for the complete graph for each repository.
 
349
        graphs = []
 
350
        for source in self._revision_sources:
 
351
            ghost_graph = source.get_revision_graph_with_ghosts()
 
352
            graphs.append(ghost_graph)
 
353
        absent = 0
 
354
        for graph in graphs:
 
355
            if not revision_id in graph.get_ancestors():
 
356
                absent += 1
 
357
        if absent == len(graphs):
 
358
            raise errors.NoSuchRevision(self._revision_sources[0], revision_id)
 
359
 
 
360
        # combine the graphs
 
361
        result = {}
 
362
        pending = set([revision_id])
 
363
        def find_parents(node_id):
 
364
            """find the parents for node_id."""
 
365
            for graph in graphs:
 
366
                ancestors = graph.get_ancestors()
 
367
                try:
 
368
                    return ancestors[node_id]
 
369
                except KeyError:
 
370
                    pass
 
371
            raise errors.NoSuchRevision(self._revision_sources[0], node_id)
 
372
        while len(pending):
 
373
            # all the graphs should have identical parent lists
 
374
            node_id = pending.pop()
 
375
            try:
 
376
                result[node_id] = find_parents(node_id)
 
377
                for parent_node in result[node_id]:
 
378
                    if not parent_node in result:
 
379
                        pending.add(parent_node)
 
380
            except errors.NoSuchRevision:
 
381
                # ghost, ignore it.
 
382
                pass
 
383
        return result
 
384
 
 
385
    def get_revision_graph_with_ghosts(self, revision_ids):
 
386
        # query all the sources for their entire graphs 
 
387
        # and then build a combined graph for just
 
388
        # revision_ids.
 
389
        graphs = []
 
390
        for source in self._revision_sources:
 
391
            ghost_graph = source.get_revision_graph_with_ghosts()
 
392
            graphs.append(ghost_graph.get_ancestors())
 
393
        for revision_id in revision_ids:
 
394
            absent = 0
 
395
            for graph in graphs:
 
396
                    if not revision_id in graph:
 
397
                        absent += 1
 
398
            if absent == len(graphs):
 
399
                raise errors.NoSuchRevision(self._revision_sources[0],
 
400
                                            revision_id)
 
401
 
 
402
        # combine the graphs
 
403
        result = Graph()
 
404
        pending = set(revision_ids)
 
405
        done = set()
 
406
        def find_parents(node_id):
 
407
            """find the parents for node_id."""
 
408
            for graph in graphs:
 
409
                try:
 
410
                    return graph[node_id]
 
411
                except KeyError:
 
412
                    pass
 
413
            raise errors.NoSuchRevision(self._revision_sources[0], node_id)
 
414
        while len(pending):
 
415
            # all the graphs should have identical parent lists
 
416
            node_id = pending.pop()
 
417
            try:
 
418
                parents = find_parents(node_id)
 
419
                for parent_node in parents:
 
420
                    # queued or done? 
 
421
                    if (parent_node not in pending and
 
422
                        parent_node not in done):
 
423
                        # no, queue
 
424
                        pending.add(parent_node)
 
425
                result.add_node(node_id, parents)
 
426
                done.add(node_id)
 
427
            except errors.NoSuchRevision:
 
428
                # ghost
 
429
                result.add_ghost(node_id)
 
430
                continue
 
431
        return result
 
432
 
 
433
    def lock_read(self):
 
434
        for source in self._revision_sources:
 
435
            source.lock_read()
 
436
 
 
437
    def unlock(self):
 
438
        for source in self._revision_sources:
 
439
            source.unlock()
 
440
 
 
441
 
 
442
def is_reserved_id(revision_id):
 
443
    """Determine whether a revision id is reserved
 
444
 
 
445
    :return: True if the revision is is reserved, False otherwise
 
446
    """
 
447
    return isinstance(revision_id, basestring) and revision_id.endswith(':')
 
448
 
 
449
 
 
450
def check_not_reserved_id(revision_id):
 
451
    """Raise ReservedId if the supplied revision_id is reserved"""
 
452
    if is_reserved_id(revision_id):
 
453
        raise errors.ReservedId(revision_id)
 
454
 
 
455
 
 
456
def ensure_null(revision_id):
 
457
    """Ensure only NULL_REVISION is used to represent the null revision"""
 
458
    if revision_id is None:
 
459
        symbol_versioning.warn('NULL_REVISION should be used for the null'
 
460
            ' revision instead of None, as of bzr 0.91.',
 
461
            DeprecationWarning, stacklevel=2)
 
462
        return NULL_REVISION
 
463
    else:
 
464
        return revision_id
 
465
 
 
466
 
 
467
def is_null(revision_id):
 
468
    if revision_id is None:
 
469
        symbol_versioning.warn('NULL_REVISION should be used for the null'
 
470
            ' revision instead of None, as of bzr 0.90.',
 
471
            DeprecationWarning, stacklevel=2)
 
472
    return revision_id in (None, NULL_REVISION)