~maria-captains/bzr-gtk/serg

« back to all changes in this revision

Viewing changes to commit.py

  • Committer: Sergei Golubchik
  • Date: 2014-05-05 15:48:54 UTC
  • Revision ID: sergii@pisem.net-20140505154854-ba4w4r97otnroxkn
cherry-pick http://bazaar.launchpad.net/~bzr-gtk/bzr-gtk/gtk3/revision/777

Avoid the use of Repository.get_ancestry().

Show diffs side-by-side

added added

removed removed

Lines of Context:
29
29
from bzrlib import (
30
30
    errors,
31
31
    osutils,
 
32
    revision as _mod_revision,
32
33
    trace,
 
34
    tsort,
33
35
    )
34
36
try:
35
37
    from bzrlib import bencode
48
50
    have_dbus = False
49
51
 
50
52
 
 
53
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
 
54
    """Get an iterator which will return the revisions in merge sorted order.
 
55
 
 
56
    This will build up a list of all nodes, such that only nodes in the list
 
57
    are referenced. It then uses MergeSorter to return them in 'merge-sorted'
 
58
    order.
 
59
 
 
60
    :param revision_ids: A set of revision_ids
 
61
    :param parent_map: The parent information for each node. Revisions which
 
62
        are considered ghosts should not be present in the map.
 
63
    :return: iterator from MergeSorter.iter_topo_order()
 
64
    """
 
65
    # MergeSorter requires that all nodes be present in the graph, so get rid
 
66
    # of any references pointing outside of this graph.
 
67
    parent_graph = {}
 
68
    for revision_id in revision_ids:
 
69
        if revision_id not in parent_map: # ghost
 
70
            parent_graph[revision_id] = []
 
71
        else:
 
72
            # Only include parents which are in this sub-graph
 
73
            parent_graph[revision_id] = [p for p in parent_map[revision_id]
 
74
                                            if p in revision_ids]
 
75
    sorter = tsort.MergeSorter(parent_graph, tip_revision)
 
76
    return sorter.iter_topo_order()
 
77
 
 
78
 
51
79
def pending_revisions(wt):
52
80
    """Return a list of pending merges or None if there are none of them.
53
81
 
58
86
    """
59
87
    parents = wt.get_parent_ids()
60
88
    if len(parents) < 2:
61
 
        return None
 
89
        return
62
90
 
63
91
    # The basic pending merge algorithm uses the same algorithm as
64
92
    # bzrlib.status.show_pending_merges
66
94
    branch = wt.branch
67
95
    last_revision = parents[0]
68
96
 
69
 
    if last_revision is not None:
70
 
        graph = branch.repository.get_graph()
71
 
        ignore = set([r for r,ps in graph.iter_ancestry([last_revision])])
72
 
    else:
73
 
        ignore = set([])
 
97
    graph = branch.repository.get_graph()
 
98
    other_revisions = [last_revision]
74
99
 
75
100
    pm = []
76
101
    for merge in pending:
77
 
        ignore.add(merge)
78
 
        try:
79
 
            rev = branch.repository.get_revision(merge)
80
 
            children = []
81
 
            pm.append((rev, children))
82
 
 
83
 
            # This does need to be topo sorted, so we search backwards
84
 
            inner_merges = branch.repository.get_ancestry(merge)
85
 
            assert inner_merges[0] is None
86
 
            inner_merges.pop(0)
87
 
            for mmerge in reversed(inner_merges):
88
 
                if mmerge in ignore:
89
 
                    continue
90
 
                rev = branch.repository.get_revision(mmerge)
91
 
                children.append(rev)
92
 
 
93
 
                ignore.add(mmerge)
94
 
        except errors.NoSuchRevision:
95
 
            print "DEBUG: NoSuchRevision:", merge
96
 
 
97
 
    return pm
 
102
        try:
 
103
            merge_rev = branch.repository.get_revision(merge)
 
104
        except errors.NoSuchRevision:
 
105
            # If we are missing a revision, just print out the revision id
 
106
            trace.mutter("ghost: %r", merge)
 
107
            other_revisions.append(merge)
 
108
            continue
 
109
 
 
110
        # Find all of the revisions in the merge source, which are not in the
 
111
        # last committed revision.
 
112
        merge_extra = graph.find_unique_ancestors(merge, other_revisions)
 
113
        other_revisions.append(merge)
 
114
        merge_extra.discard(_mod_revision.NULL_REVISION)
 
115
 
 
116
        # Get a handle to all of the revisions we will need
 
117
        try:
 
118
            revisions = dict((rev.revision_id, rev) for rev in
 
119
                             branch.repository.get_revisions(merge_extra))
 
120
        except errors.NoSuchRevision:
 
121
            # One of the sub nodes is a ghost, check each one
 
122
            revisions = {}
 
123
            for revision_id in merge_extra:
 
124
                try:
 
125
                    rev = branch.repository.get_revisions([revision_id])[0]
 
126
                except errors.NoSuchRevision:
 
127
                    revisions[revision_id] = None
 
128
                else:
 
129
                    revisions[revision_id] = rev
 
130
 
 
131
         # Display the revisions brought in by this merge.
 
132
        rev_id_iterator = _get_sorted_revisions(merge, merge_extra,
 
133
                            branch.repository.get_parent_map(merge_extra))
 
134
        # Skip the first node
 
135
        num, first, depth, eom = rev_id_iterator.next()
 
136
        if first != merge:
 
137
            raise AssertionError('Somehow we misunderstood how'
 
138
                ' iter_topo_order works %s != %s' % (first, merge))
 
139
        children = []
 
140
        for num, sub_merge, depth, eom in rev_id_iterator:
 
141
            rev = revisions[sub_merge]
 
142
            if rev is None:
 
143
                trace.warning("ghost: %r", sub_merge)
 
144
                continue
 
145
            children.append(rev)
 
146
        yield (merge_rev, children)
98
147
 
99
148
 
100
149
_newline_variants_re = re.compile(r'\r\n?')
135
184
        self._delta = None
136
185
        self._wt.lock_read()
137
186
        try:
138
 
            self._pending = pending_revisions(self._wt)
 
187
            self._pending = list(pending_revisions(self._wt))
139
188
        finally:
140
189
            self._wt.unlock()
141
190
 
255
304
                trace.mutter("networkmanager not available.")
256
305
                self._check_local.show()
257
306
                return
258
 
            
 
307
 
259
308
            dbus_iface = dbus.Interface(proxy_obj,
260
309
                                        'org.freedesktop.NetworkManager')
261
310
            try: