~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revision.py

  • Committer: Martin Pool
  • Date: 2005-08-30 01:35:40 UTC
  • Revision ID: mbp@sourcefrog.net-20050830013540-34e8996a86ba25fb
- rename FunctionalTest to TestCaseInTempDir

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
2
 
#
 
1
# (C) 2005 Canonical
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
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.lazy_import import lazy_import
22
 
lazy_import(globals(), """
23
 
from bzrlib import deprecated_graph
24
 
""")
25
 
from bzrlib import (
26
 
    errors,
27
 
    symbol_versioning,
28
 
    )
29
 
from bzrlib.osutils import contains_whitespace
30
 
from bzrlib.progress import DummyProgress
31
 
 
32
 
NULL_REVISION="null:"
33
 
CURRENT_REVISION="current:"
 
17
 
 
18
import bzrlib.errors
 
19
 
 
20
 
 
21
class RevisionReference(object):
 
22
    """
 
23
    Reference to a stored revision.
 
24
 
 
25
    Includes the revision_id and revision_sha1.
 
26
    """
 
27
    revision_id = None
 
28
    revision_sha1 = None
 
29
    def __init__(self, revision_id, revision_sha1=None):
 
30
        if revision_id == None \
 
31
           or isinstance(revision_id, basestring):
 
32
            self.revision_id = revision_id
 
33
        else:
 
34
            raise ValueError('bad revision_id %r' % revision_id)
 
35
 
 
36
        if revision_sha1 != None:
 
37
            if isinstance(revision_sha1, basestring) \
 
38
               and len(revision_sha1) == 40:
 
39
                self.revision_sha1 = revision_sha1
 
40
            else:
 
41
                raise ValueError('bad revision_sha1 %r' % revision_sha1)
 
42
                
34
43
 
35
44
 
36
45
class Revision(object):
42
51
 
43
52
    After bzr 0.0.5 revisions are allowed to have multiple parents.
44
53
 
45
 
    parent_ids
46
 
        List of parent revision_ids
47
 
 
48
 
    properties
49
 
        Dictionary of revision properties.  These are attached to the
50
 
        revision as extra metadata.  The name must be a single 
51
 
        word; the value can be an arbitrary string.
 
54
    parents
 
55
        List of parent revisions, each is a RevisionReference.
52
56
    """
 
57
    inventory_id = None
 
58
    inventory_sha1 = None
 
59
    revision_id = None
 
60
    timestamp = None
 
61
    message = None
 
62
    timezone = None
 
63
    committer = None
53
64
    
54
 
    def __init__(self, revision_id, properties=None, **args):
55
 
        self.revision_id = revision_id
56
 
        self.properties = properties or {}
57
 
        self._check_properties()
58
 
        self.parent_ids = []
59
 
        self.parent_sha1s = []
60
 
        """Not used anymore - legacy from for 4."""
 
65
    def __init__(self, **args):
61
66
        self.__dict__.update(args)
 
67
        self.parents = []
 
68
 
62
69
 
63
70
    def __repr__(self):
64
71
        return "<Revision id %s>" % self.revision_id
65
72
 
66
 
    def __eq__(self, other):
67
 
        if not isinstance(other, Revision):
68
 
            return False
69
 
        # FIXME: rbc 20050930 parent_ids are not being compared
70
 
        return (
71
 
                self.inventory_sha1 == other.inventory_sha1
72
 
                and self.revision_id == other.revision_id
73
 
                and self.timestamp == other.timestamp
74
 
                and self.message == other.message
75
 
                and self.timezone == other.timezone
76
 
                and self.committer == other.committer
77
 
                and self.properties == other.properties)
78
 
 
79
 
    def __ne__(self, other):
80
 
        return not self.__eq__(other)
81
 
 
82
 
    def _check_properties(self):
83
 
        """Verify that all revision properties are OK."""
84
 
        for name, value in self.properties.iteritems():
85
 
            if not isinstance(name, basestring) or contains_whitespace(name):
86
 
                raise ValueError("invalid property name %r" % name)
87
 
            if not isinstance(value, basestring):
88
 
                raise ValueError("invalid property value %r for %r" % 
89
 
                                 (name, value))
90
 
 
91
 
    def get_history(self, repository):
92
 
        """Return the canonical line-of-history for this revision.
93
 
 
94
 
        If ghosts are present this may differ in result from a ghost-free
95
 
        repository.
96
 
        """
97
 
        current_revision = self
98
 
        reversed_result = []
99
 
        while current_revision is not None:
100
 
            reversed_result.append(current_revision.revision_id)
101
 
            if not len (current_revision.parent_ids):
102
 
                reversed_result.append(None)
103
 
                current_revision = None
104
 
            else:
105
 
                next_revision_id = current_revision.parent_ids[0]
106
 
                current_revision = repository.get_revision(next_revision_id)
107
 
        reversed_result.reverse()
108
 
        return reversed_result
109
 
 
110
 
    def get_summary(self):
111
 
        """Get the first line of the log message for this revision.
112
 
        """
113
 
        return self.message.lstrip().split('\n', 1)[0]
114
 
 
115
 
    def get_apparent_author(self):
116
 
        """Return the apparent author of this revision.
117
 
 
118
 
        If the revision properties contain the author name,
119
 
        return it. Otherwise return the committer name.
120
 
        """
121
 
        return self.properties.get('author', self.committer)
122
 
 
 
73
        
 
74
    def to_element(self):
 
75
        from bzrlib.xml import Element, SubElement
 
76
        
 
77
        root = Element('revision',
 
78
                       committer = self.committer,
 
79
                       timestamp = '%.9f' % self.timestamp,
 
80
                       revision_id = self.revision_id,
 
81
                       inventory_id = self.inventory_id,
 
82
                       inventory_sha1 = self.inventory_sha1,
 
83
                       )
 
84
        if self.timezone:
 
85
            root.set('timezone', str(self.timezone))
 
86
        root.text = '\n'
 
87
        
 
88
        msg = SubElement(root, 'message')
 
89
        msg.text = self.message
 
90
        msg.tail = '\n'
 
91
 
 
92
        if self.parents:
 
93
            pelts = SubElement(root, 'parents')
 
94
            pelts.tail = pelts.text = '\n'
 
95
            for rr in self.parents:
 
96
                assert isinstance(rr, RevisionReference)
 
97
                p = SubElement(pelts, 'revision_ref')
 
98
                p.tail = '\n'
 
99
                assert rr.revision_id
 
100
                p.set('revision_id', rr.revision_id)
 
101
                if rr.revision_sha1:
 
102
                    p.set('revision_sha1', rr.revision_sha1)
 
103
 
 
104
        return root
 
105
 
 
106
 
 
107
    def from_element(cls, elt):
 
108
        return unpack_revision(elt)
 
109
 
 
110
    from_element = classmethod(from_element)
 
111
 
 
112
 
 
113
 
 
114
def unpack_revision(elt):
 
115
    """Convert XML element into Revision object."""
 
116
    # <changeset> is deprecated...
 
117
    if elt.tag not in ('revision', 'changeset'):
 
118
        raise bzrlib.errors.BzrError("unexpected tag in revision file: %r" % elt)
 
119
 
 
120
    rev = Revision(committer = elt.get('committer'),
 
121
                   timestamp = float(elt.get('timestamp')),
 
122
                   revision_id = elt.get('revision_id'),
 
123
                   inventory_id = elt.get('inventory_id'),
 
124
                   inventory_sha1 = elt.get('inventory_sha1')
 
125
                   )
 
126
 
 
127
    precursor = elt.get('precursor')
 
128
    precursor_sha1 = elt.get('precursor_sha1')
 
129
 
 
130
    pelts = elt.find('parents')
 
131
 
 
132
    if pelts:
 
133
        for p in pelts:
 
134
            assert p.tag == 'revision_ref', \
 
135
                   "bad parent node tag %r" % p.tag
 
136
            rev_ref = RevisionReference(p.get('revision_id'),
 
137
                                        p.get('revision_sha1'))
 
138
            rev.parents.append(rev_ref)
 
139
 
 
140
        if precursor:
 
141
            # must be consistent
 
142
            prec_parent = rev.parents[0].revision_id
 
143
            assert prec_parent == precursor
 
144
    elif precursor:
 
145
        # revisions written prior to 0.0.5 have a single precursor
 
146
        # give as an attribute
 
147
        rev_ref = RevisionReference(precursor, precursor_sha1)
 
148
        rev.parents.append(rev_ref)
 
149
 
 
150
    v = elt.get('timezone')
 
151
    rev.timezone = v and int(v)
 
152
 
 
153
    rev.message = elt.findtext('message') # text of <message>
 
154
    return rev
 
155
 
 
156
 
 
157
 
 
158
REVISION_ID_RE = None
 
159
 
 
160
def validate_revision_id(rid):
 
161
    """Check rid is syntactically valid for a revision id."""
 
162
    global REVISION_ID_RE
 
163
    if not REVISION_ID_RE:
 
164
        import re
 
165
        REVISION_ID_RE = re.compile('[\w.-]+@[\w.-]+--?\d+--?[0-9a-f]+\Z')
 
166
 
 
167
    if not REVISION_ID_RE.match(rid):
 
168
        raise ValueError("malformed revision-id %r" % rid)
 
169
 
 
170
def is_ancestor(revision_id, candidate_id, revision_source):
 
171
    """Return true if candidate_id is an ancestor of revision_id.
 
172
    A false negative will be returned if any intermediate descendent of
 
173
    candidate_id is not present in any of the revision_sources.
 
174
    
 
175
    revisions_source is an object supporting a get_revision operation that
 
176
    behaves like Branch's.
 
177
    """
 
178
 
 
179
    for ancestor_id, distance in iter_ancestors(revision_id, revision_source):
 
180
        if ancestor_id == candidate_id:
 
181
            return True
 
182
    return False
123
183
 
124
184
def iter_ancestors(revision_id, revision_source, only_present=False):
125
185
    ancestors = (revision_id,)
131
191
                yield ancestor, distance
132
192
            try:
133
193
                revision = revision_source.get_revision(ancestor)
134
 
            except errors.NoSuchRevision, e:
 
194
            except bzrlib.errors.NoSuchRevision, e:
135
195
                if e.revision == revision_id:
136
196
                    raise 
137
197
                else:
138
198
                    continue
139
199
            if only_present:
140
200
                yield ancestor, distance
141
 
            new_ancestors.extend(revision.parent_ids)
 
201
            new_ancestors.extend([p.revision_id for p in revision.parents])
142
202
        ancestors = new_ancestors
143
203
        distance += 1
144
204
 
151
211
 
152
212
    """
153
213
    found_ancestors = {}
 
214
    count = 0
154
215
    anc_iter = enumerate(iter_ancestors(revision_id, revision_source,
155
216
                         only_present=True))
156
217
    for anc_order, (anc_id, anc_distance) in anc_iter:
157
 
        if anc_id not in found_ancestors:
 
218
        if not found_ancestors.has_key(anc_id):
158
219
            found_ancestors[anc_id] = (anc_order, anc_distance)
159
220
    return found_ancestors
160
221
    
161
 
 
162
 
def __get_closest(intersection):
163
 
    intersection.sort()
164
 
    matches = [] 
165
 
    for entry in intersection:
166
 
        if entry[0] == intersection[0][0]:
167
 
            matches.append(entry[2])
168
 
    return matches
169
 
 
170
 
 
171
 
def is_reserved_id(revision_id):
172
 
    """Determine whether a revision id is reserved
173
 
 
174
 
    :return: True if the revision is is reserved, False otherwise
 
222
class AmbiguousBase(bzrlib.errors.BzrError):
 
223
    def __init__(self, bases):
 
224
        msg = "The correct base is unclear, becase %s are all equally close" %\
 
225
            ", ".join(bases)
 
226
        bzrlib.errors.BzrError.__init__(self, msg)
 
227
        self.bases = bases
 
228
 
 
229
def common_ancestor(revision_a, revision_b, revision_source):
 
230
    """Find the ancestor common to both revisions that is closest to both.
175
231
    """
176
 
    return isinstance(revision_id, basestring) and revision_id.endswith(':')
177
 
 
178
 
 
179
 
def check_not_reserved_id(revision_id):
180
 
    """Raise ReservedId if the supplied revision_id is reserved"""
181
 
    if is_reserved_id(revision_id):
182
 
        raise errors.ReservedId(revision_id)
183
 
 
184
 
 
185
 
def ensure_null(revision_id):
186
 
    """Ensure only NULL_REVISION is used to represent the null revision"""
187
 
    if revision_id is None:
188
 
        symbol_versioning.warn('NULL_REVISION should be used for the null'
189
 
            ' revision instead of None, as of bzr 0.91.',
190
 
            DeprecationWarning, stacklevel=2)
191
 
        return NULL_REVISION
 
232
    from bzrlib.trace import mutter
 
233
    a_ancestors = find_present_ancestors(revision_a, revision_source)
 
234
    b_ancestors = find_present_ancestors(revision_b, revision_source)
 
235
    a_intersection = []
 
236
    b_intersection = []
 
237
    # a_order is used as a tie-breaker when two equally-good bases are found
 
238
    for revision, (a_order, a_distance) in a_ancestors.iteritems():
 
239
        if b_ancestors.has_key(revision):
 
240
            a_intersection.append((a_distance, a_order, revision))
 
241
            b_intersection.append((b_ancestors[revision][1], a_order, revision))
 
242
    mutter("a intersection: %r" % a_intersection)
 
243
    mutter("b intersection: %r" % b_intersection)
 
244
    def get_closest(intersection):
 
245
        intersection.sort()
 
246
        matches = [] 
 
247
        for entry in intersection:
 
248
            if entry[0] == intersection[0][0]:
 
249
                matches.append(entry[2])
 
250
        return matches
 
251
 
 
252
    a_closest = get_closest(a_intersection)
 
253
    if len(a_closest) == 0:
 
254
        return None
 
255
    b_closest = get_closest(b_intersection)
 
256
    assert len(b_closest) != 0
 
257
    mutter ("a_closest %r" % a_closest)
 
258
    mutter ("b_closest %r" % b_closest)
 
259
    if a_closest[0] in b_closest:
 
260
        return a_closest[0]
 
261
    elif b_closest[0] in a_closest:
 
262
        return b_closest[0]
192
263
    else:
193
 
        return revision_id
194
 
 
195
 
 
196
 
def is_null(revision_id):
197
 
    if revision_id is None:
198
 
        symbol_versioning.warn('NULL_REVISION should be used for the null'
199
 
            ' revision instead of None, as of bzr 0.90.',
200
 
            DeprecationWarning, stacklevel=2)
201
 
    return revision_id in (None, NULL_REVISION)
 
264
        raise AmbiguousBase((a_closest[0], b_closest[0]))
 
265
    return a_closest[0]
 
266
 
 
267
class MultipleRevisionSources(object):
 
268
    def __init__(self, *args):
 
269
        object.__init__(self)
 
270
        assert len(args) != 0
 
271
        self._revision_sources = args
 
272
 
 
273
    def get_revision(self, revision_id):
 
274
        for source in self._revision_sources:
 
275
            try:
 
276
                return source.get_revision(revision_id)
 
277
            except bzrlib.errors.NoSuchRevision, e:
 
278
                pass
 
279
        raise e