← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to switch.py

  • Committer: Michael Ellerman
  • Date: 2005-11-29 07:12:26 UTC
  • mto: (0.3.1 shelf-dev) (325.1.2 bzrtools)
  • mto: This revision was merged to the branch mainline in revision 334.
  • Revision ID: michael@ellerman.id.au-20051129071226-a04b3f827880025d
Unshelve --pick was broken, because we deleted the whole patch, even when only
part of it was unshelved. So now if we unshelve part of a patch, the patch is
replaced with a new patch that has just the unshelved parts. That's a long way
of saying it does what you'd expect.

Implementing this required changing HunkSelector to return both the selected,
and unselected hunks (ie. patches to shelve, and patches to keep).

Show diffs side-by-side

added added

removed removed

Lines of Context:
switch.py
1
 
# Copyright (C) 2006 Canonical Limited.
2
 
# Authors: David Allouche <david@allouche.net>
3
 
#          Aaron Bentley <aaron.bentley@utoronto.ca>
4
 
#
5
 
#    This program is free software; you can redistribute it and/or modify
6
 
#    it under the terms of the GNU General Public License as published by
7
 
#    the Free Software Foundation; either version 2 of the License, or
8
 
#    (at your option) any later version.
9
 
#
10
 
#    This program is distributed in the hope that it will be useful,
11
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
#    GNU General Public License for more details.
14
 
#
15
 
#    You should have received a copy of the GNU General Public License
16
 
#    along with this program; if not, write to the Free Software
17
 
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
 
from bzrlib.errors import BzrCommandError
19
 
from bzrlib.commands import Command, register_command
20
 
from bzrlib.branch import Branch, BranchFormat
21
 
from bzrlib.bzrdir import BzrDir, BzrDirFormat
22
 
from bzrlib.transport import get_transport
23
 
from bzrlib.builtins import cmd_update
24
 
from bzrlib.trace import note
25
 
from bzrlib.workingtree import WorkingTree
26
 
 
27
 
class cmd_switch(Command):
28
 
    """Set the branch of a lightweight checkout and update."""
29
 
 
30
 
    takes_args = ['to_location']
31
 
 
32
 
    def run(self, to_location):
33
 
        to_branch = Branch.open(to_location)
34
 
        tree_location = '.'
35
 
        self._switch(tree_location, to_branch)
36
 
        # need to re-open tree to get the new branch
37
 
        tree = WorkingTree.open_containing(tree_location)[0]
38
 
        self._update(tree)
39
 
 
40
 
    def _switch(self, tree_location, to_branch):
41
 
        tree = WorkingTree.open_containing(tree_location)[0]
42
 
        tree.lock_write()
43
 
        try:
44
 
            self._check_switch_branch_format(tree_location)
45
 
            self.set_branch_location(tree.bzrdir, to_branch.base)
46
 
            note('Switched to branch: %s' % (to_branch.base,))
47
 
        finally:
48
 
            tree.unlock()
49
 
 
50
 
    def _update(self, tree):
51
 
        tree.lock_write()
52
 
        try:
53
 
            if tree.last_revision() == tree.branch.last_revision():
54
 
                assert tree.branch.get_master_branch() is None, (
55
 
                    "switch tried to update a fat checkout")
56
 
                note("Tree is up to date.")
57
 
                return
58
 
            tree.update()
59
 
            note('Updated to revision %d' %
60
 
                 (tree.branch.revision_id_to_revno(tree.last_revision()),))
61
 
        finally:
62
 
            tree.unlock()
63
 
 
64
 
    def _check_switch_branch_format(self, url):
65
 
        transport = get_transport(url)
66
 
        format = BzrDirFormat.find_format(transport)
67
 
        format_string = format.get_format_string()
68
 
        if not format_string.startswith("Bazaar-NG meta directory, "):
69
 
            raise BzrCommandError(
70
 
                'The switch command can only be used on a light checkout.\n'
71
 
                'Expected metadir, found %s at %s' % (
72
 
                format_string.strip(), transport.base))
73
 
        control = BzrDir.open_containing_from_transport(transport)[0]
74
 
        branch_format = BranchFormat.find_format(control)
75
 
        format_string = branch_format.get_format_string()
76
 
        if not format_string.startswith("Bazaar-NG Branch Reference Format "):
77
 
            raise BzrCommandError(
78
 
                'The switch command can only be used on a light checkout.\n'
79
 
                'Expected branch reference, found %s at %s' % (
80
 
                format_string.strip(), transport.base))
81
 
        if not format_string == "Bazaar-NG Branch Reference Format 1\n":
82
 
            raise BzrCommandError(
83
 
                'Unsupported: %r' % (format_string.strip(),))        
84
 
 
85
 
    def set_branch_location(self, control, location):
86
 
        """Set location value of a branch reference.
87
 
 
88
 
        The branch living in the control (BzrDir) object must be locked for
89
 
        writing.
90
 
 
91
 
        :param control: BzrDir containing the branch reference
92
 
        :param location: value to write to the branch reference location.
93
 
        """
94
 
        branch_format = BranchFormat.find_format(control)
95
 
        transport = control.get_branch_transport(None)
96
 
        branch = branch_format.open(control)
97
 
        location = transport.put('location', location)