/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 breezy/commit.py

  • Committer: Breezy landing bot
  • Author(s): Jelmer Vernooij
  • Date: 2019-07-07 19:22:03 UTC
  • mfrom: (7358.8.5 diff-binary-weird)
  • Revision ID: breezy.the.bot@gmail.com-20190707192203-n32v0wih963qdi6i
Don't include datestamps in filenames when reporting on binary files.

Merged from https://code.launchpad.net/~jelmer/brz/diff-binary-weird/+merge/369474

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 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
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
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
# XXX: Can we do any better about making interrupted commits change
19
 
# nothing?  Perhaps the best approach is to integrate commit of
20
 
# AtomicFiles with releasing the lock on the Branch.
21
 
 
22
 
# TODO: Separate 'prepare' phase where we find a list of potentially
23
 
# committed files.  We then can then pause the commit to prompt for a
24
 
# commit message, knowing the summary will be the same as what's
25
 
# actually used for the commit.  (But perhaps simpler to simply get
26
 
# the tree status, then use that for a selective commit?)
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
from __future__ import absolute_import
27
18
 
28
19
# The newly committed revision is going to have a shape corresponding
29
 
# to that of the working inventory.  Files that are not in the
 
20
# to that of the working tree.  Files that are not in the
30
21
# working tree and that were in the predecessor are reported as
31
22
# removed --- this can include files that were either removed from the
32
23
# inventory or deleted in the working tree.  If they were only
35
26
# We then consider the remaining entries, which will be in the new
36
27
# version.  Directory entries are simply copied across.  File entries
37
28
# must be checked to see if a new version of the file should be
38
 
# recorded.  For each parent revision inventory, we check to see what
 
29
# recorded.  For each parent revision tree, we check to see what
39
30
# version of the file was present.  If the file was present in at
40
31
# least one tree, and if it was the same version in all the trees,
41
32
# then we can just refer to that version.  Otherwise, a new version
56
47
# merges from, then it should still be reported as newly added
57
48
# relative to the basis revision.
58
49
 
59
 
# TODO: Do checks that the tree can be committed *before* running the 
60
 
# editor; this should include checks for a pointless commit and for 
61
 
# unknown or missing files.
62
 
 
63
 
# TODO: If commit fails, leave the message in a file somewhere.
64
 
 
65
 
 
66
 
import os
67
 
import re
68
 
import sys
69
 
import time
70
 
import pdb
71
 
 
72
 
from binascii import hexlify
73
 
from cStringIO import StringIO
74
 
 
75
 
from bzrlib.atomicfile import AtomicFile
76
 
from bzrlib.osutils import (local_time_offset,
77
 
                            rand_bytes, compact_date,
78
 
                            kind_marker, is_inside_any, quotefn,
79
 
                            sha_file, isdir, isfile,
80
 
                            split_lines)
81
 
import bzrlib.config
82
 
import bzrlib.errors as errors
83
 
from bzrlib.errors import (BzrError, PointlessCommit,
84
 
                           HistoryMissing,
85
 
                           ConflictsInTree,
86
 
                           StrictCommitFailed
87
 
                           )
88
 
from bzrlib.revision import Revision
89
 
from bzrlib.testament import Testament
90
 
from bzrlib.trace import mutter, note, warning
91
 
from bzrlib.xml5 import serializer_v5
92
 
from bzrlib.inventory import Inventory, ROOT_ID
93
 
from bzrlib.symbol_versioning import *
94
 
from bzrlib.workingtree import WorkingTree
95
 
 
96
 
 
97
 
@deprecated_function(zero_seven)
98
 
def commit(*args, **kwargs):
99
 
    """Commit a new revision to a branch.
100
 
 
101
 
    Function-style interface for convenience of old callers.
102
 
 
103
 
    New code should use the Commit class instead.
 
50
# TODO: Change the parameter 'rev_id' to 'revision_id' to be consistent with
 
51
# the rest of the code; add a deprecation of the old name.
 
52
 
 
53
from . import (
 
54
    debug,
 
55
    errors,
 
56
    trace,
 
57
    ui,
 
58
    )
 
59
from .branch import Branch
 
60
from .cleanup import OperationWithCleanups
 
61
import breezy.config
 
62
from .errors import (BzrError,
 
63
                     ConflictsInTree,
 
64
                     StrictCommitFailed
 
65
                     )
 
66
from .osutils import (get_user_encoding,
 
67
                      has_symlinks,
 
68
                      is_inside_any,
 
69
                      minimum_path_selection,
 
70
                      )
 
71
from .trace import mutter, note, is_quiet
 
72
from .tree import TreeChange
 
73
from .urlutils import unescape_for_display
 
74
from .i18n import gettext
 
75
 
 
76
 
 
77
class PointlessCommit(BzrError):
 
78
 
 
79
    _fmt = "No changes to commit"
 
80
 
 
81
 
 
82
class CannotCommitSelectedFileMerge(BzrError):
 
83
 
 
84
    _fmt = 'Selected-file commit of merges is not supported yet:'\
 
85
        ' files %(files_str)s'
 
86
 
 
87
    def __init__(self, files):
 
88
        files_str = ', '.join(files)
 
89
        BzrError.__init__(self, files=files, files_str=files_str)
 
90
 
 
91
 
 
92
def filter_excluded(iter_changes, exclude):
 
93
    """Filter exclude filenames.
 
94
 
 
95
    :param iter_changes: iter_changes function
 
96
    :param exclude: List of paths to exclude
 
97
    :return: iter_changes function
104
98
    """
105
 
    ## XXX: Remove this in favor of Branch.commit?
106
 
    Commit().commit(*args, **kwargs)
 
99
    for change in iter_changes:
 
100
        new_excluded = (change.path[1] is not None and
 
101
                        is_inside_any(exclude, change.path[1]))
 
102
 
 
103
        old_excluded = (change.path[0] is not None and
 
104
                        is_inside_any(exclude, change.path[0]))
 
105
 
 
106
        if old_excluded and new_excluded:
 
107
            continue
 
108
 
 
109
        if old_excluded or new_excluded:
 
110
            # TODO(jelmer): Perhaps raise an error here instead?
 
111
            continue
 
112
 
 
113
        yield change
107
114
 
108
115
 
109
116
class NullCommitReporter(object):
110
117
    """I report on progress of a commit."""
111
118
 
 
119
    def started(self, revno, revid, location):
 
120
        pass
 
121
 
112
122
    def snapshot_change(self, change, path):
113
123
        pass
114
124
 
115
125
    def completed(self, revno, rev_id):
116
126
        pass
117
127
 
118
 
    def deleted(self, file_id):
119
 
        pass
120
 
 
121
 
    def escaped(self, escape_count, message):
 
128
    def deleted(self, path):
122
129
        pass
123
130
 
124
131
    def missing(self, path):
125
132
        pass
126
133
 
 
134
    def renamed(self, change, old_path, new_path):
 
135
        pass
 
136
 
 
137
    def is_verbose(self):
 
138
        return False
 
139
 
127
140
 
128
141
class ReportCommitToLog(NullCommitReporter):
129
142
 
 
143
    def _note(self, format, *args):
 
144
        """Output a message.
 
145
 
 
146
        Subclasses may choose to override this method.
 
147
        """
 
148
        note(format, *args)
 
149
 
130
150
    def snapshot_change(self, change, path):
131
 
        if change == 'unchanged':
 
151
        if path == '' and change in (gettext('added'), gettext('modified')):
132
152
            return
133
 
        note("%s %s", change, path)
 
153
        self._note("%s %s", change, path)
 
154
 
 
155
    def started(self, revno, rev_id, location):
 
156
        self._note(
 
157
            gettext('Committing to: %s'),
 
158
            unescape_for_display(location, 'utf-8'))
134
159
 
135
160
    def completed(self, revno, rev_id):
136
 
        note('Committed revision %d.', revno)
137
 
    
138
 
    def deleted(self, file_id):
139
 
        note('deleted %s', file_id)
 
161
        if revno is not None:
 
162
            self._note(gettext('Committed revision %d.'), revno)
 
163
            # self._note goes to the console too; so while we want to log the
 
164
            # rev_id, we can't trivially only log it. (See bug 526425). Long
 
165
            # term we should rearrange the reporting structure, but for now
 
166
            # we just mutter seperately. We mutter the revid and revno together
 
167
            # so that concurrent bzr invocations won't lead to confusion.
 
168
            mutter('Committed revid %s as revno %d.', rev_id, revno)
 
169
        else:
 
170
            self._note(gettext('Committed revid %s.'), rev_id)
140
171
 
141
 
    def escaped(self, escape_count, message):
142
 
        note("replaced %d control characters in message", escape_count)
 
172
    def deleted(self, path):
 
173
        self._note(gettext('deleted %s'), path)
143
174
 
144
175
    def missing(self, path):
145
 
        note('missing %s', path)
 
176
        self._note(gettext('missing %s'), path)
 
177
 
 
178
    def renamed(self, change, old_path, new_path):
 
179
        self._note('%s %s => %s', change, old_path, new_path)
 
180
 
 
181
    def is_verbose(self):
 
182
        return True
146
183
 
147
184
 
148
185
class Commit(object):
157
194
            the working directory; these should be removed from the
158
195
            working inventory.
159
196
    """
 
197
 
160
198
    def __init__(self,
161
199
                 reporter=None,
162
 
                 config=None):
163
 
        if reporter is not None:
164
 
            self.reporter = reporter
165
 
        else:
166
 
            self.reporter = NullCommitReporter()
167
 
        if config is not None:
168
 
            self.config = config
169
 
        else:
170
 
            self.config = None
171
 
        
 
200
                 config_stack=None):
 
201
        """Create a Commit object.
 
202
 
 
203
        :param reporter: the default reporter to use or None to decide later
 
204
        """
 
205
        self.reporter = reporter
 
206
        self.config_stack = config_stack
 
207
 
 
208
    @staticmethod
 
209
    def update_revprops(revprops, branch, authors=None,
 
210
                        local=False, possible_master_transports=None):
 
211
        if revprops is None:
 
212
            revprops = {}
 
213
        if possible_master_transports is None:
 
214
            possible_master_transports = []
 
215
        if (u'branch-nick' not in revprops and
 
216
                branch.repository._format.supports_storing_branch_nick):
 
217
            revprops[u'branch-nick'] = branch._get_nick(
 
218
                local,
 
219
                possible_master_transports)
 
220
        if authors is not None:
 
221
            if u'author' in revprops or u'authors' in revprops:
 
222
                # XXX: maybe we should just accept one of them?
 
223
                raise AssertionError('author property given twice')
 
224
            if authors:
 
225
                for individual in authors:
 
226
                    if '\n' in individual:
 
227
                        raise AssertionError('\\n is not a valid character '
 
228
                                             'in an author identity')
 
229
                revprops[u'authors'] = '\n'.join(authors)
 
230
        return revprops
 
231
 
172
232
    def commit(self,
173
 
               branch=DEPRECATED_PARAMETER, message=None,
 
233
               message=None,
174
234
               timestamp=None,
175
235
               timezone=None,
176
236
               committer=None,
183
243
               working_tree=None,
184
244
               local=False,
185
245
               reporter=None,
186
 
               config=None):
 
246
               config=None,
 
247
               message_callback=None,
 
248
               recursive='down',
 
249
               exclude=None,
 
250
               possible_master_transports=None,
 
251
               lossy=False):
187
252
        """Commit working copy as a new revision.
188
253
 
189
 
        branch -- the deprecated branch to commit to. New callers should pass in 
190
 
                  working_tree instead
191
 
 
192
 
        message -- the commit message, a mandatory parameter
193
 
 
194
 
        timestamp -- if not None, seconds-since-epoch for a
195
 
             postdated/predated commit.
196
 
 
197
 
        specific_files -- If true, commit only those files.
198
 
 
199
 
        rev_id -- If set, use this as the new revision id.
 
254
        :param message: the commit message (it or message_callback is required)
 
255
        :param message_callback: A callback: message =
 
256
            message_callback(cmt_obj)
 
257
 
 
258
        :param timestamp: if not None, seconds-since-epoch for a
 
259
            postdated/predated commit.
 
260
 
 
261
        :param specific_files: If not None, commit only those files. An empty
 
262
            list means 'commit no files'.
 
263
 
 
264
        :param rev_id: If set, use this as the new revision id.
200
265
            Useful for test or import commands that need to tightly
201
266
            control what revisions are assigned.  If you duplicate
202
267
            a revision id that exists elsewhere it is your own fault.
203
268
            If null (default), a time/random revision id is generated.
204
269
 
205
 
        allow_pointless -- If true (default), commit even if nothing
 
270
        :param allow_pointless: If true (default), commit even if nothing
206
271
            has changed and no merges are recorded.
207
272
 
208
 
        strict -- If true, don't allow a commit if the working tree
 
273
        :param strict: If true, don't allow a commit if the working tree
209
274
            contains unknown files.
210
275
 
211
 
        revprops -- Properties for new revision
 
276
        :param revprops: Properties for new revision
212
277
        :param local: Perform a local only commit.
 
278
        :param reporter: the reporter to use or None for the default
 
279
        :param verbose: if True and the reporter is not None, report everything
 
280
        :param recursive: If set to 'down', commit in any subtrees that have
 
281
            pending changes of any sort during this commit.
 
282
        :param exclude: None or a list of relative paths to exclude from the
 
283
            commit. Pending changes to excluded files will be ignored by the
 
284
            commit.
 
285
        :param lossy: When committing to a foreign VCS, ignore any
 
286
            data that can not be natively represented.
213
287
        """
 
288
        operation = OperationWithCleanups(self._commit)
 
289
        self.revprops = revprops or {}
 
290
        # XXX: Can be set on __init__ or passed in - this is a bit ugly.
 
291
        self.config_stack = config or self.config_stack
 
292
        return operation.run(
 
293
            message=message,
 
294
            timestamp=timestamp,
 
295
            timezone=timezone,
 
296
            committer=committer,
 
297
            specific_files=specific_files,
 
298
            rev_id=rev_id,
 
299
            allow_pointless=allow_pointless,
 
300
            strict=strict,
 
301
            verbose=verbose,
 
302
            working_tree=working_tree,
 
303
            local=local,
 
304
            reporter=reporter,
 
305
            message_callback=message_callback,
 
306
            recursive=recursive,
 
307
            exclude=exclude,
 
308
            possible_master_transports=possible_master_transports,
 
309
            lossy=lossy)
 
310
 
 
311
    def _commit(self, operation, message, timestamp, timezone, committer,
 
312
                specific_files, rev_id, allow_pointless, strict, verbose,
 
313
                working_tree, local, reporter, message_callback, recursive,
 
314
                exclude, possible_master_transports, lossy):
214
315
        mutter('preparing to commit')
215
316
 
216
 
        if deprecated_passed(branch):
217
 
            warn("Commit.commit (branch, ...): The branch parameter is "
218
 
                 "deprecated as of bzr 0.8. Please use working_tree= instead.",
219
 
                 DeprecationWarning, stacklevel=2)
220
 
            self.branch = branch
221
 
            self.work_tree = self.branch.bzrdir.open_workingtree()
222
 
        elif working_tree is None:
223
 
            raise BzrError("One of branch and working_tree must be passed into commit().")
 
317
        if working_tree is None:
 
318
            raise BzrError("working_tree must be passed into commit().")
224
319
        else:
225
320
            self.work_tree = working_tree
226
321
            self.branch = self.work_tree.branch
227
 
        if message is None:
228
 
            raise BzrError("The message keyword parameter is required for commit().")
229
 
 
230
 
        self.weave_store = self.branch.repository.weave_store
 
322
            if getattr(self.work_tree, 'requires_rich_root', lambda: False)():
 
323
                if not self.branch.repository.supports_rich_root():
 
324
                    raise errors.RootNotRich()
 
325
        if message_callback is None:
 
326
            if message is not None:
 
327
                if isinstance(message, bytes):
 
328
                    message = message.decode(get_user_encoding())
 
329
 
 
330
                def message_callback(x):
 
331
                    return message
 
332
            else:
 
333
                raise BzrError("The message or message_callback keyword"
 
334
                               " parameter is required for commit().")
 
335
 
231
336
        self.bound_branch = None
 
337
        self.any_entries_deleted = False
 
338
        if exclude is not None:
 
339
            self.exclude = sorted(
 
340
                minimum_path_selection(exclude))
 
341
        else:
 
342
            self.exclude = []
232
343
        self.local = local
233
344
        self.master_branch = None
234
 
        self.master_locked = False
235
 
        self.rev_id = rev_id
236
 
        self.specific_files = specific_files
 
345
        self.recursive = recursive
 
346
        self.rev_id = None
 
347
        # self.specific_files is None to indicate no filter, or any iterable to
 
348
        # indicate a filter - [] means no files at all, as per iter_changes.
 
349
        if specific_files is not None:
 
350
            self.specific_files = sorted(
 
351
                minimum_path_selection(specific_files))
 
352
        else:
 
353
            self.specific_files = None
 
354
 
237
355
        self.allow_pointless = allow_pointless
238
 
        self.revprops = {}
239
 
        if revprops is not None:
240
 
            self.revprops.update(revprops)
241
 
 
242
 
        if reporter is None and self.reporter is None:
243
 
            self.reporter = NullCommitReporter()
244
 
        elif reporter is not None:
245
 
            self.reporter = reporter
 
356
        self.message_callback = message_callback
 
357
        self.timestamp = timestamp
 
358
        self.timezone = timezone
 
359
        self.committer = committer
 
360
        self.strict = strict
 
361
        self.verbose = verbose
246
362
 
247
363
        self.work_tree.lock_write()
 
364
        operation.add_cleanup(self.work_tree.unlock)
 
365
        self.parents = self.work_tree.get_parent_ids()
 
366
        self.pb = ui.ui_factory.nested_progress_bar()
 
367
        operation.add_cleanup(self.pb.finished)
 
368
        self.basis_revid = self.work_tree.last_revision()
 
369
        self.basis_tree = self.work_tree.basis_tree()
 
370
        self.basis_tree.lock_read()
 
371
        operation.add_cleanup(self.basis_tree.unlock)
 
372
        # Cannot commit with conflicts present.
 
373
        if len(self.work_tree.conflicts()) > 0:
 
374
            raise ConflictsInTree
 
375
 
 
376
        # Setup the bound branch variables as needed.
 
377
        self._check_bound_branch(operation, possible_master_transports)
 
378
 
 
379
        if self.config_stack is None:
 
380
            self.config_stack = self.work_tree.get_config_stack()
 
381
 
 
382
        # Check that the working tree is up to date
 
383
        old_revno, old_revid, new_revno = self._check_out_of_date_tree()
 
384
 
 
385
        # Complete configuration setup
 
386
        if reporter is not None:
 
387
            self.reporter = reporter
 
388
        elif self.reporter is None:
 
389
            self.reporter = self._select_reporter()
 
390
 
 
391
        # Setup the progress bar. As the number of files that need to be
 
392
        # committed in unknown, progress is reported as stages.
 
393
        # We keep track of entries separately though and include that
 
394
        # information in the progress bar during the relevant stages.
 
395
        self.pb_stage_name = ""
 
396
        self.pb_stage_count = 0
 
397
        self.pb_stage_total = 5
 
398
        if self.bound_branch:
 
399
            # 2 extra stages: "Uploading data to master branch" and "Merging
 
400
            # tags to master branch"
 
401
            self.pb_stage_total += 2
 
402
        self.pb.show_pct = False
 
403
        self.pb.show_spinner = False
 
404
        self.pb.show_eta = False
 
405
        self.pb.show_count = True
 
406
        self.pb.show_bar = True
 
407
 
 
408
        # After a merge, a selected file commit is not supported.
 
409
        # See 'bzr help merge' for an explanation as to why.
 
410
        if len(self.parents) > 1 and self.specific_files is not None:
 
411
            raise CannotCommitSelectedFileMerge(self.specific_files)
 
412
        # Excludes are a form of selected file commit.
 
413
        if len(self.parents) > 1 and self.exclude:
 
414
            raise CannotCommitSelectedFileMerge(self.exclude)
 
415
 
 
416
        # Collect the changes
 
417
        self._set_progress_stage("Collecting changes", counter=True)
 
418
        self._lossy = lossy
 
419
        self.builder = self.branch.get_commit_builder(
 
420
            self.parents, self.config_stack, timestamp, timezone, committer,
 
421
            self.revprops, rev_id, lossy=lossy)
 
422
 
 
423
        if self.builder.updates_branch and self.bound_branch:
 
424
            self.builder.abort()
 
425
            raise AssertionError(
 
426
                "bound branches not supported for commit builders "
 
427
                "that update the branch")
 
428
 
248
429
        try:
249
 
            # setup the bound branch variables as needed.
250
 
            self._check_bound_branch()
251
 
 
252
 
            # check for out of date working trees
253
 
            # if we are bound, then self.branch is the master branch and this
254
 
            # test is thus all we need.
255
 
            if self.work_tree.last_revision() != self.master_branch.last_revision():
256
 
                raise errors.OutOfDateTree(self.work_tree)
257
 
    
258
 
            if strict:
259
 
                # raise an exception as soon as we find a single unknown.
260
 
                for unknown in self.work_tree.unknowns():
261
 
                    raise StrictCommitFailed()
262
 
    
263
 
            if timestamp is None:
264
 
                self.timestamp = time.time()
265
 
            else:
266
 
                self.timestamp = long(timestamp)
267
 
                
268
 
            if self.config is None:
269
 
                self.config = bzrlib.config.BranchConfig(self.branch)
270
 
    
271
 
            if rev_id is None:
272
 
                self.rev_id = _gen_revision_id(self.config, self.timestamp)
273
 
            else:
274
 
                self.rev_id = rev_id
275
 
    
276
 
            if committer is None:
277
 
                self.committer = self.config.username()
278
 
            else:
279
 
                assert isinstance(committer, basestring), type(committer)
280
 
                self.committer = committer
281
 
    
282
 
            if timezone is None:
283
 
                self.timezone = local_time_offset()
284
 
            else:
285
 
                self.timezone = int(timezone)
286
 
    
287
 
            if isinstance(message, str):
288
 
                message = message.decode(bzrlib.user_encoding)
289
 
            assert isinstance(message, unicode), type(message)
 
430
            # find the location being committed to
 
431
            if self.bound_branch:
 
432
                master_location = self.master_branch.base
 
433
            else:
 
434
                master_location = self.branch.base
 
435
 
 
436
            # report the start of the commit
 
437
            self.reporter.started(new_revno, self.rev_id, master_location)
 
438
 
 
439
            self._update_builder_with_changes()
 
440
            self._check_pointless()
 
441
 
 
442
            # TODO: Now the new inventory is known, check for conflicts.
 
443
            # ADHB 2006-08-08: If this is done, populate_new_inv should not add
 
444
            # weave lines, because nothing should be recorded until it is known
 
445
            # that commit will succeed.
 
446
            self._set_progress_stage("Saving data locally")
 
447
            self.builder.finish_inventory()
 
448
 
 
449
            # Prompt the user for a commit message if none provided
 
450
            message = message_callback(self)
290
451
            self.message = message
291
 
            self._escape_commit_message()
292
 
 
293
 
            self.work_inv = self.work_tree.inventory
294
 
            self.basis_tree = self.work_tree.basis_tree()
295
 
            self.basis_inv = self.basis_tree.inventory
296
 
 
297
 
            self._gather_parents()
298
 
            if len(self.parents) > 1 and self.specific_files:
299
 
                raise NotImplementedError('selected-file commit of merges is not supported yet')
300
 
            self._check_parents_present()
301
 
            
302
 
            self._remove_deleted()
303
 
            self._populate_new_inv()
304
 
            self._store_snapshot()
305
 
            self._report_deletes()
306
 
 
307
 
            if not (self.allow_pointless
308
 
                    or len(self.parents) > 1
309
 
                    or self.new_inv != self.basis_inv):
310
 
                raise PointlessCommit()
311
 
 
312
 
            if len(self.work_tree.conflicts())>0:
313
 
                raise ConflictsInTree
314
 
 
315
 
            self.inv_sha1 = self.branch.repository.add_inventory(
316
 
                self.rev_id,
317
 
                self.new_inv,
318
 
                self.present_parents
319
 
                )
320
 
            self._make_revision()
321
 
            # revision data is in the local branch now.
322
 
            
323
 
            # upload revision data to the master.
324
 
            # this will propogate merged revisions too if needed.
 
452
 
 
453
            # Add revision data to the local branch
 
454
            self.rev_id = self.builder.commit(self.message)
 
455
 
 
456
        except Exception:
 
457
            mutter("aborting commit write group because of exception:")
 
458
            trace.log_exception_quietly()
 
459
            self.builder.abort()
 
460
            raise
 
461
 
 
462
        self._update_branches(old_revno, old_revid, new_revno)
 
463
 
 
464
        # Make the working tree be up to date with the branch. This
 
465
        # includes automatic changes scheduled to be made to the tree, such
 
466
        # as updating its basis and unversioning paths that were missing.
 
467
        self.work_tree.unversion(self.deleted_paths)
 
468
        self._set_progress_stage("Updating the working tree")
 
469
        self.work_tree.update_basis_by_delta(self.rev_id,
 
470
                                             self.builder.get_basis_delta())
 
471
        self.reporter.completed(new_revno, self.rev_id)
 
472
        self._process_post_hooks(old_revno, new_revno)
 
473
        return self.rev_id
 
474
 
 
475
    def _update_branches(self, old_revno, old_revid, new_revno):
 
476
        """Update the master and local branch to the new revision.
 
477
 
 
478
        This will try to make sure that the master branch is updated
 
479
        before the local branch.
 
480
 
 
481
        :param old_revno: Revision number of master branch before the
 
482
            commit
 
483
        :param old_revid: Tip of master branch before the commit
 
484
        :param new_revno: Revision number of the new commit
 
485
        """
 
486
        if not self.builder.updates_branch:
 
487
            self._process_pre_hooks(old_revno, new_revno)
 
488
 
 
489
            # Upload revision data to the master.
 
490
            # this will propagate merged revisions too if needed.
325
491
            if self.bound_branch:
326
 
                self.master_branch.repository.fetch(self.branch.repository,
327
 
                                                    revision_id=self.rev_id)
328
 
                # now the master has the revision data
329
 
                # 'commit' to the master first so a timeout here causes the local
330
 
                # branch to be out of date
331
 
                self.master_branch.append_revision(self.rev_id)
 
492
                self._set_progress_stage("Uploading data to master branch")
 
493
                # 'commit' to the master first so a timeout here causes the
 
494
                # local branch to be out of date
 
495
                (new_revno, self.rev_id) = self.master_branch.import_last_revision_info_and_tags(
 
496
                    self.branch, new_revno, self.rev_id, lossy=self._lossy)
 
497
                if self._lossy:
 
498
                    self.branch.fetch(self.master_branch, self.rev_id)
332
499
 
333
500
            # and now do the commit locally.
334
 
            self.branch.append_revision(self.rev_id)
335
 
 
336
 
            self.work_tree.set_pending_merges([])
337
 
            self.work_tree.set_last_revision(self.rev_id)
338
 
            # now the work tree is up to date with the branch
339
 
            
340
 
            self.reporter.completed(self.branch.revno(), self.rev_id)
341
 
            if self.config.post_commit() is not None:
342
 
                hooks = self.config.post_commit().split(' ')
343
 
                # this would be nicer with twisted.python.reflect.namedAny
344
 
                for hook in hooks:
345
 
                    result = eval(hook + '(branch, rev_id)',
346
 
                                  {'branch':self.branch,
347
 
                                   'bzrlib':bzrlib,
348
 
                                   'rev_id':self.rev_id})
349
 
        finally:
350
 
            self._cleanup_bound_branch()
351
 
            self.work_tree.unlock()
352
 
 
353
 
    def _check_bound_branch(self):
 
501
            if new_revno is None:
 
502
                # Keep existing behaviour around ghosts
 
503
                new_revno = 1
 
504
            self.branch.set_last_revision_info(new_revno, self.rev_id)
 
505
        else:
 
506
            try:
 
507
                self._process_pre_hooks(old_revno, new_revno)
 
508
            except BaseException:
 
509
                # The commit builder will already have updated the branch,
 
510
                # revert it.
 
511
                self.branch.set_last_revision_info(old_revno, old_revid)
 
512
                raise
 
513
 
 
514
        # Merge local tags to remote
 
515
        if self.bound_branch:
 
516
            self._set_progress_stage("Merging tags to master branch")
 
517
            tag_updates, tag_conflicts = self.branch.tags.merge_to(
 
518
                self.master_branch.tags)
 
519
            if tag_conflicts:
 
520
                warning_lines = ['    ' + name for name, _, _ in tag_conflicts]
 
521
                note(gettext("Conflicting tags in bound branch:\n{0}".format(
 
522
                    "\n".join(warning_lines))))
 
523
 
 
524
    def _select_reporter(self):
 
525
        """Select the CommitReporter to use."""
 
526
        if is_quiet():
 
527
            return NullCommitReporter()
 
528
        return ReportCommitToLog()
 
529
 
 
530
    def _check_pointless(self):
 
531
        if self.allow_pointless:
 
532
            return
 
533
        # A merge with no effect on files
 
534
        if len(self.parents) > 1:
 
535
            return
 
536
        if self.builder.any_changes():
 
537
            return
 
538
        raise PointlessCommit()
 
539
 
 
540
    def _check_bound_branch(self, operation, possible_master_transports=None):
354
541
        """Check to see if the local branch is bound.
355
542
 
356
543
        If it is bound, then most of the commit will actually be
361
548
            raise errors.LocalRequiresBoundBranch()
362
549
 
363
550
        if not self.local:
364
 
            self.master_branch = self.branch.get_master_branch()
 
551
            self.master_branch = self.branch.get_master_branch(
 
552
                possible_master_transports)
365
553
 
366
554
        if not self.master_branch:
367
555
            # make this branch the reference branch for out of date checks.
371
559
        # If the master branch is bound, we must fail
372
560
        master_bound_location = self.master_branch.get_bound_location()
373
561
        if master_bound_location:
374
 
            raise errors.CommitToDoubleBoundBranch(self.branch,
375
 
                    self.master_branch, master_bound_location)
 
562
            raise errors.CommitToDoubleBoundBranch(
 
563
                self.branch, self.master_branch, master_bound_location)
376
564
 
377
565
        # TODO: jam 20051230 We could automatically push local
378
566
        #       commits to the remote branch if they would fit.
379
567
        #       But for now, just require remote to be identical
380
568
        #       to local.
381
 
        
 
569
 
382
570
        # Make sure the local branch is identical to the master
383
 
        master_rh = self.master_branch.revision_history()
384
 
        local_rh = self.branch.revision_history()
385
 
        if local_rh != master_rh:
 
571
        master_revid = self.master_branch.last_revision()
 
572
        local_revid = self.branch.last_revision()
 
573
        if local_revid != master_revid:
386
574
            raise errors.BoundBranchOutOfDate(self.branch,
387
 
                    self.master_branch)
 
575
                                              self.master_branch)
388
576
 
389
577
        # Now things are ready to change the master branch
390
578
        # so grab the lock
391
579
        self.bound_branch = self.branch
392
580
        self.master_branch.lock_write()
393
 
        self.master_locked = True
394
 
####        
395
 
####        # Check to see if we have any pending merges. If we do
396
 
####        # those need to be pushed into the master branch
397
 
####        pending_merges = self.work_tree.pending_merges()
398
 
####        if pending_merges:
399
 
####            for revision_id in pending_merges:
400
 
####                self.master_branch.repository.fetch(self.bound_branch.repository,
401
 
####                                                    revision_id=revision_id)
402
 
 
403
 
    def _cleanup_bound_branch(self):
404
 
        """Executed at the end of a try/finally to cleanup a bound branch.
405
 
 
406
 
        If the branch wasn't bound, this is a no-op.
407
 
        If it was, it resents self.branch to the local branch, instead
408
 
        of being the master.
 
581
        operation.add_cleanup(self.master_branch.unlock)
 
582
 
 
583
    def _check_out_of_date_tree(self):
 
584
        """Check that the working tree is up to date.
 
585
 
 
586
        :return: old_revision_number, old_revision_id, new_revision_number
 
587
            tuple
409
588
        """
 
589
        try:
 
590
            first_tree_parent = self.work_tree.get_parent_ids()[0]
 
591
        except IndexError:
 
592
            # if there are no parents, treat our parent as 'None'
 
593
            # this is so that we still consider the master branch
 
594
            # - in a checkout scenario the tree may have no
 
595
            # parents but the branch may do.
 
596
            first_tree_parent = breezy.revision.NULL_REVISION
 
597
        if (self.master_branch._format.stores_revno() or
 
598
                self.config_stack.get('calculate_revnos')):
 
599
            try:
 
600
                old_revno, master_last = self.master_branch.last_revision_info()
 
601
            except errors.UnsupportedOperation:
 
602
                master_last = self.master_branch.last_revision()
 
603
                old_revno = self.branch.revision_id_to_revno(master_last)
 
604
        else:
 
605
            master_last = self.master_branch.last_revision()
 
606
            old_revno = None
 
607
        if master_last != first_tree_parent:
 
608
            if master_last != breezy.revision.NULL_REVISION:
 
609
                raise errors.OutOfDateTree(self.work_tree)
 
610
        if (old_revno is not None and
 
611
                self.branch.repository.has_revision(first_tree_parent)):
 
612
            new_revno = old_revno + 1
 
613
        else:
 
614
            # ghost parents never appear in revision history.
 
615
            new_revno = None
 
616
        return old_revno, master_last, new_revno
 
617
 
 
618
    def _process_pre_hooks(self, old_revno, new_revno):
 
619
        """Process any registered pre commit hooks."""
 
620
        self._set_progress_stage("Running pre_commit hooks")
 
621
        self._process_hooks("pre_commit", old_revno, new_revno)
 
622
 
 
623
    def _process_post_hooks(self, old_revno, new_revno):
 
624
        """Process any registered post commit hooks."""
 
625
        # Process the post commit hooks, if any
 
626
        self._set_progress_stage("Running post_commit hooks")
 
627
        # old style commit hooks - should be deprecated ? (obsoleted in
 
628
        # 0.15^H^H^H^H 2.5.0)
 
629
        post_commit = self.config_stack.get('post_commit')
 
630
        if post_commit is not None:
 
631
            hooks = post_commit.split(' ')
 
632
            # this would be nicer with twisted.python.reflect.namedAny
 
633
            for hook in hooks:
 
634
                result = eval(hook + '(branch, rev_id)',
 
635
                              {'branch': self.branch,
 
636
                               'breezy': breezy,
 
637
                               'rev_id': self.rev_id})
 
638
        # process new style post commit hooks
 
639
        self._process_hooks("post_commit", old_revno, new_revno)
 
640
 
 
641
    def _process_hooks(self, hook_name, old_revno, new_revno):
 
642
        if not Branch.hooks[hook_name]:
 
643
            return
 
644
 
 
645
        # new style commit hooks:
410
646
        if not self.bound_branch:
411
 
            return
412
 
        if self.master_locked:
413
 
            self.master_branch.unlock()
414
 
 
415
 
    def _escape_commit_message(self):
416
 
        """Replace xml-incompatible control characters."""
417
 
        # Python strings can include characters that can't be
418
 
        # represented in well-formed XML; escape characters that
419
 
        # aren't listed in the XML specification
420
 
        # (http://www.w3.org/TR/REC-xml/#NT-Char).
421
 
        self.message, escape_count = re.subn(
422
 
            u'[^\x09\x0A\x0D\u0020-\uD7FF\uE000-\uFFFD]+',
423
 
            lambda match: match.group(0).encode('unicode_escape'),
424
 
            self.message)
425
 
        if escape_count:
426
 
            self.reporter.escaped(escape_count, self.message)
427
 
 
428
 
    def _gather_parents(self):
429
 
        """Record the parents of a merge for merge detection."""
430
 
        pending_merges = self.work_tree.pending_merges()
431
 
        self.parents = []
432
 
        self.parent_invs = []
433
 
        self.present_parents = []
434
 
        precursor_id = self.branch.last_revision()
435
 
        if precursor_id:
436
 
            self.parents.append(precursor_id)
437
 
        self.parents += pending_merges
438
 
        for revision in self.parents:
439
 
            if self.branch.repository.has_revision(revision):
440
 
                inventory = self.branch.repository.get_inventory(revision)
441
 
                self.parent_invs.append(inventory)
442
 
                self.present_parents.append(revision)
443
 
 
444
 
    def _check_parents_present(self):
445
 
        for parent_id in self.parents:
446
 
            mutter('commit parent revision {%s}', parent_id)
447
 
            if not self.branch.repository.has_revision(parent_id):
448
 
                if parent_id == self.branch.last_revision():
449
 
                    warning("parent is missing %r", parent_id)
450
 
                    raise HistoryMissing(self.branch, 'revision', parent_id)
451
 
                else:
452
 
                    mutter("commit will ghost revision %r", parent_id)
453
 
            
454
 
    def _make_revision(self):
455
 
        """Record a new revision object for this commit."""
456
 
        rev = Revision(timestamp=self.timestamp,
457
 
                       timezone=self.timezone,
458
 
                       committer=self.committer,
459
 
                       message=self.message,
460
 
                       inventory_sha1=self.inv_sha1,
461
 
                       revision_id=self.rev_id,
462
 
                       properties=self.revprops)
463
 
        rev.parent_ids = self.parents
464
 
        self.branch.repository.add_revision(self.rev_id, rev, self.new_inv, self.config)
465
 
 
466
 
    def _remove_deleted(self):
467
 
        """Remove deleted files from the working inventories.
468
 
 
469
 
        This is done prior to taking the working inventory as the
470
 
        basis for the new committed inventory.
471
 
 
472
 
        This returns true if any files
473
 
        *that existed in the basis inventory* were deleted.
474
 
        Files that were added and deleted
475
 
        in the working copy don't matter.
476
 
        """
477
 
        specific = self.specific_files
478
 
        deleted_ids = []
479
 
        for path, ie in self.work_inv.iter_entries():
480
 
            if specific and not is_inside_any(specific, path):
481
 
                continue
482
 
            if not self.work_tree.has_filename(path):
483
 
                self.reporter.missing(path)
484
 
                deleted_ids.append((path, ie.file_id))
485
 
        if deleted_ids:
486
 
            deleted_ids.sort(reverse=True)
487
 
            for path, file_id in deleted_ids:
488
 
                del self.work_inv[file_id]
489
 
            self.work_tree._write_inventory(self.work_inv)
490
 
 
491
 
    def _store_snapshot(self):
492
 
        """Pass over inventory and record a snapshot.
493
 
 
494
 
        Entries get a new revision when they are modified in 
495
 
        any way, which includes a merge with a new set of
496
 
        parents that have the same entry. 
497
 
        """
498
 
        # XXX: Need to think more here about when the user has
499
 
        # made a specific decision on a particular value -- c.f.
500
 
        # mark-merge.  
501
 
        for path, ie in self.new_inv.iter_entries():
502
 
            previous_entries = ie.find_previous_heads(
503
 
                self.parent_invs,
504
 
                self.weave_store,
505
 
                self.branch.repository.get_transaction())
506
 
            if ie.revision is None:
507
 
                change = ie.snapshot(self.rev_id, path, previous_entries,
508
 
                                     self.work_tree, self.weave_store,
509
 
                                     self.branch.get_transaction())
510
 
            else:
511
 
                change = "unchanged"
512
 
            self.reporter.snapshot_change(change, path)
513
 
 
514
 
    def _populate_new_inv(self):
515
 
        """Build revision inventory.
516
 
 
517
 
        This creates a new empty inventory. Depending on
518
 
        which files are selected for commit, and what is present in the
519
 
        current tree, the new inventory is populated. inventory entries 
520
 
        which are candidates for modification have their revision set to
521
 
        None; inventory entries that are carried over untouched have their
522
 
        revision set to their prior value.
523
 
        """
524
 
        mutter("Selecting files for commit with filter %s", self.specific_files)
525
 
        self.new_inv = Inventory(revision_id=self.rev_id)
526
 
        for path, new_ie in self.work_inv.iter_entries():
527
 
            file_id = new_ie.file_id
528
 
            mutter('check %s {%s}', path, new_ie.file_id)
529
 
            if self.specific_files:
530
 
                if not is_inside_any(self.specific_files, path):
531
 
                    mutter('%s not selected for commit', path)
532
 
                    self._carry_entry(file_id)
 
647
            hook_master = self.branch
 
648
            hook_local = None
 
649
        else:
 
650
            hook_master = self.master_branch
 
651
            hook_local = self.branch
 
652
        # With bound branches, when the master is behind the local branch,
 
653
        # the 'old_revno' and old_revid values here are incorrect.
 
654
        # XXX: FIXME ^. RBC 20060206
 
655
        if self.parents:
 
656
            old_revid = self.parents[0]
 
657
        else:
 
658
            old_revid = breezy.revision.NULL_REVISION
 
659
 
 
660
        if hook_name == "pre_commit":
 
661
            future_tree = self.builder.revision_tree()
 
662
            tree_delta = future_tree.changes_from(self.basis_tree,
 
663
                                                  include_root=True)
 
664
 
 
665
        for hook in Branch.hooks[hook_name]:
 
666
            # show the running hook in the progress bar. As hooks may
 
667
            # end up doing nothing (e.g. because they are not configured by
 
668
            # the user) this is still showing progress, not showing overall
 
669
            # actions - its up to each plugin to show a UI if it want's to
 
670
            # (such as 'Emailing diff to foo@example.com').
 
671
            self.pb_stage_name = "Running %s hooks [%s]" % \
 
672
                (hook_name, Branch.hooks.get_hook_name(hook))
 
673
            self._emit_progress()
 
674
            if 'hooks' in debug.debug_flags:
 
675
                mutter("Invoking commit hook: %r", hook)
 
676
            if hook_name == "post_commit":
 
677
                hook(hook_local, hook_master, old_revno, old_revid, new_revno,
 
678
                     self.rev_id)
 
679
            elif hook_name == "pre_commit":
 
680
                hook(hook_local, hook_master,
 
681
                     old_revno, old_revid, new_revno, self.rev_id,
 
682
                     tree_delta, future_tree)
 
683
 
 
684
    def _update_builder_with_changes(self):
 
685
        """Update the commit builder with the data about what has changed.
 
686
        """
 
687
        specific_files = self.specific_files
 
688
        mutter("Selecting files for commit with filter %r", specific_files)
 
689
 
 
690
        self._check_strict()
 
691
        iter_changes = self.work_tree.iter_changes(
 
692
            self.basis_tree, specific_files=specific_files)
 
693
        if self.exclude:
 
694
            iter_changes = filter_excluded(iter_changes, self.exclude)
 
695
        iter_changes = self._filter_iter_changes(iter_changes)
 
696
        for path, fs_hash in self.builder.record_iter_changes(
 
697
                self.work_tree, self.basis_revid, iter_changes):
 
698
            self.work_tree._observed_sha1(path, fs_hash)
 
699
 
 
700
    def _filter_iter_changes(self, iter_changes):
 
701
        """Process iter_changes.
 
702
 
 
703
        This method reports on the changes in iter_changes to the user, and
 
704
        converts 'missing' entries in the iter_changes iterator to 'deleted'
 
705
        entries. 'missing' entries have their
 
706
 
 
707
        :param iter_changes: An iter_changes to process.
 
708
        :return: A generator of changes.
 
709
        """
 
710
        reporter = self.reporter
 
711
        report_changes = reporter.is_verbose()
 
712
        deleted_paths = []
 
713
        for change in iter_changes:
 
714
            if report_changes:
 
715
                old_path = change.path[0]
 
716
                new_path = change.path[1]
 
717
                versioned = change.versioned[1]
 
718
            kind = change.kind[1]
 
719
            versioned = change.versioned[1]
 
720
            if kind is None and versioned:
 
721
                # 'missing' path
 
722
                if report_changes:
 
723
                    reporter.missing(new_path)
 
724
                if change.kind[0] == 'symlink' and not self.work_tree.supports_symlinks():
 
725
                    trace.warning('Ignoring "%s" as symlinks are not '
 
726
                                  'supported on this filesystem.' % (change.path[0],))
533
727
                    continue
534
 
                else:
535
 
                    # this is selected, ensure its parents are too.
536
 
                    parent_id = new_ie.parent_id
537
 
                    while parent_id != ROOT_ID:
538
 
                        if not self.new_inv.has_id(parent_id):
539
 
                            ie = self._select_entry(self.work_inv[parent_id])
540
 
                            mutter('%s selected for commit because of %s',
541
 
                                   self.new_inv.id2path(parent_id), path)
542
 
 
543
 
                        ie = self.new_inv[parent_id]
544
 
                        if ie.revision is not None:
545
 
                            ie.revision = None
546
 
                            mutter('%s selected for commit because of %s',
547
 
                                   self.new_inv.id2path(parent_id), path)
548
 
                        parent_id = ie.parent_id
549
 
            mutter('%s selected for commit', path)
550
 
            self._select_entry(new_ie)
551
 
 
552
 
    def _select_entry(self, new_ie):
553
 
        """Make new_ie be considered for committing."""
554
 
        ie = new_ie.copy()
555
 
        ie.revision = None
556
 
        self.new_inv.add(ie)
557
 
        return ie
558
 
 
559
 
    def _carry_entry(self, file_id):
560
 
        """Carry the file unchanged from the basis revision."""
561
 
        if self.basis_inv.has_id(file_id):
562
 
            self.new_inv.add(self.basis_inv[file_id].copy())
563
 
 
564
 
    def _report_deletes(self):
565
 
        for file_id in self.basis_inv:
566
 
            if file_id not in self.new_inv:
567
 
                self.reporter.deleted(self.basis_inv.id2path(file_id))
568
 
 
569
 
def _gen_revision_id(config, when):
570
 
    """Return new revision-id."""
571
 
    s = '%s-%s-' % (config.user_email(), compact_date(when))
572
 
    s += hexlify(rand_bytes(8))
573
 
    return s
 
728
                deleted_paths.append(change[1][1])
 
729
                # Reset the new path (None) and new versioned flag (False)
 
730
                change = change.discard_new()
 
731
                new_path = change.path[1]
 
732
                versioned = False
 
733
            elif kind == 'tree-reference':
 
734
                if self.recursive == 'down':
 
735
                    self._commit_nested_tree(change[1][1])
 
736
            if change.versioned[0] or change.versioned[1]:
 
737
                yield change
 
738
                if report_changes:
 
739
                    if new_path is None:
 
740
                        reporter.deleted(old_path)
 
741
                    elif old_path is None:
 
742
                        reporter.snapshot_change(gettext('added'), new_path)
 
743
                    elif old_path != new_path:
 
744
                        reporter.renamed(gettext('renamed'),
 
745
                                         old_path, new_path)
 
746
                    else:
 
747
                        if (new_path
 
748
                                or self.work_tree.branch.repository._format.rich_root_data):
 
749
                            # Don't report on changes to '' in non rich root
 
750
                            # repositories.
 
751
                            reporter.snapshot_change(
 
752
                                gettext('modified'), new_path)
 
753
            self._next_progress_entry()
 
754
        # Unversion files that were found to be deleted
 
755
        self.deleted_paths = deleted_paths
 
756
 
 
757
    def _check_strict(self):
 
758
        # XXX: when we use iter_changes this would likely be faster if
 
759
        # iter_changes would check for us (even in the presence of
 
760
        # selected_files).
 
761
        if self.strict:
 
762
            # raise an exception as soon as we find a single unknown.
 
763
            for unknown in self.work_tree.unknowns():
 
764
                raise StrictCommitFailed()
 
765
 
 
766
    def _commit_nested_tree(self, path):
 
767
        "Commit a nested tree."
 
768
        sub_tree = self.work_tree.get_nested_tree(path)
 
769
        # FIXME: be more comprehensive here:
 
770
        # this works when both trees are in --trees repository,
 
771
        # but when both are bound to a different repository,
 
772
        # it fails; a better way of approaching this is to
 
773
        # finally implement the explicit-caches approach design
 
774
        # a while back - RBC 20070306.
 
775
        if sub_tree.branch.repository.has_same_location(
 
776
                self.work_tree.branch.repository):
 
777
            sub_tree.branch.repository = \
 
778
                self.work_tree.branch.repository
 
779
        try:
 
780
            return sub_tree.commit(message=None, revprops=self.revprops,
 
781
                                   recursive=self.recursive,
 
782
                                   message_callback=self.message_callback,
 
783
                                   timestamp=self.timestamp,
 
784
                                   timezone=self.timezone,
 
785
                                   committer=self.committer,
 
786
                                   allow_pointless=self.allow_pointless,
 
787
                                   strict=self.strict, verbose=self.verbose,
 
788
                                   local=self.local, reporter=self.reporter)
 
789
        except PointlessCommit:
 
790
            return self.work_tree.get_reference_revision(path)
 
791
 
 
792
    def _set_progress_stage(self, name, counter=False):
 
793
        """Set the progress stage and emit an update to the progress bar."""
 
794
        self.pb_stage_name = name
 
795
        self.pb_stage_count += 1
 
796
        if counter:
 
797
            self.pb_entries_count = 0
 
798
        else:
 
799
            self.pb_entries_count = None
 
800
        self._emit_progress()
 
801
 
 
802
    def _next_progress_entry(self):
 
803
        """Emit an update to the progress bar and increment the entry count."""
 
804
        self.pb_entries_count += 1
 
805
        self._emit_progress()
 
806
 
 
807
    def _emit_progress(self):
 
808
        if self.pb_entries_count is not None:
 
809
            text = gettext("{0} [{1}] - Stage").format(self.pb_stage_name,
 
810
                                                       self.pb_entries_count)
 
811
        else:
 
812
            text = gettext("%s - Stage") % (self.pb_stage_name, )
 
813
        self.pb.update(text, self.pb_stage_count, self.pb_stage_total)