66
71
If available, rsvg is used to antialias PNG and JPEG output, but this can
67
72
be disabled with --no-antialias.
69
takes_args = ['branch', 'file']
70
takes_options = ['no-collapse', 'no-antialias', 'merge-branch', 'cluster']
71
def run(self, branch, file, no_collapse=False, no_antialias=False,
72
merge_branch=None, cluster=False):
74
takes_args = ['file', 'merge_branch?']
75
takes_options = [Option('no-collapse', help="Do not skip simple nodes."),
76
Option('no-antialias',
77
help="Do not use rsvg to produce antialiased output."),
78
Option('merge-branch', type=str,
79
help="Use this branch to calcuate a merge base."),
80
Option('cluster', help="Use clustered output."),
81
Option('max-distance',
82
help="Show no nodes farther than this.", type=int),
84
help='Source branch to use (default is current'
89
def run(self, file, merge_branch=None, no_collapse=False,
90
no_antialias=False, cluster=False, max_distance=100,
92
if max_distance == -1:
75
96
ranking = "cluster"
78
graph.write_ancestry_file(branch, file, not no_collapse,
79
not no_antialias, merge_branch, ranking)
81
class cmd_fetch_ghosts(bzrlib.commands.Command):
99
graph.write_ancestry_file(directory, file, not no_collapse,
100
not no_antialias, merge_branch, ranking,
101
max_distance=max_distance)
104
class cmd_fetch_ghosts(BzrToolsCommand):
82
105
"""Attempt to retrieve ghosts from another branch.
83
106
If the other branch is not supplied, the last-pulled branch is used.
85
108
aliases = ['fetch-missing']
86
109
takes_args = ['branch?']
87
def run(self, branch=None):
110
takes_options = [Option('no-fix', help="Skip additional synchonization.")]
111
def run(self, branch=None, no_fix=False):
88
112
from fetch_ghosts import fetch_ghosts
113
fetch_ghosts(branch, do_reconcile=not no_fix)
91
115
strip_help="""Strip the smallest prefix containing num leading slashes from \
92
116
each file name found in the patch file."""
93
Option.OPTIONS['strip'] = Option('strip', type=int, help=strip_help)
94
class cmd_patch(bzrlib.commands.Command):
119
class cmd_patch(BzrToolsCommand):
95
120
"""Apply a named patch to the current tree.
97
122
takes_args = ['filename?']
98
takes_options = ['strip']
99
def run(self, filename=None, strip=0):
123
takes_options = [Option('strip', type=int, short_name='p',
125
Option('silent', help='Suppress chatter.')]
126
def run(self, filename=None, strip=None, silent=False):
100
127
from patch import patch
101
from bzrlib.branch import Branch
102
b = Branch.open_containing('.')[0]
103
return patch(b, filename, strip)
106
class cmd_shelve(bzrlib.commands.Command):
107
"""Temporarily remove some changes from the current tree.
108
Use 'unshelve' to restore these changes.
110
If filenames are specified, only changes to those files will be unshelved.
111
If a revision is specified, all changes since that revision will may be
128
from bzrlib.workingtree import WorkingTree
129
wt = WorkingTree.open_containing('.')[0]
132
return patch(wt, filename, strip, silent)
135
class cmd_shelve1(BzrToolsCommand):
136
"""Temporarily set aside some changes from the current tree.
138
Shelve allows you to temporarily put changes you've made "on the shelf",
139
ie. out of the way, until a later time when you can bring them back from
140
the shelf with the 'unshelve1' command.
142
Shelve is intended to help separate several sets of text changes that have
143
been inappropriately mingled. If you just want to get rid of all changes
144
(text and otherwise) and you don't need to restore them later, use revert.
145
If you want to shelve all text changes at once, use shelve1 --all.
147
By default shelve1 asks you what you want to shelve, press '?' at the
148
prompt to get help. To shelve everything run shelve1 --all.
150
If filenames are specified, only the changes to those files will be
151
shelved, other files will be left untouched.
153
If a revision is specified, changes since that revision will be shelved.
155
You can put multiple items on the shelf. Normally each time you run
156
unshelve1 the most recently shelved changes will be reinstated. However,
157
you can also unshelve changes in a different order by explicitly
158
specifiying which changes to unshelve1. This works best when the changes
159
don't depend on each other.
161
While you have patches on the shelf you can view and manipulate them with
162
the 'shelf1' command. Run 'bzr shelf1 -h' for more info.
114
165
takes_args = ['file*']
115
takes_options = ['all', 'message', 'revision']
116
def run(self, all=False, file_list=None, message=None, revision=None):
117
if file_list is not None and len(file_list) > 0:
118
branchdir = file_list[0]
166
takes_options = [Option('message',
167
help='A message to associate with the shelved changes.',
168
short_name='m', type=unicode),
170
Option('all', help='Shelve all changes without prompting.'),
171
Option('no-color', help='Never display changes in color.')]
173
def run(self, all=False, file_list=None, message=None, revision=None,
122
175
if revision is not None and revision:
123
176
if len(revision) == 1:
124
177
revision = revision[0]
126
raise BzrCommandError("shelve only accepts a single revision "
179
raise CommandError("shelve only accepts a single revision "
130
return s.shelve(all, message, revision, file_list)
133
class cmd_unshelve(bzrlib.commands.Command):
134
"""Restore previously-shelved changes to the current tree.
141
class cmd_shell(bzrlib.commands.Command):
182
source = BzrPatchSource(revision, file_list)
183
s = shelf.Shelf(source.base)
184
s.shelve(source, all, message, no_color)
188
# The following classes are only used as subcommands for 'shelf1', they're
189
# not to be registered directly with bzr.
191
class cmd_shelf_list(bzrlib.commands.Command):
192
"""List the patches on the current shelf."""
193
aliases = ['list', 'ls']
198
class cmd_shelf_delete(bzrlib.commands.Command):
199
"""Delete the patch from the current shelf."""
200
aliases = ['delete', 'del']
201
takes_args = ['patch']
202
def run(self, patch):
203
self.shelf.delete(patch)
206
class cmd_shelf_switch(bzrlib.commands.Command):
207
"""Switch to the other shelf, create it if necessary."""
209
takes_args = ['othershelf']
210
def run(self, othershelf):
211
s = shelf.Shelf(self.shelf.base, othershelf)
215
class cmd_shelf_show(bzrlib.commands.Command):
216
"""Show the contents of the specified or topmost patch."""
217
aliases = ['show', 'cat', 'display']
218
takes_args = ['patch?']
219
def run(self, patch=None):
220
self.shelf.display(patch)
223
class cmd_shelf_upgrade(bzrlib.commands.Command):
224
"""Upgrade old format shelves."""
225
aliases = ['upgrade']
230
class cmd_shelf1(BzrToolsCommand):
231
"""Perform various operations on your shelved patches. See also shelve1."""
232
takes_args = ['subcommand', 'args*']
234
subcommands = [cmd_shelf_list, cmd_shelf_delete, cmd_shelf_switch,
235
cmd_shelf_show, cmd_shelf_upgrade]
237
def run(self, subcommand, args_list):
240
if args_list is None:
242
cmd = self._get_cmd_object(subcommand)
243
source = BzrPatchSource()
244
s = shelf.Shelf(source.base)
247
if args_list is None:
249
return cmd.run_argv_aliases(args_list)
251
def _get_cmd_object(self, cmd_name):
252
for cmd_class in self.subcommands:
253
for alias in cmd_class.aliases:
254
if alias == cmd_name:
256
raise CommandError("Unknown shelf subcommand '%s'" % cmd_name)
259
text = ["%s\n\nSubcommands:\n" % self.__doc__]
261
for cmd_class in self.subcommands:
262
text.extend(self.sub_help(cmd_class) + ['\n'])
266
def sub_help(self, cmd_class):
268
cmd_obj = cmd_class()
271
usage = cmd_obj._usage()
272
usage = usage.replace('bzr shelf-', '')
273
text.append('%s%s\n' % (indent, usage))
275
text.append('%s%s\n' % (2 * indent, cmd_class.__doc__))
277
# Somewhat copied from bzrlib.help.help_on_command_options
279
for option_name, option in sorted(cmd_obj.options().items()):
280
if option_name == 'help':
282
option_help.append('%s--%s' % (3 * indent, option_name))
283
if option.type is not None:
284
option_help.append(' %s' % option.argname.upper())
285
if option.short_name():
286
option_help.append(', -%s' % option.short_name())
287
option_help.append('%s%s\n' % (2 * indent, option.help))
289
if len(option_help) > 0:
290
text.append('%soptions:\n' % (2 * indent))
291
text.extend(option_help)
296
class cmd_unshelve1(BzrToolsCommand):
297
"""Restore shelved changes.
299
By default the most recently shelved changes are restored. However if you
300
specify a patch by name those changes will be restored instead.
302
See 'shelve1' for more information.
305
Option('all', help='Unshelve all changes without prompting.'),
306
Option('force', help='Force unshelving even if errors occur.'),
307
Option('no-color', help='Never display changes in color.')
309
takes_args = ['patch?']
310
def run(self, patch=None, all=False, force=False, no_color=False):
311
source = BzrPatchSource()
312
s = shelf.Shelf(source.base)
313
s.unshelve(source, patch, all, force, no_color)
317
class cmd_shell(BzrToolsCommand):
318
"""Begin an interactive shell tailored for bzr.
319
Bzr commands can be used without typing bzr first, and will be run natively
320
when possible. Tab completion is tailored for bzr. The shell prompt shows
321
the branch nick, revno, and path.
323
If it encounters any moderately complicated shell command, it will punt to
328
bzr bzrtools:287/> status
331
bzr bzrtools:287/> status --[TAB][TAB]
332
--all --help --revision --show-ids
333
bzr bzrtools:287/> status --
337
help='Branch in which to start the shell, '
338
'rather than the one containing the working directory.',
343
def run(self, directory=None):
146
commands = [cmd_shelve, cmd_unshelve, cmd_clean_tree, cmd_graph_ancestry,
147
cmd_fetch_ghosts, cmd_patch, cmd_shell]
149
import bzrlib.builtins
150
if not hasattr(bzrlib.builtins, "cmd_annotate"):
151
commands.append(annotate.cmd_annotate)
152
if not hasattr(bzrlib.builtins, "cmd_push"):
153
commands.append(push.cmd_push)
155
from errors import NoPyBaz
158
commands.append(baz_import.cmd_baz_import)
161
class cmd_baz_import(bzrlib.commands.Command):
162
"""Disabled. (Requires PyBaz)"""
163
takes_args = ['to_root_dir?', 'from_archive?']
164
takes_options = ['verbose']
165
def run(self, to_root_dir=None, from_archive=None, verbose=False):
166
print "This command is disabled. Please install PyBaz."
167
commands.append(cmd_baz_import)
169
if hasattr(bzrlib.commands, 'register_command'):
170
for command in commands:
171
bzrlib.commands.register_command(command)
174
from doctest import DocTestSuite
175
from unittest import TestSuite, TestLoader
180
result.addTest(DocTestSuite(bzrtools))
181
result.addTest(clean_tree.test_suite())
182
result.addTest(TestLoader().loadTestsFromModule(shelf_tests))
183
result.addTest(blackbox.test_suite())
345
return shell.run_shell(directory)
348
class cmd_branch_history(BzrToolsCommand):
350
Display the development history of a branch.
352
Each different committer or branch nick is considered a different line of
353
development. Committers are treated as the same if they have the same
354
name, or if they have the same email address.
356
takes_args = ["branch?"]
357
def run(self, branch=None):
358
from branchhistory import branch_history
359
return branch_history(branch)
362
class cmd_zap(BzrToolsCommand):
364
Remove a lightweight checkout, if it can be done safely.
366
This command will remove a lightweight checkout without losing data. That
367
means it only removes lightweight checkouts, and only if they have no
370
If --branch is specified, the branch will be deleted too, but only if the
371
the branch has no new commits (relative to its parent).
373
If bzr-pipeline is also installed, the --store option will store changes
374
in the branch before deleting the tree. To restore the changes, do::
376
bzr checkout --lightweight $BRANCH $CHECKOUT
377
bzr switch-pipe -d $CHECKOUT `bzr nick -d $CHECKOUT`
379
takes_options = [Option("branch", help="Remove associated branch from"
381
RegistryOption('change_policy',
382
'How to handle changed files',
384
('bzrlib.plugins.bzrtools.zap',
385
'change_policy_registry'),
388
takes_args = ["checkout"]
389
def run(self, checkout, branch=False, change_policy=None):
391
change_policy_registry,
395
if change_policy is None:
396
change_policy = change_policy_registry.get()
397
if change_policy is StoreChanges:
399
import bzrlib.plugins.pipeline
401
raise BzrCommandError('--store requires bzr-pipeline.')
402
return zap(checkout, remove_branch=branch, policy=change_policy)
405
class cmd_cbranch(BzrToolsCommand):
407
Create a new checkout, associated with a new repository branch.
409
When you cbranch, bzr looks up a target location in locations.conf, and
410
creates the branch there.
412
In your locations.conf, add the following lines:
413
[/working_directory_root]
414
cbranch_target = /branch_root
415
cbranch_target:policy = appendpath
417
This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
418
working directory root, the branch will be created in
419
"/branch_root/foo/baz"
421
NOTE: cbranch also supports "cbranch_root", but that behaviour is
424
takes_options = [Option("lightweight",
425
help="Create a lightweight checkout."), 'revision',
426
Option('files-from', type=unicode,
427
help='Accelerate checkout using files from this'
430
help='Hard-link files from source/files-from tree'
432
takes_args = ["source", "target?"]
433
def run(self, source, target=None, lightweight=False, revision=None,
434
files_from=None, hardlink=False):
435
from cbranch import cbranch
436
return cbranch(source, target, lightweight=lightweight,
437
revision=revision, files_from=files_from,
441
class cmd_list_branches(BzrToolsCommand):
442
"""Scan a location for branches"""
445
from bzrlib import commands
446
return commands.plugin_cmds.get_info('list-branches').aliases
448
takes_args = ["location?"]
449
def run(self, location=None):
450
from branches import branches
451
return branches(location)
453
class cmd_trees(BzrToolsCommand):
454
"""Scan a location for trees"""
455
takes_args = ['location?']
456
def run(self, location='.'):
457
from bzrlib.workingtree import WorkingTree
458
from bzrlib.transport import get_transport
459
t = get_transport(location)
460
for tree in WorkingTree.find_trees(location):
461
self.outf.write('%s\n' % t.relpath(
462
tree.bzrdir.root_transport.base))
464
class cmd_multi_pull(BzrToolsCommand):
465
"""Pull all the branches under a location, e.g. a repository.
467
Both branches present in the directory and the branches of checkouts are
470
takes_args = ["location?"]
471
def run(self, location=None):
472
from bzrlib.transport import get_transport
473
from bzrtools import iter_branch_tree
476
t = get_transport(location)
477
possible_transports = []
479
print "Can't list this type of location."
481
for branch, wt in iter_branch_tree(t):
486
parent = branch.get_parent()
493
if base.startswith(t.base):
494
relpath = base[len(t.base):].rstrip('/')
497
print "Pulling %s from %s" % (relpath, parent)
499
branch_t = get_transport(parent, possible_transports)
500
pullable.pull(Branch.open_from_transport(branch_t))
506
class cmd_import(BzrToolsCommand):
507
"""Import sources from a directory, tarball or zip file
509
This command will import a directory, tarball or zip file into a bzr
510
tree, replacing any versioned files already present. If a directory is
511
specified, it is used as the target. If the directory does not exist, or
512
is not versioned, it is created.
514
Tarballs may be gzip or bzip2 compressed. This is autodetected.
516
If the tarball or zip has a single root directory, that directory is
517
stripped when extracting the tarball. This is not done for directories.
520
takes_args = ['source', 'tree?']
521
def run(self, source, tree=None):
522
from upstream_import import do_import
523
do_import(source, tree)
526
class cmd_cdiff(BzrToolsCommand):
527
"""A color version of bzr's diff"""
528
takes_args = property(lambda x: get_cmd_object('diff').takes_args)
529
takes_options = list(get_cmd_object('diff').takes_options) + [
530
RegistryOption.from_kwargs('color',
531
'Color mode to use.',
532
title='Color Mode', value_switches=False, enum_switch=True,
533
never='Never colorize output.',
534
auto='Only colorize output if terminal supports it and STDOUT is a'
536
always='Always colorize output (default).'),
537
Option('check-style',
538
help='Warn if trailing whitespace or spurious changes have been'
541
def run(self, color='always', check_style=False, *args, **kwargs):
542
from colordiff import colordiff
543
colordiff(color, check_style, *args, **kwargs)
546
class cmd_conflict_diff(BzrToolsCommand):
548
"""Compare a conflicted file against BASE."""
550
encoding_type = 'exact'
551
takes_args = ['file*']
553
RegistryOption.from_kwargs('direction', 'Direction of comparison.',
554
value_switches=True, enum_switch=False,
555
other='Compare OTHER against common base.',
556
this='Compare THIS against common base.')]
558
def run(self, file_list, direction='other'):
559
from bzrlib.plugins.bzrtools.colordiff import DiffWriter
560
from conflict_diff import ConflictDiffer
561
dw = DiffWriter(self.outf, check_style=False, color='auto')
562
ConflictDiffer().run(dw, file_list, direction)
565
class cmd_rspush(BzrToolsCommand):
566
"""Upload this branch to another location using rsync.
568
If no location is specified, the last-used location will be used. To
569
prevent dirty trees from being uploaded, rspush will error out if there are
570
unknown files or local changes. It will also error out if the upstream
571
directory is non-empty and not an earlier version of the branch.
573
takes_args = ['location?']
574
takes_options = [Option('overwrite', help='Ignore differences between'
575
' branches and overwrite unconditionally.'),
576
Option('no-tree', help='Do not push the working tree,'
579
def run(self, location=None, overwrite=False, no_tree=False):
580
from bzrlib import workingtree
582
cur_branch = workingtree.WorkingTree.open_containing(".")[0]
583
bzrtools.rspush(cur_branch, location, overwrite=overwrite,
584
working_tree=not no_tree)
587
class cmd_link_tree(BzrToolsCommand):
588
"""Hardlink matching files to another tree.
590
Only files with identical content and execute bit will be linked.
592
takes_args = ['location']
594
def run(self, location):
595
from bzrlib import workingtree
596
from bzrlib.plugins.bzrtools.link_tree import link_tree
597
target_tree = workingtree.WorkingTree.open_containing(".")[0]
598
source_tree = workingtree.WorkingTree.open(location)
599
target_tree.lock_write()
601
source_tree.lock_read()
603
link_tree(target_tree, source_tree)
610
class cmd_create_mirror(BzrToolsCommand):
611
"""Create a mirror of another branch.
613
This is similar to `bzr branch`, but copies more settings, including the
614
submit branch and nickname.
616
It sets the public branch and parent of the target to the source location.
619
takes_args = ['source', 'target']
621
def run(self, source, target):
622
source_branch = Branch.open(source)
623
from bzrlib.plugins.bzrtools.mirror import create_mirror
624
create_mirror(source_branch, target, [])
added
removed
command_classes.py
