~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Aaron Bentley
  • Date: 2008-10-06 23:24:16 UTC
  • Revision ID: aaron@aaronbentley.com-20081006232416-mbohnubye0z5tiev
Add target_file param

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@aaronbentley.com>
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
 
 
65
 
from errors import CommandError, NoPyBaz
 
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
 
33
from errors import CommandError
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__), 
79
 
                                                 "external")))
80
47
 
81
 
import show_paths
 
48
from command import BzrToolsCommand
82
49
 
83
50
bzrlib.ignores.add_runtime_ignores(['./.shelf'])
84
51
 
85
52
 
86
 
class cmd_clean_tree(bzrlib.commands.Command):
 
53
class cmd_clean_tree(BzrToolsCommand):
87
54
    """Remove unwanted files from working tree.
88
55
 
89
56
    By default, only unknown files, not ignored files, are deleted.  Versioned
98
65
 
99
66
    To check what clean-tree will do, use --dry-run.
100
67
    """
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'
 
68
    takes_options = [Option('ignored', help='Delete all ignored files.'),
 
69
                     Option('detritus', help='Delete conflict files, merge'
 
70
                            ' backups, and failed selftest dirs.'),
 
71
                     Option('unknown',
 
72
                            help='Delete files unknown to bzr (default).'),
 
73
                     Option('dry-run', help='Show files to delete instead of'
107
74
                            ' deleting them.')]
108
75
    def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
109
76
        from clean_tree import clean_tree
113
80
                   dry_run=dry_run)
114
81
 
115
82
 
116
 
class cmd_graph_ancestry(bzrlib.commands.Command):
 
83
class cmd_graph_ancestry(BzrToolsCommand):
117
84
    """Produce ancestry graphs using dot.
118
85
    
119
86
    Output format is detected according to file extension.  Some of the more
125
92
    with the last 5 characters of their revision identifier are used instead.
126
93
 
127
94
    The value starting with d is "(maximum) distance from the null revision".
128
 
    
 
95
 
129
96
    If --merge-branch is specified, the two branches are compared and a merge
130
97
    base is selected.
131
 
    
 
98
 
132
99
    Legend:
133
100
    white    normal revision
134
101
    yellow   THIS  history
148
115
    If available, rsvg is used to antialias PNG and JPEG output, but this can
149
116
    be disabled with --no-antialias.
150
117
    """
151
 
    takes_args = ['branch', 'file']
152
 
    takes_options = [Option('no-collapse', help="Do not skip simple nodes"), 
 
118
    takes_args = ['file', 'merge_branch?']
 
119
    takes_options = [Option('no-collapse', help="Do not skip simple nodes."),
153
120
                     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):
 
121
                     help="Do not use rsvg to produce antialiased output."),
 
122
                     Option('merge-branch', type=str,
 
123
                     help="Use this branch to calcuate a merge base."),
 
124
                     Option('cluster', help="Use clustered output."),
 
125
                     Option('max-distance',
 
126
                            help="Show no nodes farther than this.", type=int),
 
127
                     Option('directory',
 
128
                            help='Source branch to use (default is current'
 
129
                            ' directory).',
 
130
                            short_name='d',
 
131
                            type=unicode),
 
132
                    ]
 
133
    def run(self, file, merge_branch=None, no_collapse=False,
 
134
            no_antialias=False, cluster=False, max_distance=100,
 
135
            directory='.'):
 
136
        if max_distance == -1:
 
137
            max_distance = None
160
138
        import graph
161
139
        if cluster:
162
140
            ranking = "cluster"
163
141
        else:
164
142
            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):
 
143
        graph.write_ancestry_file(directory, file, not no_collapse,
 
144
                                  not no_antialias, merge_branch, ranking,
 
145
                                  max_distance=max_distance)
 
146
 
 
147
 
 
148
class cmd_fetch_ghosts(BzrToolsCommand):
170
149
    """Attempt to retrieve ghosts from another branch.
171
150
    If the other branch is not supplied, the last-pulled branch is used.
172
151
    """
173
152
    aliases = ['fetch-missing']
174
153
    takes_args = ['branch?']
175
 
    takes_options = [Option('no-fix')]
 
154
    takes_options = [Option('no-fix', help="Skip additional synchonization.")]
176
155
    def run(self, branch=None, no_fix=False):
177
156
        from fetch_ghosts import fetch_ghosts
178
157
        fetch_ghosts(branch, no_fix)
179
158
 
180
159
strip_help="""Strip the smallest prefix containing num leading slashes  from \
181
160
each file name found in the patch file."""
182
 
Option.OPTIONS['bzrdiff'] = Option('bzrdiff',type=None,
183
 
                                help="""Handle extra bzr tags""")
184
 
 
185
 
 
186
 
class cmd_patch(bzrlib.commands.Command):
 
161
 
 
162
 
 
163
class cmd_patch(BzrToolsCommand):
187
164
    """Apply a named patch to the current tree.
188
165
    """
189
166
    takes_args = ['filename?']
190
 
    takes_options = [Option('strip', type=int, help=strip_help)]
191
 
    def run(self, filename=None, strip=-1, bzrdiff=0):
 
167
    takes_options = [Option('strip', type=int, help=strip_help),
 
168
                     Option('silent', help='Suppress chatter.')]
 
169
    def run(self, filename=None, strip=None, silent=False):
192
170
        from patch import patch
193
171
        from bzrlib.workingtree import WorkingTree
194
172
        wt = WorkingTree.open_containing('.')[0]
195
 
        if strip == -1:
196
 
            if bzrdiff: strip = 0
197
 
            else:       strip = 0
198
 
 
199
 
        return patch(wt, filename, strip, legacy= not bzrdiff)
200
 
 
201
 
 
202
 
class cmd_shelve(bzrlib.commands.Command):
 
173
        if strip is None:
 
174
            strip = 0
 
175
        return patch(wt, filename, strip, silent)
 
176
 
 
177
 
 
178
class cmd_shelve(BzrToolsCommand):
203
179
    """Temporarily set aside some changes from the current tree.
204
180
 
205
181
    Shelve allows you to temporarily put changes you've made "on the shelf",
230
206
    """
231
207
 
232
208
    takes_args = ['file*']
233
 
    takes_options = ['message', 'revision',
234
 
            Option('all', help='Shelve all changes without prompting'), 
235
 
            Option('no-color', help='Never display changes in color')]
 
209
    takes_options = [Option('message',
 
210
            help='A message to associate with the shelved changes.',
 
211
            short_name='m', type=unicode),
 
212
            'revision',
 
213
            Option('all', help='Shelve all changes without prompting.'),
 
214
            Option('no-color', help='Never display changes in color.')]
236
215
 
237
216
    def run(self, all=False, file_list=None, message=None, revision=None,
238
217
            no_color=False):
244
223
                                  "parameter.")
245
224
 
246
225
        source = BzrPatchSource(revision, file_list)
247
 
        s = Shelf(source.base)
 
226
        s = shelf.Shelf(source.base)
248
227
        s.shelve(source, all, message, no_color)
249
228
        return 0
250
229
 
272
251
    aliases = ['switch']
273
252
    takes_args = ['othershelf']
274
253
    def run(self, othershelf):
275
 
        s = Shelf(self.shelf.base, othershelf)
 
254
        s = shelf.Shelf(self.shelf.base, othershelf)
276
255
        s.make_default()
277
256
 
278
257
 
291
270
        self.shelf.upgrade()
292
271
 
293
272
 
294
 
class cmd_shelf(bzrlib.commands.Command):
 
273
class cmd_shelf(BzrToolsCommand):
295
274
    """Perform various operations on your shelved patches. See also shelve."""
296
275
    takes_args = ['subcommand', 'args*']
297
276
 
305
284
            args_list = []
306
285
        cmd = self._get_cmd_object(subcommand)
307
286
        source = BzrPatchSource()
308
 
        s = Shelf(source.base)
 
287
        s = shelf.Shelf(source.base)
309
288
        cmd.shelf = s
 
289
 
 
290
        if args_list is None:
 
291
            args_list = []
310
292
        return cmd.run_argv_aliases(args_list)
311
293
 
312
294
    def _get_cmd_object(self, cmd_name):
329
311
        cmd_obj = cmd_class()
330
312
        indent = 2 * ' '
331
313
 
332
 
        usage = command_usage(cmd_obj)
 
314
        usage = cmd_obj._usage()
333
315
        usage = usage.replace('bzr shelf-', '')
334
316
        text.append('%s%s\n' % (indent, usage))
335
317
 
354
336
        return text
355
337
 
356
338
 
357
 
class cmd_unshelve(bzrlib.commands.Command):
 
339
class cmd_unshelve(BzrToolsCommand):
358
340
    """Restore shelved changes.
359
341
 
360
342
    By default the most recently shelved changes are restored. However if you
363
345
    See 'shelve' for more information.
364
346
    """
365
347
    takes_options = [
366
 
            Option('all', help='Unshelve all changes without prompting'),
367
 
            Option('force', help='Force unshelving even if errors occur'),
368
 
            Option('no-color', help='Never display changes in color')
 
348
            Option('all', help='Unshelve all changes without prompting.'),
 
349
            Option('force', help='Force unshelving even if errors occur.'),
 
350
            Option('no-color', help='Never display changes in color.')
369
351
        ]
370
352
    takes_args = ['patch?']
371
353
    def run(self, patch=None, all=False, force=False, no_color=False):
372
354
        source = BzrPatchSource()
373
 
        s = Shelf(source.base)
 
355
        s = shelf.Shelf(source.base)
374
356
        s.unshelve(source, patch, all, force, no_color)
375
357
        return 0
376
358
 
377
359
 
378
 
class cmd_shell(bzrlib.commands.Command):
 
360
class cmd_shell(BzrToolsCommand):
379
361
    """Begin an interactive shell tailored for bzr.
380
362
    Bzr commands can be used without typing bzr first, and will be run natively
381
363
    when possible.  Tab completion is tailored for bzr.  The shell prompt shows
398
380
        return shell.run_shell()
399
381
 
400
382
 
401
 
class cmd_branch_history(bzrlib.commands.Command):
 
383
class cmd_branch_history(BzrToolsCommand):
402
384
    """\
403
385
    Display the development history of a branch.
404
386
 
408
390
    """
409
391
    takes_args = ["branch?"]
410
392
    def run(self, branch=None):
411
 
        from branchhistory import branch_history 
 
393
        from branchhistory import branch_history
412
394
        return branch_history(branch)
413
395
 
414
396
 
415
 
class cmd_zap(bzrlib.commands.Command):
 
397
class cmd_zap(BzrToolsCommand):
416
398
    """\
417
399
    Remove a lightweight checkout, if it can be done safely.
418
400
 
423
405
    If --branch is specified, the branch will be deleted too, but only if the
424
406
    the branch has no new commits (relative to its parent).
425
407
    """
426
 
    takes_options = [Option("branch", help="Remove associtated branch from"
427
 
                                           " repository")]
 
408
    takes_options = [Option("branch", help="Remove associated branch from"
 
409
                                           " repository."),
 
410
                     Option('force', help='Delete tree even if contents are'
 
411
                     ' modified.')]
428
412
    takes_args = ["checkout"]
429
 
    def run(self, checkout, branch=False):
 
413
    def run(self, checkout, branch=False, force=False):
430
414
        from zap import zap
431
 
        return zap(checkout, remove_branch=branch)
432
 
 
433
 
 
434
 
class cmd_cbranch(bzrlib.commands.Command):
 
415
        return zap(checkout, remove_branch=branch, allow_modified=force)
 
416
 
 
417
 
 
418
class cmd_cbranch(BzrToolsCommand):
435
419
    """
436
420
    Create a new checkout, associated with a new repository branch.
437
 
    
438
 
    When you cbranch, bzr looks up the repository associated with your current
439
 
    directory in locations.conf.  It creates a new branch in that repository
440
 
    with the same name and relative path as the checkout you request.
441
 
 
442
 
    The locations.conf parameter is "cbranch_root".  So if you want 
443
 
    cbranch operations in /home/jrandom/bigproject to produce branches in 
444
 
    /home/jrandom/bigproject/repository, you'd add this:
445
 
 
446
 
    [/home/jrandom/bigproject]
447
 
    cbranch_root = /home/jrandom/bigproject/repository
448
 
 
449
 
    Note that if "/home/jrandom/bigproject/repository" isn't a repository,
450
 
    standalone branches will be produced.  Standalone branches will also
451
 
    be produced if the source branch is in 0.7 format (or earlier).
 
421
 
 
422
    When you cbranch, bzr looks up a target location in locations.conf, and
 
423
    creates the branch there.
 
424
 
 
425
    In your locations.conf, add the following lines:
 
426
    [/working_directory_root]
 
427
    cbranch_target = /branch_root
 
428
    cbranch_target:policy = appendpath
 
429
 
 
430
    This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
 
431
    working directory root, the branch will be created in
 
432
    "/branch_root/foo/baz"
 
433
 
 
434
    NOTE: cbranch also supports "cbranch_root", but that behaviour is
 
435
    deprecated.
452
436
    """
453
 
    takes_options = [Option("lightweight", 
454
 
                            help="Create a lightweight checkout"), 'revision']
 
437
    takes_options = [Option("lightweight",
 
438
                            help="Create a lightweight checkout."), 'revision',
 
439
                     Option('files-from', type=unicode,
 
440
                            help='Accelerate checkout using files from this'
 
441
                                 ' tree.'),
 
442
                     Option('hardlink',
 
443
                            help='Hard-link files from source/files-from tree'
 
444
                            ' where posible.')]
455
445
    takes_args = ["source", "target?"]
456
 
    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, hardlink=False):
457
448
        from cbranch import cbranch
458
 
        return cbranch(source, target, lightweight=lightweight, 
459
 
                       revision=revision)
460
 
 
461
 
 
462
 
class cmd_branches(bzrlib.commands.Command):
 
449
        return cbranch(source, target, lightweight=lightweight,
 
450
                       revision=revision, files_from=files_from,
 
451
                       hardlink=hardlink)
 
452
 
 
453
 
 
454
class cmd_branches(BzrToolsCommand):
463
455
    """Scan a location for branches"""
464
456
    takes_args = ["location?"]
465
457
    def run(self, location=None):
466
458
        from branches import branches
467
459
        return branches(location)
468
460
 
 
461
class cmd_trees(BzrToolsCommand):
 
462
    """Scan a location for trees"""
 
463
    takes_args = ['location?']
 
464
    def run(self, location='.'):
 
465
        from bzrlib.workingtree import WorkingTree
 
466
        from bzrlib.transport import get_transport
 
467
        t = get_transport(location)
 
468
        for tree in WorkingTree.find_trees(location):
 
469
            self.outf.write('%s\n' % t.relpath(
 
470
                tree.bzrdir.root_transport.base))
469
471
 
470
 
class cmd_multi_pull(bzrlib.commands.Command):
 
472
class cmd_multi_pull(BzrToolsCommand):
471
473
    """Pull all the branches under a location, e.g. a repository.
472
 
    
 
474
 
473
475
    Both branches present in the directory and the branches of checkouts are
474
476
    pulled.
475
477
    """
476
478
    takes_args = ["location?"]
477
479
    def run(self, location=None):
478
 
        from bzrlib.branch import Branch
479
480
        from bzrlib.transport import get_transport
480
481
        from bzrtools import iter_branch_tree
481
482
        if location is None:
482
483
            location = '.'
483
484
        t = get_transport(location)
 
485
        possible_transports = []
484
486
        if not t.listable():
485
487
            print "Can't list this type of location."
486
488
            return 3
502
504
                relpath = base
503
505
            print "Pulling %s from %s" % (relpath, parent)
504
506
            try:
505
 
                pullable.pull(Branch.open(parent))
 
507
                branch_t = get_transport(parent, possible_transports)
 
508
                pullable.pull(Branch.open_from_transport(branch_t))
506
509
            except Exception, e:
507
510
                print e
508
511
 
509
512
 
510
 
class cmd_branch_mark(bzrlib.commands.Command):
511
 
    """
512
 
    Add, view or list branch markers <EXPERIMENTAL>
513
 
 
514
 
    To add a mark, do 'bzr branch-mark MARK'.
515
 
    To list marks, do 'bzr branch-mark' (this lists all marks for the branch's
516
 
    repository).
517
 
    To delete a mark, do 'bzr branch-mark --delete MARK'
518
 
 
519
 
    These marks can be used to track a branch's status.
520
 
    """
521
 
    takes_args = ['mark?', 'branch?']
522
 
    takes_options = [Option('delete', help='Delete this mark')]
523
 
    def run(self, mark=None, branch=None, delete=False):
524
 
        from branch_mark import branch_mark
525
 
        branch_mark(mark, branch, delete)
526
 
 
527
 
 
528
 
class cmd_import(bzrlib.commands.Command):
529
 
    """Import sources from a tarball
530
 
    
531
 
    This command will import a tarball into a bzr tree, replacing any versioned
532
 
    files already present.  If a directory is specified, it is used as the
533
 
    target.  If the directory does not exist, or is not versioned, it is
534
 
    created.
 
513
 
 
514
class cmd_import(BzrToolsCommand):
 
515
    """Import sources from a directory, tarball or zip file
 
516
 
 
517
    This command will import a directory, tarball or zip file into a bzr
 
518
    tree, replacing any versioned files already present.  If a directory is
 
519
    specified, it is used as the target.  If the directory does not exist, or
 
520
    is not versioned, it is created.
535
521
 
536
522
    Tarballs may be gzip or bzip2 compressed.  This is autodetected.
537
523
 
538
 
    If the tarball has a single root directory, that directory is stripped
539
 
    when extracting the tarball.
 
524
    If the tarball or zip has a single root directory, that directory is
 
525
    stripped when extracting the tarball.  This is not done for directories.
540
526
    """
541
 
    
 
527
 
542
528
    takes_args = ['source', 'tree?']
543
529
    def run(self, source, tree=None):
544
530
        from upstream_import import do_import
545
531
        do_import(source, tree)
546
532
 
547
533
 
548
 
class cmd_cdiff(bzrlib.commands.Command):
 
534
class cmd_cdiff(BzrToolsCommand):
549
535
    """A color version of bzr's diff"""
550
536
    takes_args = property(lambda x: get_cmd_object('diff').takes_args)
551
 
    takes_options = property(lambda x: get_cmd_object('diff').takes_options)
552
 
    def run(*args, **kwargs):
 
537
    takes_options = list(get_cmd_object('diff').takes_options) + [
 
538
        Option('check-style',
 
539
            help='Warn if trailing whitespace or spurious changes have been'
 
540
                 ' added.')]
 
541
 
 
542
    def run(self, check_style=False, *args, **kwargs):
553
543
        from colordiff import colordiff
554
 
        colordiff(*args, **kwargs)
555
 
 
556
 
 
557
 
class cmd_baz_import(bzrlib.commands.Command):
558
 
    """Import an Arch or Baz archive into a bzr repository.
559
 
 
560
 
    This command should be used on local archives (or mirrors) only.  It is
561
 
    quite slow on remote archives.
562
 
    
563
 
    reuse_history allows you to specify any previous imports you 
564
 
    have done of different archives, which this archive has branches
565
 
    tagged from. This will dramatically reduce the time to convert 
566
 
    the archive as it will not have to convert the history already
567
 
    converted in that other branch.
568
 
 
569
 
    If you specify prefixes, only branches whose names start with that prefix
570
 
    will be imported.  Skipped branches will be listed, so you can import any
571
 
    branches you missed by accident.  Here's an example of doing a partial
572
 
    import from thelove@canonical.com:
573
 
    bzr baz-import thelove thelove@canonical.com --prefixes dists:talloc-except
574
 
 
575
 
    WARNING: Encoding should not be specified unless necessary, because if you
576
 
    specify an encoding, your converted branch will not interoperate with
577
 
    independently-converted branches, unless the other branches were converted
578
 
    with exactly the same encoding.  Any encoding recognized by Python may
579
 
    be specified.  Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
580
 
    are incompatible.
581
 
    """
582
 
    takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
583
 
    takes_options = ['verbose', Option('prefixes', type=str,
584
 
                     help="Prefixes of branches to import, colon-separated"),
585
 
                     Option('encoding', type=str, 
586
 
                     help='Force encoding to specified value.  See WARNING.')]
587
 
 
588
 
    def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
589
 
            reuse_history_list=[], prefixes=None):
590
 
        from errors import NoPyBaz
591
 
        try:
592
 
            import baz_import
593
 
            baz_import.baz_import(to_root_dir, from_archive, encoding,
594
 
                                  verbose, reuse_history_list, prefixes)
595
 
        except NoPyBaz:
596
 
            print "This command is disabled.  Please install PyBaz."
597
 
 
598
 
 
599
 
class cmd_baz_import_branch(bzrlib.commands.Command):
600
 
    """Import an Arch or Baz branch into a bzr branch.
601
 
 
602
 
    WARNING: Encoding should not be specified unless necessary, because if you
603
 
    specify an encoding, your converted branch will not interoperate with
604
 
    independently-converted branches, unless the other branches were converted
605
 
    with exactly the same encoding.  Any encoding recognized by Python may
606
 
    be specified.  Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
607
 
    are incompatible.
608
 
    """
609
 
    takes_args = ['to_location', 'from_branch?', 'reuse_history*']
610
 
    takes_options = ['verbose', Option('max-count', type=int),
611
 
                     Option('encoding', type=str, 
612
 
                     help='Force encoding to specified value.  See WARNING.')]
613
 
 
614
 
    def run(self, to_location, from_branch=None, fast=False, max_count=None,
615
 
            encoding=None, verbose=False, dry_run=False,
616
 
            reuse_history_list=[]):
617
 
        from errors import NoPyBaz
618
 
        try:
619
 
            import baz_import
620
 
            baz_import.baz_import_branch(to_location, from_branch, fast, 
621
 
                                         max_count, verbose, encoding, dry_run,
622
 
                                         reuse_history_list)
623
 
        except NoPyBaz:
624
 
            print "This command is disabled.  Please install PyBaz."
625
 
 
626
 
 
627
 
class cmd_rspush(bzrlib.commands.Command):
 
544
        colordiff(check_style, *args, **kwargs)
 
545
 
 
546
 
 
547
class cmd_rspush(BzrToolsCommand):
628
548
    """Upload this branch to another location using rsync.
629
549
 
630
 
    If no location is specified, the last-used location will be used.  To 
631
 
    prevent dirty trees from being uploaded, rspush will error out if there are 
632
 
    unknown files or local changes.  It will also error out if the upstream 
633
 
    directory is non-empty and not an earlier version of the branch. 
 
550
    If no location is specified, the last-used location will be used.  To
 
551
    prevent dirty trees from being uploaded, rspush will error out if there are
 
552
    unknown files or local changes.  It will also error out if the upstream
 
553
    directory is non-empty and not an earlier version of the branch.
634
554
    """
635
555
    takes_args = ['location?']
636
556
    takes_options = [Option('overwrite', help='Ignore differences between'
637
 
                            ' branches and overwrite unconditionally'),
 
557
                            ' branches and overwrite unconditionally.'),
638
558
                     Option('no-tree', help='Do not push the working tree,'
639
559
                            ' just the .bzr.')]
640
560
 
642
562
        from bzrlib import workingtree
643
563
        import bzrtools
644
564
        cur_branch = workingtree.WorkingTree.open_containing(".")[0]
645
 
        bzrtools.rspush(cur_branch, location, overwrite=overwrite, 
 
565
        bzrtools.rspush(cur_branch, location, overwrite=overwrite,
646
566
                      working_tree=not no_tree)
647
567
 
648
568
 
649
 
class cmd_switch(bzrlib.commands.Command):
650
 
    """Set the branch of a lightweight checkout and update."""
651
 
 
652
 
    takes_args = ['to_location']
653
 
 
654
 
    def run(self, to_location):
655
 
        from switch import cmd_switch
656
 
        cmd_switch().run(to_location)
657
 
 
658
 
 
 
569
class cmd_link_tree(BzrToolsCommand):
 
570
    """Hardlink matching files to another tree.
 
571
 
 
572
    Only files with identical content and execute bit will be linked.
 
573
    """
 
574
    takes_args = ['location']
 
575
 
 
576
    def run(self, location):
 
577
        from bzrlib import workingtree
 
578
        from bzrlib.plugins.bzrtools.link_tree import link_tree
 
579
        target_tree = workingtree.WorkingTree.open_containing(".")[0]
 
580
        source_tree = workingtree.WorkingTree.open(location)
 
581
        target_tree.lock_write()
 
582
        try:
 
583
            source_tree.lock_read()
 
584
            try:
 
585
                link_tree(target_tree, source_tree)
 
586
            finally:
 
587
                source_tree.unlock()
 
588
        finally:
 
589
            target_tree.unlock()
 
590
 
 
591
from heads import cmd_heads
659
592
commands = [
660
 
            cmd_baz_import,
661
 
            cmd_baz_import_branch,
662
593
            cmd_branches,
663
594
            cmd_branch_history,
664
 
            cmd_branch_mark,
665
 
            cmd_cbranch,  
 
595
            cmd_cbranch,
666
596
            cmd_cdiff,
667
597
            cmd_clean_tree,
668
598
            cmd_fetch_ghosts,
669
599
            cmd_graph_ancestry,
 
600
            cmd_heads,
670
601
            cmd_import,
 
602
            cmd_link_tree,
671
603
            cmd_multi_pull,
672
604
            cmd_patch,
673
605
            cmd_rspush,
674
 
            cmd_shelf, 
 
606
            cmd_shelf,
675
607
            cmd_shell,
676
 
            cmd_shelve, 
677
 
            cmd_switch,
678
 
            cmd_unshelve, 
679
 
            cmd_zap,            
 
608
            cmd_shelve,
 
609
            cmd_trees,
 
610
            cmd_unshelve,
 
611
            cmd_zap,
680
612
            ]
681
613
 
682
614
 
692
624
    from unittest import TestSuite
693
625
    import bzrtools
694
626
    import tests.clean_tree
695
 
    import upstream_import
 
627
    import tests.test_dotgraph
 
628
    import tests.is_clean
 
629
    import tests.test_cbranch
 
630
    import tests.test_link_tree
 
631
    import tests.test_patch
 
632
    import tests.test_rspush
 
633
    import tests.upstream_import
696
634
    import zap
697
635
    import tests.blackbox
698
636
    import tests.shelf_tests
699
637
    result = TestSuite()
700
638
    result.addTest(DocTestSuite(bzrtools, optionflags=ELLIPSIS))
701
639
    result.addTest(tests.clean_tree.test_suite())
702
 
    try:
703
 
        import baz_import
704
 
        result.addTest(DocTestSuite(baz_import))
705
 
    except NoPyBaz:
706
 
        pass
707
640
    result.addTest(tests.test_suite())
708
641
    result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))
709
642
    result.addTest(tests.blackbox.test_suite())
710
 
    result.addTest(upstream_import.test_suite())
 
643
    result.addTest(tests.upstream_import.test_suite())
711
644
    result.addTest(zap.test_suite())
 
645
    result.addTest(TestLoader().loadTestsFromModule(tests.test_dotgraph))
 
646
    result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))
 
647
    result.addTest(TestLoader().loadTestsFromModule(tests.test_link_tree))
 
648
    result.addTest(TestLoader().loadTestsFromModule(tests.test_patch))
 
649
    result.addTest(TestLoader().loadTestsFromModule(tests.test_rspush))
 
650
    result.addTest(TestLoader().loadTestsFromModule(tests.test_cbranch))
712
651
    return result