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

  • Committer: Martin
  • Date: 2011-01-26 20:02:52 UTC
  • mfrom: (5633 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5634.
  • Revision ID: gzlist@googlemail.com-20110126200252-s4yy1eywfgomxup7
Merge bzr.dev to add release notes for 2.4

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005-2011 Canonical Ltd
2
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
28
28
from bzrlib.lazy_import import lazy_import
29
29
lazy_import(globals(), """
30
30
from bzrlib import (
 
31
    graph,
31
32
    tsort,
32
33
    versionedfile,
33
34
    )
93
94
        try:
94
95
            pb.update("Finding revisions", 0, 2)
95
96
            search = self._revids_to_fetch()
96
 
            if search is None:
 
97
            mutter('fetching: %s', search)
 
98
            if search.is_empty():
97
99
                return
98
100
            pb.update("Fetching revisions", 1, 2)
99
101
            self._fetch_everything_for_search(search)
125
127
            pb.update("Inserting stream")
126
128
            resume_tokens, missing_keys = self.sink.insert_stream(
127
129
                stream, from_format, [])
128
 
            if self.to_repository._fallback_repositories:
129
 
                missing_keys.update(
130
 
                    self._parent_inventories(search.get_keys()))
131
130
            if missing_keys:
132
131
                pb.update("Missing keys")
133
132
                stream = source.get_stream_for_missing_keys(missing_keys)
151
150
        """Determines the exact revisions needed from self.from_repository to
152
151
        install self._last_revision in self.to_repository.
153
152
 
154
 
        If no revisions need to be fetched, then this just returns None.
 
153
        :returns: A SearchResult of some sort.  (Possibly a
 
154
        PendingAncestryResult, EmptySearchResult, etc.)
155
155
        """
156
 
        if self._fetch_spec is not None:
 
156
        mutter("self._fetch_spec, self._last_revision: %r, %r",
 
157
                self._fetch_spec, self._last_revision)
 
158
        get_search_result = getattr(self._fetch_spec, 'get_search_result', None)
 
159
        if get_search_result is not None:
 
160
            mutter(
 
161
                'resolving fetch_spec into search result: %s', self._fetch_spec)
 
162
            # This is EverythingNotInOther or a similar kind of fetch_spec.
 
163
            # Turn it into a search result.
 
164
            return get_search_result()
 
165
        elif self._fetch_spec is not None:
 
166
            # The fetch spec is already a concrete search result.
157
167
            return self._fetch_spec
158
 
        mutter('fetch up to rev {%s}', self._last_revision)
159
 
        if self._last_revision is NULL_REVISION:
 
168
        elif self._last_revision == NULL_REVISION:
 
169
            # fetch_spec is None + last_revision is null => empty fetch.
160
170
            # explicit limit of no revisions needed
161
 
            return None
162
 
        return self.to_repository.search_missing_revision_ids(
163
 
            self.from_repository, self._last_revision,
164
 
            find_ghosts=self.find_ghosts)
165
 
 
166
 
    def _parent_inventories(self, revision_ids):
167
 
        # Find all the parent revisions referenced by the stream, but
168
 
        # not present in the stream, and make sure we send their
169
 
        # inventories.
170
 
        parent_maps = self.to_repository.get_parent_map(revision_ids)
171
 
        parents = set()
172
 
        map(parents.update, parent_maps.itervalues())
173
 
        parents.discard(NULL_REVISION)
174
 
        parents.difference_update(revision_ids)
175
 
        missing_keys = set(('inventories', rev_id) for rev_id in parents)
176
 
        return missing_keys
 
171
            return graph.EmptySearchResult()
 
172
        elif self._last_revision is not None:
 
173
            return graph.NotInOtherForRevs(self.to_repository,
 
174
                self.from_repository, [self._last_revision],
 
175
                find_ghosts=self.find_ghosts).get_search_result()
 
176
        else: # self._last_revision is None:
 
177
            return graph.EverythingNotInOther(self.to_repository,
 
178
                self.from_repository,
 
179
                find_ghosts=self.find_ghosts).get_search_result()
177
180
 
178
181
 
179
182
class Inter1and2Helper(object):
182
185
    This is for use by fetchers and converters.
183
186
    """
184
187
 
 
188
    # This is a class variable so that the test suite can override it.
 
189
    known_graph_threshold = 100
 
190
 
185
191
    def __init__(self, source):
186
192
        """Constructor.
187
193
 
243
249
        # yet, and are unlikely to in non-rich-root environments anyway.
244
250
        root_id_order.sort(key=operator.itemgetter(0))
245
251
        # Create a record stream containing the roots to create.
246
 
        if len(revs) > 100:
247
 
            # XXX: not covered by tests, should have a flag to always run
248
 
            # this. -- mbp 20100129
249
 
            graph = self.source_repo.get_known_graph_ancestry(revs)
 
252
        if len(revs) > self.known_graph_threshold:
 
253
            graph = self.source.get_known_graph_ancestry(revs)
250
254
        new_roots_stream = _new_root_data_stream(
251
255
            root_id_order, rev_id_to_root_id, parent_map, self.source, graph)
252
256
        return [('texts', new_roots_stream)]
253
257
 
254
258
 
255
 
def _get_rich_root_heads_graph(source_repo, revision_ids):
256
 
    """Get a Graph object suitable for asking heads() for new rich roots."""
257
 
    return 
258
 
 
259
 
 
260
259
def _new_root_data_stream(
261
260
    root_keys_to_create, rev_id_to_root_id_map, parent_map, repo, graph=None):
262
261
    """Generate a texts substream of synthesised root entries.