~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commit.py

  • Committer: John Arbash Meinel
  • Date: 2006-09-10 20:39:26 UTC
  • mto: This revision was merged to the branch mainline in revision 2004.
  • Revision ID: john@arbash-meinel.com-20060910203926-ae731f6bb165d6fa
Adding a ScopeReplacer class, which can replace itself on demand

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
# Copyright (C) 2005, 2006 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
5
5
# the Free Software Foundation; either version 2 of the License, or
16
16
 
17
17
 
18
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.
 
19
# nothing?  
21
20
 
22
21
# TODO: Separate 'prepare' phase where we find a list of potentially
23
22
# committed files.  We then can then pause the commit to prompt for a
67
66
import re
68
67
import sys
69
68
import time
70
 
import pdb
71
69
 
72
 
from binascii import hexlify
73
70
from cStringIO import StringIO
74
71
 
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
72
import bzrlib.config
82
73
import bzrlib.errors as errors
83
74
from bzrlib.errors import (BzrError, PointlessCommit,
84
 
                           HistoryMissing,
85
75
                           ConflictsInTree,
86
76
                           StrictCommitFailed
87
77
                           )
88
 
from bzrlib.revision import Revision
 
78
from bzrlib.osutils import (kind_marker, isdir,isfile, is_inside_any, 
 
79
                            is_inside_or_parent_of_any,
 
80
                            quotefn, sha_file, split_lines)
89
81
from bzrlib.testament import Testament
90
82
from bzrlib.trace import mutter, note, warning
91
83
from bzrlib.xml5 import serializer_v5
92
84
from bzrlib.inventory import Inventory, ROOT_ID, InventoryEntry
93
 
from bzrlib.symbol_versioning import *
 
85
from bzrlib import symbol_versioning
 
86
from bzrlib.symbol_versioning import (deprecated_passed,
 
87
        deprecated_function,
 
88
        DEPRECATED_PARAMETER)
94
89
from bzrlib.workingtree import WorkingTree
95
90
 
96
91
 
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.
104
 
    """
105
 
    ## XXX: Remove this in favor of Branch.commit?
106
 
    Commit().commit(*args, **kwargs)
107
 
 
108
 
 
109
92
class NullCommitReporter(object):
110
93
    """I report on progress of a commit."""
111
94
 
225
208
        mutter('preparing to commit')
226
209
 
227
210
        if deprecated_passed(branch):
228
 
            warn("Commit.commit (branch, ...): The branch parameter is "
 
211
            symbol_versioning.warn("Commit.commit (branch, ...): The branch parameter is "
229
212
                 "deprecated as of bzr 0.8. Please use working_tree= instead.",
230
213
                 DeprecationWarning, stacklevel=2)
231
214
            self.branch = branch
238
221
        if message is None:
239
222
            raise BzrError("The message keyword parameter is required for commit().")
240
223
 
241
 
        self.weave_store = self.branch.repository.weave_store
242
224
        self.bound_branch = None
243
225
        self.local = local
244
226
        self.master_branch = None
245
227
        self.master_locked = False
246
 
        self.rev_id = rev_id
 
228
        self.rev_id = None
247
229
        self.specific_files = specific_files
248
230
        self.allow_pointless = allow_pointless
249
 
        self.revprops = {}
250
 
        if revprops is not None:
251
 
            self.revprops.update(revprops)
252
231
 
253
232
        if reporter is None and self.reporter is None:
254
233
            self.reporter = NullCommitReporter()
266
245
            self._check_bound_branch()
267
246
 
268
247
            # check for out of date working trees
269
 
            # if we are bound, then self.branch is the master branch and this
270
 
            # test is thus all we need.
271
 
            if self.work_tree.last_revision() != self.master_branch.last_revision():
 
248
            try:
 
249
                first_tree_parent = self.work_tree.get_parent_ids()[0]
 
250
            except IndexError:
 
251
                # if there are no parents, treat our parent as 'None'
 
252
                # this is so that we still consier the master branch
 
253
                # - in a checkout scenario the tree may have no
 
254
                # parents but the branch may do.
 
255
                first_tree_parent = None
 
256
            master_last = self.master_branch.last_revision()
 
257
            if (master_last is not None and
 
258
                master_last != first_tree_parent):
272
259
                raise errors.OutOfDateTree(self.work_tree)
273
260
    
274
261
            if strict:
275
262
                # raise an exception as soon as we find a single unknown.
276
263
                for unknown in self.work_tree.unknowns():
277
264
                    raise StrictCommitFailed()
278
 
    
279
 
            if timestamp is None:
280
 
                self.timestamp = time.time()
281
 
            else:
282
 
                self.timestamp = long(timestamp)
283
 
                
 
265
                   
284
266
            if self.config is None:
285
 
                self.config = bzrlib.config.BranchConfig(self.branch)
286
 
    
287
 
            if rev_id is None:
288
 
                self.rev_id = _gen_revision_id(self.config, self.timestamp)
289
 
            else:
290
 
                self.rev_id = rev_id
291
 
    
292
 
            if committer is None:
293
 
                self.committer = self.config.username()
294
 
            else:
295
 
                assert isinstance(committer, basestring), type(committer)
296
 
                self.committer = committer
297
 
    
298
 
            if timezone is None:
299
 
                self.timezone = local_time_offset()
300
 
            else:
301
 
                self.timezone = int(timezone)
302
 
    
 
267
                self.config = self.branch.get_config()
 
268
      
303
269
            if isinstance(message, str):
304
270
                message = message.decode(bzrlib.user_encoding)
305
271
            assert isinstance(message, unicode), type(message)
314
280
            # note that this estimate is too long when we do a partial tree
315
281
            # commit which excludes some new files from being considered.
316
282
            # The estimate is corrected when we populate the new inv.
317
 
            self.pb_total = len(self.basis_inv) + len(self.work_inv) + 3 - 1
 
283
            self.pb_total = len(self.work_inv) + 5
318
284
            self.pb_count = 0
319
285
 
320
286
            self._gather_parents()
321
287
            if len(self.parents) > 1 and self.specific_files:
322
288
                raise NotImplementedError('selected-file commit of merges is not supported yet: files %r',
323
289
                        self.specific_files)
324
 
            self._check_parents_present()
 
290
            
 
291
            self.builder = self.branch.get_commit_builder(self.parents, 
 
292
                self.config, timestamp, timezone, committer, revprops, rev_id)
325
293
            
326
294
            self._remove_deleted()
327
295
            self._populate_new_inv()
328
 
            self._store_snapshot()
329
296
            self._report_deletes()
330
297
 
331
298
            if not (self.allow_pointless
332
299
                    or len(self.parents) > 1
333
 
                    or self.new_inv != self.basis_inv):
 
300
                    or self.builder.new_inventory != self.basis_inv):
334
301
                raise PointlessCommit()
335
302
 
336
303
            self._emit_progress_update()
337
 
            self.inv_sha1 = self.branch.repository.add_inventory(
338
 
                self.rev_id,
339
 
                self.new_inv,
340
 
                self.present_parents
341
 
                )
342
 
            self._emit_progress_update()
343
 
            self._make_revision()
 
304
            # TODO: Now the new inventory is known, check for conflicts and
 
305
            # prompt the user for a commit message.
 
306
            # ADHB 2006-08-08: If this is done, populate_new_inv should not add
 
307
            # weave lines, because nothing should be recorded until it is known
 
308
            # that commit will succeed.
 
309
            self.builder.finish_inventory()
 
310
            self._emit_progress_update()
 
311
            self.rev_id = self.builder.commit(self.message)
 
312
            self._emit_progress_update()
344
313
            # revision data is in the local branch now.
345
314
            
346
315
            # upload revision data to the master.
347
 
            # this will propogate merged revisions too if needed.
 
316
            # this will propagate merged revisions too if needed.
348
317
            if self.bound_branch:
349
318
                self.master_branch.repository.fetch(self.branch.repository,
350
319
                                                    revision_id=self.rev_id)
356
325
            # and now do the commit locally.
357
326
            self.branch.append_revision(self.rev_id)
358
327
 
359
 
            self.work_tree.set_pending_merges([])
360
 
            self.work_tree.set_last_revision(self.rev_id)
 
328
            # if the builder gave us the revisiontree it created back, we
 
329
            # could use it straight away here.
 
330
            # TODO: implement this.
 
331
            self.work_tree.set_parent_trees([(self.rev_id,
 
332
                self.branch.repository.revision_tree(self.rev_id))])
361
333
            # now the work tree is up to date with the branch
362
334
            
363
335
            self.reporter.completed(self.branch.revno(), self.rev_id)
372
344
            self._emit_progress_update()
373
345
        finally:
374
346
            self._cleanup()
 
347
        return self.rev_id
375
348
 
376
349
    def _check_bound_branch(self):
377
350
        """Check to see if the local branch is bound.
414
387
        self.bound_branch = self.branch
415
388
        self.master_branch.lock_write()
416
389
        self.master_locked = True
417
 
####        
418
 
####        # Check to see if we have any pending merges. If we do
419
 
####        # those need to be pushed into the master branch
420
 
####        pending_merges = self.work_tree.pending_merges()
421
 
####        if pending_merges:
422
 
####            for revision_id in pending_merges:
423
 
####                self.master_branch.repository.fetch(self.bound_branch.repository,
424
 
####                                                    revision_id=revision_id)
425
390
 
426
391
    def _cleanup(self):
427
392
        """Cleanup any open locks, progress bars etc."""
439
404
            except Exception, e:
440
405
                found_exception = e
441
406
        if found_exception is not None: 
442
 
            # dont do a plan raise, because the last exception may have been
 
407
            # don't do a plan raise, because the last exception may have been
443
408
            # trashed, e is our sure-to-work exception even though it loses the
444
409
            # full traceback. XXX: RBC 20060421 perhaps we could check the
445
410
            # exc_info and if its the same one do a plain raise otherwise 
478
443
 
479
444
    def _gather_parents(self):
480
445
        """Record the parents of a merge for merge detection."""
481
 
        pending_merges = self.work_tree.pending_merges()
482
 
        self.parents = []
 
446
        # TODO: Make sure that this list doesn't contain duplicate 
 
447
        # entries and the order is preserved when doing this.
 
448
        self.parents = self.work_tree.get_parent_ids()
483
449
        self.parent_invs = []
484
 
        self.present_parents = []
485
 
        precursor_id = self.branch.last_revision()
486
 
        if precursor_id:
487
 
            self.parents.append(precursor_id)
488
 
        self.parents += pending_merges
489
450
        for revision in self.parents:
490
451
            if self.branch.repository.has_revision(revision):
 
452
                mutter('commit parent revision {%s}', revision)
491
453
                inventory = self.branch.repository.get_inventory(revision)
492
454
                self.parent_invs.append(inventory)
493
 
                self.present_parents.append(revision)
494
 
 
495
 
    def _check_parents_present(self):
496
 
        for parent_id in self.parents:
497
 
            mutter('commit parent revision {%s}', parent_id)
498
 
            if not self.branch.repository.has_revision(parent_id):
499
 
                if parent_id == self.branch.last_revision():
500
 
                    warning("parent is missing %r", parent_id)
501
 
                    raise HistoryMissing(self.branch, 'revision', parent_id)
502
 
                else:
503
 
                    mutter("commit will ghost revision %r", parent_id)
504
 
            
505
 
    def _make_revision(self):
506
 
        """Record a new revision object for this commit."""
507
 
        rev = Revision(timestamp=self.timestamp,
508
 
                       timezone=self.timezone,
509
 
                       committer=self.committer,
510
 
                       message=self.message,
511
 
                       inventory_sha1=self.inv_sha1,
512
 
                       revision_id=self.rev_id,
513
 
                       properties=self.revprops)
514
 
        rev.parent_ids = self.parents
515
 
        self.branch.repository.add_revision(self.rev_id, rev, self.new_inv, self.config)
 
455
            else:
 
456
                mutter('commit parent ghost revision {%s}', revision)
516
457
 
517
458
    def _remove_deleted(self):
518
459
        """Remove deleted files from the working inventories.
527
468
        """
528
469
        specific = self.specific_files
529
470
        deleted_ids = []
 
471
        deleted_paths = set()
530
472
        for path, ie in self.work_inv.iter_entries():
 
473
            if is_inside_any(deleted_paths, path):
 
474
                # The tree will delete the required ids recursively.
 
475
                continue
531
476
            if specific and not is_inside_any(specific, path):
532
477
                continue
533
478
            if not self.work_tree.has_filename(path):
 
479
                deleted_paths.add(path)
534
480
                self.reporter.missing(path)
535
 
                deleted_ids.append((path, ie.file_id))
536
 
        if deleted_ids:
537
 
            deleted_ids.sort(reverse=True)
538
 
            for path, file_id in deleted_ids:
539
 
                del self.work_inv[file_id]
540
 
            self.work_tree._write_inventory(self.work_inv)
541
 
 
542
 
    def _store_snapshot(self):
543
 
        """Pass over inventory and record a snapshot.
544
 
 
545
 
        Entries get a new revision when they are modified in 
546
 
        any way, which includes a merge with a new set of
547
 
        parents that have the same entry. 
548
 
        """
549
 
        # XXX: Need to think more here about when the user has
550
 
        # made a specific decision on a particular value -- c.f.
551
 
        # mark-merge.  
552
 
 
553
 
        # iter_entries does not visit the ROOT_ID node so we need to call
554
 
        # self._emit_progress_update once by hand.
555
 
        self._emit_progress_update()
556
 
        for path, ie in self.new_inv.iter_entries():
557
 
            self._emit_progress_update()
558
 
            previous_entries = ie.find_previous_heads(
559
 
                self.parent_invs,
560
 
                self.weave_store,
561
 
                self.branch.repository.get_transaction())
562
 
            if ie.revision is None:
563
 
                # we are creating a new revision for ie in the history store
564
 
                # and inventory.
565
 
                ie.snapshot(self.rev_id, path, previous_entries,
566
 
                    self.work_tree, self.weave_store,
567
 
                    self.branch.repository.get_transaction())
568
 
            # describe the nature of the change that has occured relative to
569
 
            # the basis inventory.
570
 
            if (self.basis_inv.has_id(ie.file_id)):
571
 
                basis_ie = self.basis_inv[ie.file_id]
572
 
            else:
573
 
                basis_ie = None
574
 
            change = ie.describe_change(basis_ie, ie)
575
 
            if change in (InventoryEntry.RENAMED, 
576
 
                InventoryEntry.MODIFIED_AND_RENAMED):
577
 
                old_path = self.basis_inv.id2path(ie.file_id)
578
 
                self.reporter.renamed(change, old_path, path)
579
 
            else:
580
 
                self.reporter.snapshot_change(change, path)
 
481
                deleted_ids.append(ie.file_id)
 
482
        self.work_tree.unversion(deleted_ids)
581
483
 
582
484
    def _populate_new_inv(self):
583
485
        """Build revision inventory.
589
491
        None; inventory entries that are carried over untouched have their
590
492
        revision set to their prior value.
591
493
        """
 
494
        # ESEPARATIONOFCONCERNS: this function is diffing and using the diff
 
495
        # results to create a new inventory at the same time, which results
 
496
        # in bugs like #46635.  Any reason not to use/enhance Tree.changes_from?
 
497
        # ADHB 11-07-2006
592
498
        mutter("Selecting files for commit with filter %s", self.specific_files)
593
 
        self.new_inv = Inventory(revision_id=self.rev_id)
594
 
        # iter_entries does not visit the ROOT_ID node so we need to call
595
 
        # self._emit_progress_update once by hand.
596
 
        self._emit_progress_update()
597
 
        for path, new_ie in self.work_inv.iter_entries():
 
499
        entries = self.work_inv.iter_entries()
 
500
        if not self.builder.record_root_entry:
 
501
            symbol_versioning.warn('CommitBuilders should support recording'
 
502
                ' the root entry as of bzr 0.10.', DeprecationWarning, 
 
503
                stacklevel=1)
 
504
            self.builder.new_inventory.add(self.basis_inv.root.copy())
 
505
            entries.next()
 
506
            self._emit_progress_update()
 
507
        for path, new_ie in entries:
598
508
            self._emit_progress_update()
599
509
            file_id = new_ie.file_id
600
 
            mutter('check %s {%s}', path, new_ie.file_id)
601
 
            if self.specific_files:
602
 
                if not is_inside_any(self.specific_files, path):
603
 
                    mutter('%s not selected for commit', path)
604
 
                    self._carry_entry(file_id)
 
510
            # mutter('check %s {%s}', path, file_id)
 
511
            if (not self.specific_files or 
 
512
                is_inside_or_parent_of_any(self.specific_files, path)):
 
513
                    # mutter('%s selected for commit', path)
 
514
                    ie = new_ie.copy()
 
515
                    ie.revision = None
 
516
            else:
 
517
                # mutter('%s not selected for commit', path)
 
518
                if self.basis_inv.has_id(file_id):
 
519
                    ie = self.basis_inv[file_id].copy()
 
520
                else:
 
521
                    # this entry is new and not being committed
605
522
                    continue
606
 
                else:
607
 
                    # this is selected, ensure its parents are too.
608
 
                    parent_id = new_ie.parent_id
609
 
                    while parent_id != ROOT_ID:
610
 
                        if not self.new_inv.has_id(parent_id):
611
 
                            ie = self._select_entry(self.work_inv[parent_id])
612
 
                            mutter('%s selected for commit because of %s',
613
 
                                   self.new_inv.id2path(parent_id), path)
614
 
 
615
 
                        ie = self.new_inv[parent_id]
616
 
                        if ie.revision is not None:
617
 
                            ie.revision = None
618
 
                            mutter('%s selected for commit because of %s',
619
 
                                   self.new_inv.id2path(parent_id), path)
620
 
                        parent_id = ie.parent_id
621
 
            mutter('%s selected for commit', path)
622
 
            self._select_entry(new_ie)
 
523
 
 
524
            self.builder.record_entry_contents(ie, self.parent_invs, 
 
525
                path, self.work_tree)
 
526
            # describe the nature of the change that has occurred relative to
 
527
            # the basis inventory.
 
528
            if (self.basis_inv.has_id(ie.file_id)):
 
529
                basis_ie = self.basis_inv[ie.file_id]
 
530
            else:
 
531
                basis_ie = None
 
532
            change = ie.describe_change(basis_ie, ie)
 
533
            if change in (InventoryEntry.RENAMED, 
 
534
                InventoryEntry.MODIFIED_AND_RENAMED):
 
535
                old_path = self.basis_inv.id2path(ie.file_id)
 
536
                self.reporter.renamed(change, old_path, path)
 
537
            else:
 
538
                self.reporter.snapshot_change(change, path)
 
539
 
 
540
        if not self.specific_files:
 
541
            return
 
542
 
 
543
        # ignore removals that don't match filespec
 
544
        for path, new_ie in self.basis_inv.iter_entries():
 
545
            if new_ie.file_id in self.work_inv:
 
546
                continue
 
547
            if is_inside_any(self.specific_files, path):
 
548
                continue
 
549
            ie = new_ie.copy()
 
550
            ie.revision = None
 
551
            self.builder.record_entry_contents(ie, self.parent_invs, path,
 
552
                                               self.basis_tree)
623
553
 
624
554
    def _emit_progress_update(self):
625
555
        """Emit an update to the progress bar."""
626
556
        self.pb.update("Committing", self.pb_count, self.pb_total)
627
557
        self.pb_count += 1
628
558
 
629
 
    def _select_entry(self, new_ie):
630
 
        """Make new_ie be considered for committing."""
631
 
        ie = new_ie.copy()
632
 
        ie.revision = None
633
 
        self.new_inv.add(ie)
634
 
        return ie
635
 
 
636
 
    def _carry_entry(self, file_id):
637
 
        """Carry the file unchanged from the basis revision."""
638
 
        if self.basis_inv.has_id(file_id):
639
 
            self.new_inv.add(self.basis_inv[file_id].copy())
640
 
        else:
641
 
            # this entry is new and not being committed
642
 
            self.pb_total -= 1
643
 
 
644
559
    def _report_deletes(self):
645
560
        for path, ie in self.basis_inv.iter_entries():
646
 
            if ie.file_id not in self.new_inv:
 
561
            if ie.file_id not in self.builder.new_inventory:
647
562
                self.reporter.deleted(path)
648
563
 
649
 
def _gen_revision_id(config, when):
650
 
    """Return new revision-id."""
651
 
    s = '%s-%s-' % (config.user_email(), compact_date(when))
652
 
    s += hexlify(rand_bytes(8))
653
 
    return s
 
564