~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Aaron Bentley
  • Date: 2007-12-27 16:30:52 UTC
  • Revision ID: abentley@panoramicfeedback.com-20071227163052-l4wmn2vsl91nbfjh
Remove test due to intentional behavior change

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Aaron Bentley <aaron.bentley@utoronto.ca>
 
1
# Copyright (C) 2005, 2006, 2007 Aaron Bentley <aaron.bentley@utoronto.ca>
2
2
# Copyright (C) 2005, 2006 Canonical Limited.
3
3
# Copyright (C) 2006 Michael Ellerman.
4
4
#
22
22
 
23
23
import bzrlib
24
24
 
25
 
 
26
 
__version__ = '0.14.0'
27
 
 
28
 
 
29
 
version_info = tuple(int(n) for n in __version__.split('.'))
30
 
 
31
 
 
32
 
def check_bzrlib_version(desired):
33
 
    """Check that bzrlib is compatible.
34
 
 
35
 
    If version is < bzrtools version, assume incompatible.
36
 
    If version == bzrtools version, assume completely compatible
37
 
    If version == bzrtools version + 1, assume compatible, with deprecations
38
 
    Otherwise, assume incompatible.
39
 
    """
40
 
    desired_plus = (desired[0], desired[1]+1)
41
 
    bzrlib_version = bzrlib.version_info[:2]
42
 
    if bzrlib_version == desired:
43
 
        return
44
 
    try:
45
 
        from bzrlib.trace import warning
46
 
    except ImportError:
47
 
        # get the message out any way we can
48
 
        from warnings import warn as warning
49
 
    if bzrlib_version < desired:
50
 
        warning('Installed bzr version %s is too old to be used with bzrtools'
51
 
                ' %s.' % (bzrlib.__version__, __version__))
52
 
        # Not using BzrNewError, because it may not exist.
53
 
        raise Exception, ('Version mismatch', version_info)
54
 
    else:
55
 
        warning('Bzrtools is not up to date with installed bzr version %s.'
56
 
                ' \nThere should be a newer version available, e.g. %i.%i.' 
57
 
                % (bzrlib.__version__, bzrlib_version[0], bzrlib_version[1]))
58
 
        if bzrlib_version != desired_plus:
59
 
            raise Exception, 'Version mismatch'
60
 
 
61
 
 
62
 
check_bzrlib_version(version_info[:2])
63
 
 
64
 
 
 
25
from bzrlib.lazy_import import lazy_import
 
26
lazy_import(globals(), """
 
27
from bzrlib import help, urlutils
 
28
import shelf
 
29
""")
 
30
 
 
31
from version import version_info, __version__
 
32
from command import BzrToolsCommand
65
33
from errors import CommandError, NoPyBaz
66
34
from patchsource import BzrPatchSource
67
 
from shelf import Shelf
68
35
import sys
69
36
import os.path
70
37
 
71
38
import bzrlib.builtins
72
39
import bzrlib.commands
 
40
from bzrlib.branch import Branch
 
41
from bzrlib.bzrdir import BzrDir
73
42
from bzrlib.commands import get_cmd_object
74
43
from bzrlib.errors import BzrCommandError
75
 
from bzrlib.help import command_usage
76
44
import bzrlib.ignores
 
45
from bzrlib.trace import note
77
46
from bzrlib.option import Option
78
 
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__), 
 
47
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
79
48
                                                 "external")))
80
49
 
81
50
import show_paths
 
51
from command import BzrToolsCommand
82
52
 
83
53
bzrlib.ignores.add_runtime_ignores(['./.shelf'])
84
54
 
85
55
 
86
 
class cmd_clean_tree(bzrlib.commands.Command):
 
56
class cmd_clean_tree(BzrToolsCommand):
87
57
    """Remove unwanted files from working tree.
88
58
 
89
59
    By default, only unknown files, not ignored files, are deleted.  Versioned
98
68
 
99
69
    To check what clean-tree will do, use --dry-run.
100
70
    """
101
 
    takes_options = [Option('ignored', help='delete all ignored files.'), 
102
 
                     Option('detritus', help='delete conflict files, merge'
103
 
                            ' backups, and failed selftest dirs.'), 
104
 
                     Option('unknown', 
105
 
                            help='delete files unknown to bzr.  (default)'),
106
 
                     Option('dry-run', help='show files to delete instead of'
 
71
    takes_options = [Option('ignored', help='Delete all ignored files.'),
 
72
                     Option('detritus', help='Delete conflict files, merge'
 
73
                            ' backups, and failed selftest dirs.'),
 
74
                     Option('unknown',
 
75
                            help='Delete files unknown to bzr (default).'),
 
76
                     Option('dry-run', help='Show files to delete instead of'
107
77
                            ' deleting them.')]
108
78
    def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
109
79
        from clean_tree import clean_tree
113
83
                   dry_run=dry_run)
114
84
 
115
85
 
116
 
class cmd_graph_ancestry(bzrlib.commands.Command):
 
86
class cmd_graph_ancestry(BzrToolsCommand):
117
87
    """Produce ancestry graphs using dot.
118
88
    
119
89
    Output format is detected according to file extension.  Some of the more
125
95
    with the last 5 characters of their revision identifier are used instead.
126
96
 
127
97
    The value starting with d is "(maximum) distance from the null revision".
128
 
    
 
98
 
129
99
    If --merge-branch is specified, the two branches are compared and a merge
130
100
    base is selected.
131
 
    
 
101
 
132
102
    Legend:
133
103
    white    normal revision
134
104
    yellow   THIS  history
148
118
    If available, rsvg is used to antialias PNG and JPEG output, but this can
149
119
    be disabled with --no-antialias.
150
120
    """
151
 
    takes_args = ['branch', 'file']
152
 
    takes_options = [Option('no-collapse', help="Do not skip simple nodes"), 
 
121
    takes_args = ['file', 'merge_branch?']
 
122
    takes_options = [Option('no-collapse', help="Do not skip simple nodes."),
153
123
                     Option('no-antialias',
154
 
                     help="Do not use rsvg to produce antialiased output"), 
155
 
                     Option('merge-branch', type=str, 
156
 
                     help="Use this branch to calcuate a merge base"), 
157
 
                     Option('cluster', help="Use clustered output.")]
158
 
    def run(self, branch, file, no_collapse=False, no_antialias=False,
159
 
        merge_branch=None, cluster=False):
 
124
                     help="Do not use rsvg to produce antialiased output."),
 
125
                     Option('merge-branch', type=str,
 
126
                     help="Use this branch to calcuate a merge base."),
 
127
                     Option('cluster', help="Use clustered output."),
 
128
                     Option('max-distance',
 
129
                            help="Show no nodes farther than this.", type=int),
 
130
                     Option('directory',
 
131
                            help='Source branch to use (default is current'
 
132
                            ' directory).',
 
133
                            short_name='d',
 
134
                            type=unicode),
 
135
                    ]
 
136
    def run(self, file, merge_branch=None, no_collapse=False,
 
137
            no_antialias=False, cluster=False, max_distance=100,
 
138
            directory='.'):
 
139
        if max_distance == -1:
 
140
            max_distance = None
160
141
        import graph
161
142
        if cluster:
162
143
            ranking = "cluster"
163
144
        else:
164
145
            ranking = "forced"
165
 
        graph.write_ancestry_file(branch, file, not no_collapse, 
166
 
                                  not no_antialias, merge_branch, ranking)
167
 
 
168
 
 
169
 
class cmd_fetch_ghosts(bzrlib.commands.Command):
 
146
        graph.write_ancestry_file(directory, file, not no_collapse,
 
147
                                  not no_antialias, merge_branch, ranking,
 
148
                                  max_distance=max_distance)
 
149
 
 
150
 
 
151
class cmd_fetch_ghosts(BzrToolsCommand):
170
152
    """Attempt to retrieve ghosts from another branch.
171
153
    If the other branch is not supplied, the last-pulled branch is used.
172
154
    """
173
155
    aliases = ['fetch-missing']
174
156
    takes_args = ['branch?']
175
 
    takes_options = [Option('no-fix')]
 
157
    takes_options = [Option('no-fix', help="Skip additional synchonization.")]
176
158
    def run(self, branch=None, no_fix=False):
177
159
        from fetch_ghosts import fetch_ghosts
178
160
        fetch_ghosts(branch, no_fix)
181
163
each file name found in the patch file."""
182
164
 
183
165
 
184
 
class cmd_patch(bzrlib.commands.Command):
 
166
class cmd_patch(BzrToolsCommand):
185
167
    """Apply a named patch to the current tree.
186
168
    """
187
169
    takes_args = ['filename?']
188
 
    takes_options = [Option('strip', type=int, help=strip_help)]
189
 
    def run(self, filename=None, strip=None):
 
170
    takes_options = [Option('strip', type=int, help=strip_help),
 
171
                     Option('silent', help='Suppress chatter.')]
 
172
    def run(self, filename=None, strip=None, silent=False):
190
173
        from patch import patch
191
174
        from bzrlib.workingtree import WorkingTree
192
175
        wt = WorkingTree.open_containing('.')[0]
193
176
        if strip is None:
194
177
            strip = 0
195
 
        return patch(wt, filename, strip)
196
 
 
197
 
 
198
 
class cmd_shelve(bzrlib.commands.Command):
 
178
        return patch(wt, filename, strip, silent)
 
179
 
 
180
 
 
181
class cmd_shelve(BzrToolsCommand):
199
182
    """Temporarily set aside some changes from the current tree.
200
183
 
201
184
    Shelve allows you to temporarily put changes you've made "on the shelf",
226
209
    """
227
210
 
228
211
    takes_args = ['file*']
229
 
    takes_options = ['message', 'revision',
230
 
            Option('all', help='Shelve all changes without prompting'), 
231
 
            Option('no-color', help='Never display changes in color')]
 
212
    takes_options = [Option('message',
 
213
            help='A message to associate with the shelved changes.',
 
214
            short_name='m', type=unicode),
 
215
            'revision',
 
216
            Option('all', help='Shelve all changes without prompting.'),
 
217
            Option('no-color', help='Never display changes in color.')]
232
218
 
233
219
    def run(self, all=False, file_list=None, message=None, revision=None,
234
220
            no_color=False):
240
226
                                  "parameter.")
241
227
 
242
228
        source = BzrPatchSource(revision, file_list)
243
 
        s = Shelf(source.base)
 
229
        s = shelf.Shelf(source.base)
244
230
        s.shelve(source, all, message, no_color)
245
231
        return 0
246
232
 
268
254
    aliases = ['switch']
269
255
    takes_args = ['othershelf']
270
256
    def run(self, othershelf):
271
 
        s = Shelf(self.shelf.base, othershelf)
 
257
        s = shelf.Shelf(self.shelf.base, othershelf)
272
258
        s.make_default()
273
259
 
274
260
 
287
273
        self.shelf.upgrade()
288
274
 
289
275
 
290
 
class cmd_shelf(bzrlib.commands.Command):
 
276
class cmd_shelf(BzrToolsCommand):
291
277
    """Perform various operations on your shelved patches. See also shelve."""
292
278
    takes_args = ['subcommand', 'args*']
293
279
 
301
287
            args_list = []
302
288
        cmd = self._get_cmd_object(subcommand)
303
289
        source = BzrPatchSource()
304
 
        s = Shelf(source.base)
 
290
        s = shelf.Shelf(source.base)
305
291
        cmd.shelf = s
 
292
 
 
293
        if args_list is None:
 
294
            args_list = []
306
295
        return cmd.run_argv_aliases(args_list)
307
296
 
308
297
    def _get_cmd_object(self, cmd_name):
325
314
        cmd_obj = cmd_class()
326
315
        indent = 2 * ' '
327
316
 
328
 
        usage = command_usage(cmd_obj)
 
317
        usage = cmd_obj._usage()
329
318
        usage = usage.replace('bzr shelf-', '')
330
319
        text.append('%s%s\n' % (indent, usage))
331
320
 
350
339
        return text
351
340
 
352
341
 
353
 
class cmd_unshelve(bzrlib.commands.Command):
 
342
class cmd_unshelve(BzrToolsCommand):
354
343
    """Restore shelved changes.
355
344
 
356
345
    By default the most recently shelved changes are restored. However if you
359
348
    See 'shelve' for more information.
360
349
    """
361
350
    takes_options = [
362
 
            Option('all', help='Unshelve all changes without prompting'),
363
 
            Option('force', help='Force unshelving even if errors occur'),
364
 
            Option('no-color', help='Never display changes in color')
 
351
            Option('all', help='Unshelve all changes without prompting.'),
 
352
            Option('force', help='Force unshelving even if errors occur.'),
 
353
            Option('no-color', help='Never display changes in color.')
365
354
        ]
366
355
    takes_args = ['patch?']
367
356
    def run(self, patch=None, all=False, force=False, no_color=False):
368
357
        source = BzrPatchSource()
369
 
        s = Shelf(source.base)
 
358
        s = shelf.Shelf(source.base)
370
359
        s.unshelve(source, patch, all, force, no_color)
371
360
        return 0
372
361
 
373
362
 
374
 
class cmd_shell(bzrlib.commands.Command):
 
363
class cmd_shell(BzrToolsCommand):
375
364
    """Begin an interactive shell tailored for bzr.
376
365
    Bzr commands can be used without typing bzr first, and will be run natively
377
366
    when possible.  Tab completion is tailored for bzr.  The shell prompt shows
394
383
        return shell.run_shell()
395
384
 
396
385
 
397
 
class cmd_branch_history(bzrlib.commands.Command):
 
386
class cmd_branch_history(BzrToolsCommand):
398
387
    """\
399
388
    Display the development history of a branch.
400
389
 
404
393
    """
405
394
    takes_args = ["branch?"]
406
395
    def run(self, branch=None):
407
 
        from branchhistory import branch_history 
 
396
        from branchhistory import branch_history
408
397
        return branch_history(branch)
409
398
 
410
399
 
411
 
class cmd_zap(bzrlib.commands.Command):
 
400
class cmd_zap(BzrToolsCommand):
412
401
    """\
413
402
    Remove a lightweight checkout, if it can be done safely.
414
403
 
419
408
    If --branch is specified, the branch will be deleted too, but only if the
420
409
    the branch has no new commits (relative to its parent).
421
410
    """
422
 
    takes_options = [Option("branch", help="Remove associtated branch from"
423
 
                                           " repository")]
 
411
    takes_options = [Option("branch", help="Remove associated branch from"
 
412
                                           " repository."),
 
413
                     Option('force', help='Delete tree even if contents are'
 
414
                     ' modified.')]
424
415
    takes_args = ["checkout"]
425
 
    def run(self, checkout, branch=False):
 
416
    def run(self, checkout, branch=False, force=False):
426
417
        from zap import zap
427
 
        return zap(checkout, remove_branch=branch)
428
 
 
429
 
 
430
 
class cmd_cbranch(bzrlib.commands.Command):
 
418
        return zap(checkout, remove_branch=branch, allow_modified=force)
 
419
 
 
420
 
 
421
class cmd_cbranch(BzrToolsCommand):
431
422
    """
432
423
    Create a new checkout, associated with a new repository branch.
433
 
    
434
 
    When you cbranch, bzr looks up the repository associated with your current
435
 
    directory in locations.conf.  It creates a new branch in that repository
436
 
    with the same name and relative path as the checkout you request.
437
 
 
438
 
    The locations.conf parameter is "cbranch_root".  So if you want 
439
 
    cbranch operations in /home/jrandom/bigproject to produce branches in 
440
 
    /home/jrandom/bigproject/repository, you'd add this:
441
 
 
442
 
    [/home/jrandom/bigproject]
443
 
    cbranch_root = /home/jrandom/bigproject/repository
444
 
 
445
 
    Note that if "/home/jrandom/bigproject/repository" isn't a repository,
446
 
    standalone branches will be produced.  Standalone branches will also
447
 
    be produced if the source branch is in 0.7 format (or earlier).
 
424
 
 
425
    When you cbranch, bzr looks up a target location in locations.conf, and
 
426
    creates the branch there.
 
427
 
 
428
    In your locations.conf, add the following lines:
 
429
    [/working_directory_root]
 
430
    cbranch_target = /branch_root
 
431
    cbranch_target:policy = appendpath
 
432
 
 
433
    This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
 
434
    working directory root, the branch will be created in
 
435
    "/branch_root/foo/baz"
 
436
 
 
437
    NOTE: cbranch also supports "cbranch_root", but that behaviour is
 
438
    deprecated.
448
439
    """
449
 
    takes_options = [Option("lightweight", 
450
 
                            help="Create a lightweight checkout"), 'revision']
 
440
    takes_options = [Option("lightweight",
 
441
                            help="Create a lightweight checkout."), 'revision',
 
442
                     Option('files-from', type=unicode,
 
443
                            help='Accelerate checkout using files from this'
 
444
                                 ' tree.')]
451
445
    takes_args = ["source", "target?"]
452
 
    def run(self, source, target=None, lightweight=False, revision=None):
 
446
    def run(self, source, target=None, lightweight=False, revision=None,
 
447
            files_from=None):
453
448
        from cbranch import cbranch
454
 
        return cbranch(source, target, lightweight=lightweight, 
455
 
                       revision=revision)
456
 
 
457
 
 
458
 
class cmd_branches(bzrlib.commands.Command):
 
449
        return cbranch(source, target, lightweight=lightweight,
 
450
                       revision=revision, files_from=files_from)
 
451
 
 
452
 
 
453
class cmd_branches(BzrToolsCommand):
459
454
    """Scan a location for branches"""
460
455
    takes_args = ["location?"]
461
456
    def run(self, location=None):
462
457
        from branches import branches
463
458
        return branches(location)
464
459
 
 
460
class cmd_trees(BzrToolsCommand):
 
461
    """Scan a location for trees"""
 
462
    takes_args = ['location?']
 
463
    def run(self, location='.'):
 
464
        from bzrlib.workingtree import WorkingTree
 
465
        from bzrlib.transport import get_transport
 
466
        t = get_transport(location)
 
467
        for tree in WorkingTree.find_trees(location):
 
468
            self.outf.write('%s\n' % t.relpath(
 
469
                tree.bzrdir.root_transport.base))
465
470
 
466
 
class cmd_multi_pull(bzrlib.commands.Command):
 
471
class cmd_multi_pull(BzrToolsCommand):
467
472
    """Pull all the branches under a location, e.g. a repository.
468
 
    
 
473
 
469
474
    Both branches present in the directory and the branches of checkouts are
470
475
    pulled.
471
476
    """
472
477
    takes_args = ["location?"]
473
478
    def run(self, location=None):
474
 
        from bzrlib.branch import Branch
475
479
        from bzrlib.transport import get_transport
476
480
        from bzrtools import iter_branch_tree
477
481
        if location is None:
478
482
            location = '.'
479
483
        t = get_transport(location)
 
484
        possible_transports = []
480
485
        if not t.listable():
481
486
            print "Can't list this type of location."
482
487
            return 3
498
503
                relpath = base
499
504
            print "Pulling %s from %s" % (relpath, parent)
500
505
            try:
501
 
                pullable.pull(Branch.open(parent))
 
506
                branch_t = get_transport(parent, possible_transports)
 
507
                pullable.pull(Branch.open_from_transport(branch_t))
502
508
            except Exception, e:
503
509
                print e
504
510
 
505
511
 
506
 
class cmd_branch_mark(bzrlib.commands.Command):
507
 
    """
508
 
    Add, view or list branch markers <EXPERIMENTAL>
509
 
 
510
 
    To add a mark, do 'bzr branch-mark MARK'.
511
 
    To list marks, do 'bzr branch-mark' (this lists all marks for the branch's
512
 
    repository).
513
 
    To delete a mark, do 'bzr branch-mark --delete MARK'
514
 
 
515
 
    These marks can be used to track a branch's status.
516
 
    """
517
 
    takes_args = ['mark?', 'branch?']
518
 
    takes_options = [Option('delete', help='Delete this mark')]
519
 
    def run(self, mark=None, branch=None, delete=False):
520
 
        from branch_mark import branch_mark
521
 
        branch_mark(mark, branch, delete)
522
 
 
523
 
 
524
 
class cmd_import(bzrlib.commands.Command):
525
 
    """Import sources from a tarball
526
 
    
527
 
    This command will import a tarball into a bzr tree, replacing any versioned
528
 
    files already present.  If a directory is specified, it is used as the
529
 
    target.  If the directory does not exist, or is not versioned, it is
530
 
    created.
 
512
 
 
513
class cmd_import(BzrToolsCommand):
 
514
    """Import sources from a directory, tarball or zip file
 
515
 
 
516
    This command will import a directory, tarball or zip file into a bzr
 
517
    tree, replacing any versioned files already present.  If a directory is
 
518
    specified, it is used as the target.  If the directory does not exist, or
 
519
    is not versioned, it is created.
531
520
 
532
521
    Tarballs may be gzip or bzip2 compressed.  This is autodetected.
533
522
 
534
 
    If the tarball has a single root directory, that directory is stripped
535
 
    when extracting the tarball.
 
523
    If the tarball or zip has a single root directory, that directory is
 
524
    stripped when extracting the tarball.  This is not done for directories.
536
525
    """
537
 
    
 
526
 
538
527
    takes_args = ['source', 'tree?']
539
528
    def run(self, source, tree=None):
540
529
        from upstream_import import do_import
541
530
        do_import(source, tree)
542
531
 
543
532
 
544
 
class cmd_cdiff(bzrlib.commands.Command):
 
533
class cmd_cdiff(BzrToolsCommand):
545
534
    """A color version of bzr's diff"""
546
535
    takes_args = property(lambda x: get_cmd_object('diff').takes_args)
547
 
    takes_options = property(lambda x: get_cmd_object('diff').takes_options)
548
 
    def run(*args, **kwargs):
 
536
    takes_options = list(get_cmd_object('diff').takes_options) + [
 
537
        Option('check-style',
 
538
            help='Warn if trailing whitespace or spurious changes have been'
 
539
                 ' added.')]
 
540
 
 
541
    def run(self, check_style=False, *args, **kwargs):
549
542
        from colordiff import colordiff
550
 
        colordiff(*args, **kwargs)
551
 
 
552
 
 
553
 
class cmd_baz_import(bzrlib.commands.Command):
 
543
        colordiff(check_style, *args, **kwargs)
 
544
 
 
545
 
 
546
class cmd_baz_import(BzrToolsCommand):
554
547
    """Import an Arch or Baz archive into a bzr repository.
555
548
 
556
549
    This command should be used on local archives (or mirrors) only.  It is
557
550
    quite slow on remote archives.
558
 
    
559
 
    reuse_history allows you to specify any previous imports you 
 
551
 
 
552
    reuse_history allows you to specify any previous imports you
560
553
    have done of different archives, which this archive has branches
561
 
    tagged from. This will dramatically reduce the time to convert 
 
554
    tagged from. This will dramatically reduce the time to convert
562
555
    the archive as it will not have to convert the history already
563
556
    converted in that other branch.
564
557
 
577
570
    """
578
571
    takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
579
572
    takes_options = ['verbose', Option('prefixes', type=str,
580
 
                     help="Prefixes of branches to import, colon-separated"),
581
 
                     Option('encoding', type=str, 
 
573
                     help="Prefixes of branches to import, colon-separated."),
 
574
                     Option('encoding', type=str,
582
575
                     help='Force encoding to specified value.  See WARNING.')]
583
576
 
584
577
    def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
592
585
            print "This command is disabled.  Please install PyBaz."
593
586
 
594
587
 
595
 
class cmd_baz_import_branch(bzrlib.commands.Command):
 
588
class cmd_baz_import_branch(BzrToolsCommand):
596
589
    """Import an Arch or Baz branch into a bzr branch.
597
590
 
598
591
    WARNING: Encoding should not be specified unless necessary, because if you
603
596
    are incompatible.
604
597
    """
605
598
    takes_args = ['to_location', 'from_branch?', 'reuse_history*']
606
 
    takes_options = ['verbose', Option('max-count', type=int),
607
 
                     Option('encoding', type=str, 
 
599
    takes_options = ['verbose', 
 
600
                     Option('max-count', type=int, 
 
601
                     help='Maximim revisions to import at once.'),
 
602
                     Option('encoding', type=str,
608
603
                     help='Force encoding to specified value.  See WARNING.')]
609
604
 
610
605
    def run(self, to_location, from_branch=None, fast=False, max_count=None,
613
608
        from errors import NoPyBaz
614
609
        try:
615
610
            import baz_import
616
 
            baz_import.baz_import_branch(to_location, from_branch, fast, 
 
611
            baz_import.baz_import_branch(to_location, from_branch, fast,
617
612
                                         max_count, verbose, encoding, dry_run,
618
613
                                         reuse_history_list)
619
614
        except NoPyBaz:
620
615
            print "This command is disabled.  Please install PyBaz."
621
616
 
622
617
 
623
 
class cmd_rspush(bzrlib.commands.Command):
 
618
class cmd_rspush(BzrToolsCommand):
624
619
    """Upload this branch to another location using rsync.
625
620
 
626
 
    If no location is specified, the last-used location will be used.  To 
627
 
    prevent dirty trees from being uploaded, rspush will error out if there are 
628
 
    unknown files or local changes.  It will also error out if the upstream 
629
 
    directory is non-empty and not an earlier version of the branch. 
 
621
    If no location is specified, the last-used location will be used.  To
 
622
    prevent dirty trees from being uploaded, rspush will error out if there are
 
623
    unknown files or local changes.  It will also error out if the upstream
 
624
    directory is non-empty and not an earlier version of the branch.
630
625
    """
631
626
    takes_args = ['location?']
632
627
    takes_options = [Option('overwrite', help='Ignore differences between'
633
 
                            ' branches and overwrite unconditionally'),
 
628
                            ' branches and overwrite unconditionally.'),
634
629
                     Option('no-tree', help='Do not push the working tree,'
635
630
                            ' just the .bzr.')]
636
631
 
638
633
        from bzrlib import workingtree
639
634
        import bzrtools
640
635
        cur_branch = workingtree.WorkingTree.open_containing(".")[0]
641
 
        bzrtools.rspush(cur_branch, location, overwrite=overwrite, 
 
636
        bzrtools.rspush(cur_branch, location, overwrite=overwrite,
642
637
                      working_tree=not no_tree)
643
638
 
644
639
 
645
 
class cmd_switch(bzrlib.commands.Command):
646
 
    """Set the branch of a lightweight checkout and update."""
647
 
 
648
 
    takes_args = ['to_location']
649
 
 
650
 
    def run(self, to_location):
651
 
        from switch import cmd_switch
652
 
        cmd_switch().run(to_location)
653
 
 
654
 
 
655
640
commands = [
656
641
            cmd_baz_import,
657
642
            cmd_baz_import_branch,
658
643
            cmd_branches,
 
644
            cmd_trees,
659
645
            cmd_branch_history,
660
 
            cmd_branch_mark,
661
 
            cmd_cbranch,  
 
646
            cmd_cbranch,
662
647
            cmd_cdiff,
663
648
            cmd_clean_tree,
664
649
            cmd_fetch_ghosts,
667
652
            cmd_multi_pull,
668
653
            cmd_patch,
669
654
            cmd_rspush,
670
 
            cmd_shelf, 
 
655
            cmd_shelf,
671
656
            cmd_shell,
672
 
            cmd_shelve, 
673
 
            cmd_switch,
674
 
            cmd_unshelve, 
675
 
            cmd_zap,            
 
657
            cmd_shelve,
 
658
            cmd_unshelve,
 
659
            cmd_zap,
676
660
            ]
677
661
 
678
662
 
688
672
    from unittest import TestSuite
689
673
    import bzrtools
690
674
    import tests.clean_tree
691
 
    import upstream_import
 
675
    import tests.is_clean
 
676
    import tests.test_rspush
 
677
    import tests.upstream_import
692
678
    import zap
693
679
    import tests.blackbox
694
680
    import tests.shelf_tests
703
689
    result.addTest(tests.test_suite())
704
690
    result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))
705
691
    result.addTest(tests.blackbox.test_suite())
706
 
    result.addTest(upstream_import.test_suite())
 
692
    result.addTest(tests.upstream_import.test_suite())
707
693
    result.addTest(zap.test_suite())
 
694
    result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))
 
695
    result.addTest(TestLoader().loadTestsFromModule(tests.test_rspush))
708
696
    return result