← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/fetch.py

(gz) Backslash escape selftest output when printing to non-unicode consoles
 (Martin [gz])

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/fetch.py
1
 
# Copyright (C) 2005, 2006, 2008, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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 as _mod_graph,
32
 
    static_tuple,
33
31
    tsort,
34
32
    versionedfile,
35
33
    )
37
35
import bzrlib
38
36
from bzrlib import (
39
37
    errors,
40
 
    symbol_versioning,
41
38
    ui,
42
39
    )
43
40
from bzrlib.revision import NULL_REVISION
52
49
    """
53
50
 
54
51
    def __init__(self, to_repository, from_repository, last_revision=None,
55
 
        pb=None, find_ghosts=True, fetch_spec=None):
 
52
        find_ghosts=True, fetch_spec=None):
56
53
        """Create a repo fetcher.
57
54
 
58
55
        :param last_revision: If set, try to limit to the data this revision
59
56
            references.
60
57
        :param find_ghosts: If True search the entire history for ghosts.
61
 
        :param pb: ProgressBar object to use; deprecated and ignored.
62
 
            This method will just create one on top of the stack.
63
58
        """
64
 
        if pb is not None:
65
 
            symbol_versioning.warn(
66
 
                symbol_versioning.deprecated_in((1, 14, 0))
67
 
                % "pb parameter to RepoFetcher.__init__")
68
 
            # and for simplicity it is in fact ignored
69
59
        # repository.fetch has the responsibility for short-circuiting
70
60
        # attempts to copy between a repository and itself.
71
61
        self.to_repository = to_repository
192
182
    This is for use by fetchers and converters.
193
183
    """
194
184
 
 
185
    # This is a class variable so that the test suite can override it.
 
186
    known_graph_threshold = 100
 
187
 
195
188
    def __init__(self, source):
196
189
        """Constructor.
197
190
 
253
246
        # yet, and are unlikely to in non-rich-root environments anyway.
254
247
        root_id_order.sort(key=operator.itemgetter(0))
255
248
        # Create a record stream containing the roots to create.
256
 
        if len(revs) > 100:
257
 
            # XXX: not covered by tests, should have a flag to always run
258
 
            # this. -- mbp 20100129
259
 
            graph = _get_rich_root_heads_graph(self.source, revs)
 
249
        if len(revs) > self.known_graph_threshold:
 
250
            graph = self.source.get_known_graph_ancestry(revs)
260
251
        new_roots_stream = _new_root_data_stream(
261
252
            root_id_order, rev_id_to_root_id, parent_map, self.source, graph)
262
253
        return [('texts', new_roots_stream)]
263
254
 
264
255
 
265
 
def _get_rich_root_heads_graph(source_repo, revision_ids):
266
 
    """Get a Graph object suitable for asking heads() for new rich roots."""
267
 
    st = static_tuple.StaticTuple
268
 
    revision_keys = [st(r_id).intern() for r_id in revision_ids]
269
 
    known_graph = source_repo.revisions.get_known_graph_ancestry(
270
 
                    revision_keys)
271
 
    return _mod_graph.GraphThunkIdsToKeys(known_graph)
272
 
 
273
 
 
274
256
def _new_root_data_stream(
275
257
    root_keys_to_create, rev_id_to_root_id_map, parent_map, repo, graph=None):
276
258
    """Generate a texts substream of synthesised root entries.