← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to push.py

Make work with integration again.

Show diffs side-by-side

added added

removed removed

Lines of Context:
push.py
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
 
16
from bzrlib.branch import Branch
 
17
from bzrlib.commands import Command
 
18
from bzrlib.errors import MustUseDecorated, BzrOptionError
16
19
from bzrlib.option import Option
17
 
from bzrlib.workingtree import WorkingTree
18
20
import sys
19
21
import os
20
 
import bzrtools
21
 
 
22
 
from command import BzrToolsCommand
23
 
 
24
 
class cmd_rspush(BzrToolsCommand):
 
22
import bzrtools 
 
23
 
 
24
 
 
25
class cmd_push(Command):
25
26
    """Upload this branch to another location using rsync.
26
27
 
27
 
    If no location is specified, the last-used location will be used.  To
28
 
    prevent dirty trees from being uploaded, rspush will error out if there are
29
 
    unknown files or local changes.  It will also error out if the upstream
30
 
    directory is non-empty and not an earlier version of the branch.
 
28
    If no location is specified, the last-used location will be used.  To 
 
29
    prevent dirty trees from being uploaded, push will error out if there are 
 
30
    unknown files or local changes.  It will also error out if the upstream 
 
31
    directory is non-empty and not an earlier version of the branch. 
31
32
    """
 
33
    _original_command = None
32
34
    takes_args = ['location?']
33
35
    takes_options = [Option('overwrite', help='Ignore differences between'
34
 
                            ' branches and overwrite unconditionally'),
35
 
                     Option('no-tree', help='Do not push the working tree,'
36
 
                            ' just the .bzr.')]
 
36
                            ' branches and overwrite unconditionally')]
37
37
 
38
 
    def run(self, location=None, overwrite=False, no_tree=False):
39
 
        cur_branch = WorkingTree.open_containing(".")[0]
40
 
        bzrtools.rspush(cur_branch, location, overwrite=overwrite,
41
 
                      working_tree=not no_tree)
 
38
    def run(self, location=None, overwrite=False):
 
39
        cur_branch = Branch.open_containing(".")[0]
 
40
        bzrtools.push(cur_branch, location, overwrite=overwrite)
 
41
        
 
42
    def run_argv(self, argv):
 
43
        """Parse command line and run.
 
44
        
 
45
        If the command requests it, run the decorated version.
 
46
        """
 
47
        try:
 
48
            super(cmd_push, self).run_argv(list(argv))
 
49
        except (MustUseDecorated, BzrOptionError):
 
50
            if self._original_command is None:
 
51
                raise
 
52
            self._original_command().run_argv(argv)