~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/fetch.py

  • Committer: John Arbash Meinel
  • Date: 2008-08-18 22:34:21 UTC
  • mto: (3606.5.6 1.6)
  • mto: This revision was merged to the branch mainline in revision 3641.
  • Revision ID: john@arbash-meinel.com-20080818223421-todjny24vj4faj4t
Add tests for the fetching behavior.

The proper parameter passed is 'unordered' add an assert for it, and
fix callers that were passing 'unsorted' instead.
Add tests that we make the right get_record_stream call based
on the value of _fetch_uses_deltas.
Fix the fetch request for signatures.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 by Canonical Ltd
2
 
 
 
1
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
 
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
27
27
stored, so that if a revision is present we can totally recreate it.
28
28
However, we can't know what files are included in a revision until we
29
29
read its inventory.  So we query the inventory store of the source for
30
 
the ids we need, and then pull those ids and finally actually join
31
 
the inventories.
 
30
the ids we need, and then pull those ids and then return to the inventories.
32
31
"""
33
32
 
 
33
import operator
 
34
 
34
35
import bzrlib
35
36
import bzrlib.errors as errors
36
 
from bzrlib.errors import (InstallFailed, NoSuchRevision,
37
 
                           MissingText)
 
37
from bzrlib.errors import InstallFailed
 
38
from bzrlib.progress import ProgressPhase
 
39
from bzrlib.revision import is_null, NULL_REVISION
 
40
from bzrlib.symbol_versioning import (deprecated_function,
 
41
        deprecated_method,
 
42
        )
 
43
from bzrlib.tsort import topo_sort
38
44
from bzrlib.trace import mutter
39
 
from bzrlib.progress import ProgressBar, ProgressPhase
40
 
from bzrlib.reconcile import RepoReconciler
41
 
from bzrlib.revision import NULL_REVISION
42
 
from bzrlib.symbol_versioning import *
43
 
 
 
45
import bzrlib.ui
 
46
from bzrlib.versionedfile import filter_absent, FulltextContentFactory
44
47
 
45
48
# TODO: Avoid repeatedly opening weaves so many times.
46
49
 
60
63
#   and add in all file versions
61
64
 
62
65
 
63
 
@deprecated_function(zero_eight)
64
 
def greedy_fetch(to_branch, from_branch, revision=None, pb=None):
65
 
    """Legacy API, please see branch.fetch(from_branch, last_revision, pb)."""
66
 
    f = Fetcher(to_branch, from_branch, revision, pb)
67
 
    return f.count_copied, f.failed_revisions
68
 
 
69
 
fetch = greedy_fetch
70
 
 
71
 
 
72
66
class RepoFetcher(object):
73
67
    """Pull revisions and texts from one repository to another.
74
68
 
78
72
    after running:
79
73
    count_copied -- number of revisions copied
80
74
 
81
 
    This should not be used directory, its essential a object to encapsulate
 
75
    This should not be used directly, it's essential a object to encapsulate
82
76
    the logic in InterRepository.fetch().
83
77
    """
84
 
    def __init__(self, to_repository, from_repository, last_revision=None, pb=None):
 
78
 
 
79
    def __init__(self, to_repository, from_repository, last_revision=None, pb=None,
 
80
        find_ghosts=True):
 
81
        """Create a repo fetcher.
 
82
 
 
83
        :param find_ghosts: If True search the entire history for ghosts.
 
84
        """
85
85
        # result variables.
86
86
        self.failed_revisions = []
87
87
        self.count_copied = 0
88
 
        if to_repository.control_files._transport.base == from_repository.control_files._transport.base:
89
 
            # check that last_revision is in 'from' and then return a no-operation.
90
 
            if last_revision not in (None, NULL_REVISION):
91
 
                from_repository.get_revision(last_revision)
92
 
            return
 
88
        if to_repository.has_same_location(from_repository):
 
89
            # repository.fetch should be taking care of this case.
 
90
            raise errors.BzrError('RepoFetcher run '
 
91
                    'between two objects at the same location: '
 
92
                    '%r and %r' % (to_repository, from_repository))
93
93
        self.to_repository = to_repository
94
94
        self.from_repository = from_repository
95
95
        # must not mutate self._last_revision as its potentially a shared instance
96
96
        self._last_revision = last_revision
 
97
        self.find_ghosts = find_ghosts
97
98
        if pb is None:
98
99
            self.pb = bzrlib.ui.ui_factory.nested_progress_bar()
99
100
            self.nested_pb = self.pb
104
105
        try:
105
106
            self.to_repository.lock_write()
106
107
            try:
107
 
                self.__fetch()
 
108
                self.to_repository.start_write_group()
 
109
                try:
 
110
                    self.__fetch()
 
111
                except:
 
112
                    self.to_repository.abort_write_group()
 
113
                    raise
 
114
                else:
 
115
                    self.to_repository.commit_write_group()
108
116
            finally:
109
 
                if self.nested_pb is not None:
110
 
                    self.nested_pb.finished()
111
 
                self.to_repository.unlock()
 
117
                try:
 
118
                    if self.nested_pb is not None:
 
119
                        self.nested_pb.finished()
 
120
                finally:
 
121
                    self.to_repository.unlock()
112
122
        finally:
113
123
            self.from_repository.unlock()
114
124
 
118
128
        This initialises all the needed variables, and then fetches the 
119
129
        requested revisions, finally clearing the progress bar.
120
130
        """
121
 
        self.to_weaves = self.to_repository.weave_store
122
 
        self.to_control = self.to_repository.control_weaves
123
 
        self.from_weaves = self.from_repository.weave_store
124
 
        self.from_control = self.from_repository.control_weaves
125
131
        self.count_total = 0
126
132
        self.file_ids_names = {}
127
 
        pp = ProgressPhase('fetch phase', 4, self.pb)
 
133
        pp = ProgressPhase('Transferring', 4, self.pb)
128
134
        try:
129
 
            revs = self._revids_to_fetch()
130
 
            # something to do ?
131
 
            if revs:
132
 
                pp.next_phase()
133
 
                self._fetch_weave_texts(revs)
134
 
                pp.next_phase()
135
 
                self._fetch_inventory_weave(revs)
136
 
                pp.next_phase()
137
 
                self._fetch_revision_texts(revs)
138
 
                self.count_copied += len(revs)
 
135
            pp.next_phase()
 
136
            search = self._revids_to_fetch()
 
137
            if search is None:
 
138
                return
 
139
            if getattr(self, '_fetch_everything_for_search', None) is not None:
 
140
                self._fetch_everything_for_search(search, pp)
 
141
            else:
 
142
                # backward compatibility
 
143
                self._fetch_everything_for_revisions(search.get_keys, pp)
139
144
        finally:
140
145
            self.pb.clear()
141
146
 
 
147
    def _fetch_everything_for_search(self, search, pp):
 
148
        """Fetch all data for the given set of revisions."""
 
149
        # The first phase is "file".  We pass the progress bar for it directly
 
150
        # into item_keys_introduced_by, which has more information about how
 
151
        # that phase is progressing than we do.  Progress updates for the other
 
152
        # phases are taken care of in this function.
 
153
        # XXX: there should be a clear owner of the progress reporting.  Perhaps
 
154
        # item_keys_introduced_by should have a richer API than it does at the
 
155
        # moment, so that it can feed the progress information back to this
 
156
        # function?
 
157
        phase = 'file'
 
158
        pb = bzrlib.ui.ui_factory.nested_progress_bar()
 
159
        try:
 
160
            revs = search.get_keys()
 
161
            graph = self.from_repository.get_graph()
 
162
            revs = list(graph.iter_topo_order(revs))
 
163
            data_to_fetch = self.from_repository.item_keys_introduced_by(revs,
 
164
                                                                         pb)
 
165
            text_keys = []
 
166
            for knit_kind, file_id, revisions in data_to_fetch:
 
167
                if knit_kind != phase:
 
168
                    phase = knit_kind
 
169
                    # Make a new progress bar for this phase
 
170
                    pb.finished()
 
171
                    pp.next_phase()
 
172
                    pb = bzrlib.ui.ui_factory.nested_progress_bar()
 
173
                if knit_kind == "file":
 
174
                    # Accumulate file texts
 
175
                    text_keys.extend([(file_id, revision) for revision in
 
176
                        revisions])
 
177
                elif knit_kind == "inventory":
 
178
                    # Now copy the file texts.
 
179
                    to_texts = self.to_repository.texts
 
180
                    from_texts = self.from_repository.texts
 
181
                    to_texts.insert_record_stream(from_texts.get_record_stream(
 
182
                        text_keys, self.to_repository._fetch_order,
 
183
                        not self.to_repository._fetch_uses_deltas))
 
184
                    # Cause an error if a text occurs after we have done the
 
185
                    # copy.
 
186
                    text_keys = None
 
187
                    # Before we process the inventory we generate the root
 
188
                    # texts (if necessary) so that the inventories references
 
189
                    # will be valid.
 
190
                    self._generate_root_texts(revs)
 
191
                    # NB: This currently reopens the inventory weave in source;
 
192
                    # using a single stream interface instead would avoid this.
 
193
                    self._fetch_inventory_weave(revs, pb)
 
194
                elif knit_kind == "signatures":
 
195
                    # Nothing to do here; this will be taken care of when
 
196
                    # _fetch_revision_texts happens.
 
197
                    pass
 
198
                elif knit_kind == "revisions":
 
199
                    self._fetch_revision_texts(revs, pb)
 
200
                else:
 
201
                    raise AssertionError("Unknown knit kind %r" % knit_kind)
 
202
            if self.to_repository._fetch_reconcile:
 
203
                self.to_repository.reconcile()
 
204
        finally:
 
205
            if pb is not None:
 
206
                pb.finished()
 
207
        self.count_copied += len(revs)
 
208
        
142
209
    def _revids_to_fetch(self):
 
210
        """Determines the exact revisions needed from self.from_repository to
 
211
        install self._last_revision in self.to_repository.
 
212
 
 
213
        If no revisions need to be fetched, then this just returns None.
 
214
        """
143
215
        mutter('fetch up to rev {%s}', self._last_revision)
144
216
        if self._last_revision is NULL_REVISION:
145
217
            # explicit limit of no revisions needed
146
218
            return None
147
 
        if (self._last_revision != None and
 
219
        if (self._last_revision is not None and
148
220
            self.to_repository.has_revision(self._last_revision)):
149
221
            return None
150
 
            
151
222
        try:
152
 
            return self.to_repository.missing_revision_ids(self.from_repository,
153
 
                                                           self._last_revision)
154
 
        except errors.NoSuchRevision:
 
223
            return self.to_repository.search_missing_revision_ids(
 
224
                self.from_repository, self._last_revision,
 
225
                find_ghosts=self.find_ghosts)
 
226
        except errors.NoSuchRevision, e:
155
227
            raise InstallFailed([self._last_revision])
156
228
 
157
 
    def _fetch_weave_texts(self, revs):
158
 
        texts_pb = bzrlib.ui.ui_factory.nested_progress_bar()
 
229
    def _fetch_inventory_weave(self, revs, pb):
 
230
        pb.update("fetch inventory", 0, 2)
 
231
        to_weave = self.to_repository.inventories
 
232
        child_pb = bzrlib.ui.ui_factory.nested_progress_bar()
159
233
        try:
160
 
            file_ids = self.from_repository.fileids_altered_by_revision_ids(revs)
161
 
            count = 0
162
 
            num_file_ids = len(file_ids)
163
 
            for file_id, required_versions in file_ids.items():
164
 
                texts_pb.update("fetch texts", count, num_file_ids)
165
 
                count +=1
166
 
                to_weave = self.to_weaves.get_weave_or_empty(file_id,
167
 
                    self.to_repository.get_transaction())
168
 
                from_weave = self.from_weaves.get_weave(file_id,
169
 
                    self.from_repository.get_transaction())
170
 
                # we fetch all the texts, because texts do
171
 
                # not reference anything, and its cheap enough
172
 
                to_weave.join(from_weave, version_ids=required_versions) 
173
 
                # we don't need *all* of this data anymore, but we dont know
174
 
                # what we do. This cache clearing will result in a new read 
175
 
                # of the knit data when we do the checkout, but probably we
176
 
                # want to emit the needed data on the fly rather than at the
177
 
                # end anyhow.
178
 
                # the from weave should know not to cache data being joined,
179
 
                # but its ok to ask it to clear.
180
 
                from_weave.clear_cache()
181
 
                to_weave.clear_cache()
 
234
            # just merge, this is optimisable and its means we don't
 
235
            # copy unreferenced data such as not-needed inventories.
 
236
            pb.update("fetch inventory", 1, 3)
 
237
            from_weave = self.from_repository.inventories
 
238
            pb.update("fetch inventory", 2, 3)
 
239
            # we fetch only the referenced inventories because we do not
 
240
            # know for unselected inventories whether all their required
 
241
            # texts are present in the other repository - it could be
 
242
            # corrupt.
 
243
            to_weave.insert_record_stream(from_weave.get_record_stream(
 
244
                [(rev_id,) for rev_id in revs],
 
245
                self.to_repository._fetch_order,
 
246
                not self.to_repository._fetch_uses_deltas))
182
247
        finally:
183
 
            texts_pb.finished()
184
 
 
185
 
    def _fetch_inventory_weave(self, revs):
186
 
        pb = bzrlib.ui.ui_factory.nested_progress_bar()
187
 
        try:
188
 
            pb.update("fetch inventory", 0, 2)
189
 
            to_weave = self.to_control.get_weave('inventory',
190
 
                    self.to_repository.get_transaction())
 
248
            child_pb.finished()
 
249
 
 
250
    def _fetch_revision_texts(self, revs, pb):
 
251
        # may need to be a InterRevisionStore call here.
 
252
        to_sf = self.to_repository.signatures
 
253
        from_sf = self.from_repository.signatures
 
254
        # A missing signature is just skipped.
 
255
        to_sf.insert_record_stream(filter_absent(from_sf.get_record_stream(
 
256
            [(rev_id,) for rev_id in revs],
 
257
            self.to_repository._fetch_order,
 
258
            not self.to_repository._fetch_uses_deltas)))
 
259
        self._fetch_just_revision_texts(revs)
 
260
 
 
261
    def _fetch_just_revision_texts(self, version_ids):
 
262
        to_rf = self.to_repository.revisions
 
263
        from_rf = self.from_repository.revisions
 
264
        to_rf.insert_record_stream(from_rf.get_record_stream(
 
265
            [(rev_id,) for rev_id in version_ids],
 
266
            self.to_repository._fetch_order,
 
267
            not self.to_repository._fetch_uses_deltas))
 
268
 
 
269
    def _generate_root_texts(self, revs):
 
270
        """This will be called by __fetch between fetching weave texts and
 
271
        fetching the inventory weave.
 
272
 
 
273
        Subclasses should override this if they need to generate root texts
 
274
        after fetching weave texts.
 
275
        """
 
276
        pass
 
277
 
 
278
 
 
279
class Inter1and2Helper(object):
 
280
    """Helper for operations that convert data from model 1 and 2
191
281
    
192
 
            child_pb = bzrlib.ui.ui_factory.nested_progress_bar()
193
 
            try:
194
 
                # just merge, this is optimisable and its means we don't
195
 
                # copy unreferenced data such as not-needed inventories.
196
 
                pb.update("fetch inventory", 1, 3)
197
 
                from_weave = self.from_repository.get_inventory_weave()
198
 
                pb.update("fetch inventory", 2, 3)
199
 
                # we fetch only the referenced inventories because we do not
200
 
                # know for unselected inventories whether all their required
201
 
                # texts are present in the other repository - it could be
202
 
                # corrupt.
203
 
                to_weave.join(from_weave, pb=child_pb, msg='merge inventory',
204
 
                              version_ids=revs)
205
 
            finally:
206
 
                child_pb.finished()
207
 
        finally:
208
 
            pb.finished()
209
 
 
210
 
 
211
 
class GenericRepoFetcher(RepoFetcher):
212
 
    """This is a generic repo to repo fetcher.
213
 
 
214
 
    This makes minimal assumptions about repo layout and contents.
215
 
    It triggers a reconciliation after fetching to ensure integrity.
216
 
    """
217
 
 
218
 
    def _fetch_revision_texts(self, revs):
 
282
    This is for use by fetchers and converters.
 
283
    """
 
284
 
 
285
    def __init__(self, source, target):
 
286
        """Constructor.
 
287
 
 
288
        :param source: The repository data comes from
 
289
        :param target: The repository data goes to
 
290
        """
 
291
        self.source = source
 
292
        self.target = target
 
293
 
 
294
    def iter_rev_trees(self, revs):
 
295
        """Iterate through RevisionTrees efficiently.
 
296
 
 
297
        Additionally, the inventory's revision_id is set if unset.
 
298
 
 
299
        Trees are retrieved in batches of 100, and then yielded in the order
 
300
        they were requested.
 
301
 
 
302
        :param revs: A list of revision ids
 
303
        """
 
304
        # In case that revs is not a list.
 
305
        revs = list(revs)
 
306
        while revs:
 
307
            for tree in self.source.revision_trees(revs[:100]):
 
308
                if tree.inventory.revision_id is None:
 
309
                    tree.inventory.revision_id = tree.get_revision_id()
 
310
                yield tree
 
311
            revs = revs[100:]
 
312
 
 
313
    def _find_root_ids(self, revs, parent_map, graph):
 
314
        revision_root = {}
 
315
        planned_versions = {}
 
316
        for tree in self.iter_rev_trees(revs):
 
317
            revision_id = tree.inventory.root.revision
 
318
            root_id = tree.get_root_id()
 
319
            planned_versions.setdefault(root_id, []).append(revision_id)
 
320
            revision_root[revision_id] = root_id
 
321
        # Find out which parents we don't already know root ids for
 
322
        parents = set()
 
323
        for revision_parents in parent_map.itervalues():
 
324
            parents.update(revision_parents)
 
325
        parents.difference_update(revision_root.keys() + [NULL_REVISION])
 
326
        # Limit to revisions present in the versionedfile
 
327
        parents = graph.get_parent_map(parents).keys()
 
328
        for tree in self.iter_rev_trees(parents):
 
329
            root_id = tree.get_root_id()
 
330
            revision_root[tree.get_revision_id()] = root_id
 
331
        return revision_root, planned_versions
 
332
 
 
333
    def generate_root_texts(self, revs):
 
334
        """Generate VersionedFiles for all root ids.
 
335
 
 
336
        :param revs: the revisions to include
 
337
        """
 
338
        to_texts = self.target.texts
 
339
        graph = self.source.get_graph()
 
340
        parent_map = graph.get_parent_map(revs)
 
341
        rev_order = topo_sort(parent_map)
 
342
        rev_id_to_root_id, root_id_to_rev_ids = self._find_root_ids(
 
343
            revs, parent_map, graph)
 
344
        root_id_order = [(rev_id_to_root_id[rev_id], rev_id) for rev_id in
 
345
            rev_order]
 
346
        # Guaranteed stable, this groups all the file id operations together
 
347
        # retaining topological order within the revisions of a file id.
 
348
        # File id splits and joins would invalidate this, but they don't exist
 
349
        # yet, and are unlikely to in non-rich-root environments anyway.
 
350
        root_id_order.sort(key=operator.itemgetter(0))
 
351
        # Create a record stream containing the roots to create.
 
352
        def yield_roots():
 
353
            for key in root_id_order:
 
354
                root_id, rev_id = key
 
355
                rev_parents = parent_map[rev_id]
 
356
                # We drop revision parents with different file-ids, because
 
357
                # that represents a rename of the root to a different location
 
358
                # - its not actually a parent for us. (We could look for that
 
359
                # file id in the revision tree at considerably more expense,
 
360
                # but for now this is sufficient (and reconcile will catch and
 
361
                # correct this anyway).
 
362
                # When a parent revision is a ghost, we guess that its root id
 
363
                # was unchanged (rather than trimming it from the parent list).
 
364
                parent_keys = tuple((root_id, parent) for parent in rev_parents
 
365
                    if parent != NULL_REVISION and
 
366
                        rev_id_to_root_id.get(parent, root_id) == root_id)
 
367
                yield FulltextContentFactory(key, parent_keys, None, '')
 
368
        to_texts.insert_record_stream(yield_roots())
 
369
 
 
370
    def regenerate_inventory(self, revs):
 
371
        """Generate a new inventory versionedfile in target, convertin data.
 
372
        
 
373
        The inventory is retrieved from the source, (deserializing it), and
 
374
        stored in the target (reserializing it in a different format).
 
375
        :param revs: The revisions to include
 
376
        """
 
377
        for tree in self.iter_rev_trees(revs):
 
378
            parents = tree.get_parent_ids()
 
379
            self.target.add_inventory(tree.get_revision_id(), tree.inventory,
 
380
                                      parents)
 
381
 
 
382
    def fetch_revisions(self, revision_ids):
 
383
        for revision in self.source.get_revisions(revision_ids):
 
384
            self.target.add_revision(revision.revision_id, revision)
 
385
 
 
386
 
 
387
class Model1toKnit2Fetcher(RepoFetcher):
 
388
    """Fetch from a Model1 repository into a Knit2 repository
 
389
    """
 
390
    def __init__(self, to_repository, from_repository, last_revision=None,
 
391
                 pb=None, find_ghosts=True):
 
392
        self.helper = Inter1and2Helper(from_repository, to_repository)
 
393
        RepoFetcher.__init__(self, to_repository, from_repository,
 
394
            last_revision, pb, find_ghosts)
 
395
 
 
396
    def _generate_root_texts(self, revs):
 
397
        self.helper.generate_root_texts(revs)
 
398
 
 
399
    def _fetch_inventory_weave(self, revs, pb):
 
400
        self.helper.regenerate_inventory(revs)
 
401
 
 
402
    def _fetch_revision_texts(self, revs, pb):
219
403
        """Fetch revision object texts"""
220
 
        rev_pb = bzrlib.ui.ui_factory.nested_progress_bar()
221
 
        try:
222
 
            to_txn = self.to_transaction = self.to_repository.get_transaction()
223
 
            count = 0
224
 
            total = len(revs)
225
 
            to_store = self.to_repository._revision_store
226
 
            for rev in revs:
227
 
                pb = bzrlib.ui.ui_factory.nested_progress_bar()
228
 
                try:
229
 
                    pb.update('copying revisions', count, total)
230
 
                    try:
231
 
                        sig_text = self.from_repository.get_signature_text(rev)
232
 
                        to_store.add_revision_signature_text(rev, sig_text, to_txn)
233
 
                    except errors.NoSuchRevision:
234
 
                        # not signed.
235
 
                        pass
236
 
                    to_store.add_revision(self.from_repository.get_revision(rev),
237
 
                                          to_txn)
238
 
                    count += 1
239
 
                finally:
240
 
                    pb.finished()
241
 
            # fixup inventory if needed: 
242
 
            # this is expensive because we have no inverse index to current ghosts.
243
 
            # but on local disk its a few seconds and sftp push is already insane.
244
 
            # so we just-do-it.
245
 
            # FIXME: repository should inform if this is needed.
246
 
            self.to_repository.reconcile()
247
 
        finally:
248
 
            rev_pb.finished()
249
 
    
250
 
 
251
 
class KnitRepoFetcher(RepoFetcher):
252
 
    """This is a knit format repository specific fetcher.
253
 
 
254
 
    This differs from the GenericRepoFetcher by not doing a 
255
 
    reconciliation after copying, and using knit joining to
256
 
    copy revision texts.
257
 
    """
258
 
 
259
 
    def _fetch_revision_texts(self, revs):
260
 
        # may need to be a InterRevisionStore call here.
261
 
        from_transaction = self.from_repository.get_transaction()
262
 
        to_transaction = self.to_repository.get_transaction()
263
 
        to_sf = self.to_repository._revision_store.get_signature_file(
264
 
            to_transaction)
265
 
        from_sf = self.from_repository._revision_store.get_signature_file(
266
 
            from_transaction)
267
 
        to_sf.join(from_sf, version_ids=revs, ignore_missing=True)
268
 
        to_rf = self.to_repository._revision_store.get_revision_file(
269
 
            to_transaction)
270
 
        from_rf = self.from_repository._revision_store.get_revision_file(
271
 
            from_transaction)
272
 
        to_rf.join(from_rf, version_ids=revs)
273
 
 
274
 
 
275
 
class Fetcher(object):
276
 
    """Backwards compatibility glue for branch.fetch()."""
277
 
 
278
 
    @deprecated_method(zero_eight)
279
 
    def __init__(self, to_branch, from_branch, last_revision=None, pb=None):
280
 
        """Please see branch.fetch()."""
281
 
        to_branch.fetch(from_branch, last_revision, pb)
 
404
        count = 0
 
405
        total = len(revs)
 
406
        for rev in revs:
 
407
            pb.update('copying revisions', count, total)
 
408
            try:
 
409
                sig_text = self.from_repository.get_signature_text(rev)
 
410
                self.to_repository.add_signature_text(rev, sig_text)
 
411
            except errors.NoSuchRevision:
 
412
                # not signed.
 
413
                pass
 
414
            self._copy_revision(rev)
 
415
            count += 1
 
416
 
 
417
    def _copy_revision(self, rev):
 
418
        self.helper.fetch_revisions([rev])
 
419
 
 
420
 
 
421
class Knit1to2Fetcher(RepoFetcher):
 
422
    """Fetch from a Knit1 repository into a Knit2 repository"""
 
423
 
 
424
    def __init__(self, to_repository, from_repository, last_revision=None,
 
425
                 pb=None, find_ghosts=True):
 
426
        self.helper = Inter1and2Helper(from_repository, to_repository)
 
427
        RepoFetcher.__init__(self, to_repository, from_repository,
 
428
            last_revision, pb, find_ghosts)
 
429
 
 
430
    def _generate_root_texts(self, revs):
 
431
        self.helper.generate_root_texts(revs)
 
432
 
 
433
    def _fetch_inventory_weave(self, revs, pb):
 
434
        self.helper.regenerate_inventory(revs)
 
435
 
 
436
    def _fetch_just_revision_texts(self, version_ids):
 
437
        self.helper.fetch_revisions(version_ids)