1
# Copyright (C) 2005, 2006, 2007 Aaron Bentley <aaron.bentley@utoronto.ca>
2
# Copyright (C) 2005, 2006 Canonical Limited.
3
# Copyright (C) 2006 Michael Ellerman.
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.
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.
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
3
20
Various useful plugins for working with bzr.
8
from shelf import Shelf
26
__version__ = '0.18.0'
29
version_info = tuple(int(n) for n in __version__.split('.'))
32
def check_bzrlib_version(desired):
33
"""Check that bzrlib is compatible.
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.
40
desired_plus = (desired[0], desired[1]+1)
41
bzrlib_version = bzrlib.version_info[:2]
42
if bzrlib_version == desired or (bzrlib_version == desired_plus and
43
bzrlib.version_info[3] == 'dev'):
46
from bzrlib.trace import warning
48
# get the message out any way we can
49
from warnings import warn as warning
50
if bzrlib_version < desired:
51
warning('Installed Bazaar version %s is too old to be used with'
53
'"Bzrtools" %s.' % (bzrlib.__version__, __version__))
54
# Not using BzrNewError, because it may not exist.
55
raise Exception, ('Version mismatch', version_info)
57
warning('Plugin "Bzrtools" is not up to date with installed Bazaar'
59
' There should be a newer version of Bzrtools available, e.g.'
61
% (bzrlib.__version__, bzrlib_version[0], bzrlib_version[1]))
62
if bzrlib_version != desired_plus:
63
raise Exception, 'Version mismatch'
66
check_bzrlib_version(version_info[:2])
68
from bzrlib.lazy_import import lazy_import
69
lazy_import(globals(), """
70
from bzrlib import help
74
from errors import CommandError, NoPyBaz
75
from patchsource import BzrPatchSource
79
import bzrlib.builtins
80
import bzrlib.commands
81
from bzrlib.commands import get_cmd_object
82
from bzrlib.errors import BzrCommandError
11
84
from bzrlib.option import Option
13
from bzrlib.errors import BzrCommandError
14
sys.path.append(os.path.dirname(__file__))
16
Option.OPTIONS['ignored'] = Option('ignored',
17
help='delete all ignored files.')
18
Option.OPTIONS['detrius'] = Option('detrius',
19
help='delete conflict files merge backups, and failed selftest dirs.' +
20
'(*.THIS, *.BASE, *.OTHER, *~, *.tmp')
21
Option.OPTIONS['dry-run'] = Option('dry-run',
22
help='show files to delete instead of deleting them.')
85
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
90
bzrlib.ignores.add_runtime_ignores(['./.shelf'])
24
93
class cmd_clean_tree(bzrlib.commands.Command):
25
94
"""Remove unwanted files from working tree.
26
Normally, ignored files are left alone.
96
By default, only unknown files, not ignored files, are deleted. Versioned
97
files are never deleted.
99
Another class is 'detritus', which includes files emitted by bzr during
100
normal operations and selftests. (The value of these files decreases with
103
If no options are specified, unknown files are deleted. Otherwise, option
104
flags are respected, and may be combined.
106
To check what clean-tree will do, use --dry-run.
28
takes_options = ['ignored', 'detrius', 'dry-run']
29
def run(self, ignored=False, detrius=False, dry_run=False):
108
takes_options = [Option('ignored', help='delete all ignored files.'),
109
Option('detritus', help='delete conflict files, merge'
110
' backups, and failed selftest dirs.'),
112
help='delete files unknown to bzr. (default)'),
113
Option('dry-run', help='show files to delete instead of'
115
def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
30
116
from clean_tree import clean_tree
31
clean_tree('.', ignored=ignored, detrius=detrius, dry_run=dry_run)
117
if not (unknown or ignored or detritus):
119
clean_tree('.', unknown=unknown, ignored=ignored, detritus=detritus,
33
Option.OPTIONS['no-collapse'] = Option('no-collapse')
34
Option.OPTIONS['no-antialias'] = Option('no-antialias')
35
Option.OPTIONS['cluster'] = Option('cluster')
36
Option.OPTIONS['merge-branch'] = Option('merge-branch',type=str)
38
123
class cmd_graph_ancestry(bzrlib.commands.Command):
39
124
"""Produce ancestry graphs using dot.
41
126
Output format is detected according to file extension. Some of the more
42
common output formats are png, gif, svg, ps. An extension of '.dot' will
43
cause a dot graph file to be produced.
127
common output formats are html, png, gif, svg, ps. An extension of '.dot'
128
will cause a dot graph file to be produced. HTML output has mouseovers
129
that show the commit message.
45
131
Branches are labeled r?, where ? is the revno. If they have no revno,
46
132
with the last 5 characters of their revision identifier are used instead.
134
The value starting with d is "(maximum) distance from the null revision".
48
136
If --merge-branch is specified, the two branches are compared and a merge
52
140
white normal revision
53
141
yellow THIS history
55
143
orange COMMON history
56
144
blue COMMON non-history ancestor
57
dotted Missing from branch storage
145
green Merge base (COMMON ancestor farthest from the null revision)
146
dotted Ghost revision (missing from branch storage)
59
Ancestry is usually collapsed by removing revisions with a single parent
148
Ancestry is usually collapsed by skipping revisions with a single parent
60
149
and descendant. The number of skipped revisions is shown on the arrow.
61
150
This feature can be disabled with --no-collapse.
85
191
aliases = ['fetch-missing']
86
192
takes_args = ['branch?']
87
def run(self, branch=None):
193
takes_options = [Option('no-fix')]
194
def run(self, branch=None, no_fix=False):
88
195
from fetch_ghosts import fetch_ghosts
196
fetch_ghosts(branch, no_fix)
91
198
strip_help="""Strip the smallest prefix containing num leading slashes from \
92
199
each file name found in the patch file."""
93
Option.OPTIONS['strip'] = Option('strip', type=int, help=strip_help)
94
202
class cmd_patch(bzrlib.commands.Command):
95
203
"""Apply a named patch to the current tree.
97
205
takes_args = ['filename?']
98
takes_options = ['strip']
99
def run(self, filename=None, strip=0):
206
takes_options = [Option('strip', type=int, help=strip_help),
207
Option('silent', help='Suppress chatter')]
208
def run(self, filename=None, strip=None, silent=False):
100
209
from patch import patch
101
from bzrlib.branch import Branch
102
b = Branch.open_containing('.')[0]
103
return patch(b, filename, strip)
210
from bzrlib.workingtree import WorkingTree
211
wt = WorkingTree.open_containing('.')[0]
214
return patch(wt, filename, strip, silent)
106
217
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
218
"""Temporarily set aside some changes from the current tree.
220
Shelve allows you to temporarily put changes you've made "on the shelf",
221
ie. out of the way, until a later time when you can bring them back from
222
the shelf with the 'unshelve' command.
224
Shelve is intended to help separate several sets of text changes that have
225
been inappropriately mingled. If you just want to get rid of all changes
226
(text and otherwise) and you don't need to restore them later, use revert.
227
If you want to shelve all text changes at once, use shelve --all.
229
By default shelve asks you what you want to shelve, press '?' at the
230
prompt to get help. To shelve everything run shelve --all.
232
If filenames are specified, only the changes to those files will be
233
shelved, other files will be left untouched.
235
If a revision is specified, changes since that revision will be shelved.
237
You can put multiple items on the shelf. Normally each time you run
238
unshelve the most recently shelved changes will be reinstated. However,
239
you can also unshelve changes in a different order by explicitly
240
specifiying which changes to unshelve. This works best when the changes
241
don't depend on each other.
243
While you have patches on the shelf you can view and manipulate them with
244
the 'shelf' command. Run 'bzr shelf -h' for more info.
114
247
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]
248
takes_options = ['message', 'revision',
249
Option('all', help='Shelve all changes without prompting'),
250
Option('no-color', help='Never display changes in color')]
252
def run(self, all=False, file_list=None, message=None, revision=None,
122
254
if revision is not None and revision:
123
255
if len(revision) == 1:
124
256
revision = revision[0]
126
raise BzrCommandError("shelve only accepts a single revision "
258
raise CommandError("shelve only accepts a single revision "
130
return s.shelve(all, message, revision, file_list)
261
source = BzrPatchSource(revision, file_list)
262
s = shelf.Shelf(source.base)
263
s.shelve(source, all, message, no_color)
267
# The following classes are only used as subcommands for 'shelf', they're
268
# not to be registered directly with bzr.
270
class cmd_shelf_list(bzrlib.commands.Command):
271
"""List the patches on the current shelf."""
272
aliases = ['list', 'ls']
277
class cmd_shelf_delete(bzrlib.commands.Command):
278
"""Delete the patch from the current shelf."""
279
aliases = ['delete', 'del']
280
takes_args = ['patch']
281
def run(self, patch):
282
self.shelf.delete(patch)
285
class cmd_shelf_switch(bzrlib.commands.Command):
286
"""Switch to the other shelf, create it if necessary."""
288
takes_args = ['othershelf']
289
def run(self, othershelf):
290
s = shelf.Shelf(self.shelf.base, othershelf)
294
class cmd_shelf_show(bzrlib.commands.Command):
295
"""Show the contents of the specified or topmost patch."""
296
aliases = ['show', 'cat', 'display']
297
takes_args = ['patch?']
298
def run(self, patch=None):
299
self.shelf.display(patch)
302
class cmd_shelf_upgrade(bzrlib.commands.Command):
303
"""Upgrade old format shelves."""
304
aliases = ['upgrade']
309
class cmd_shelf(bzrlib.commands.Command):
310
"""Perform various operations on your shelved patches. See also shelve."""
311
takes_args = ['subcommand', 'args*']
313
subcommands = [cmd_shelf_list, cmd_shelf_delete, cmd_shelf_switch,
314
cmd_shelf_show, cmd_shelf_upgrade]
316
def run(self, subcommand, args_list):
319
if args_list is None:
321
cmd = self._get_cmd_object(subcommand)
322
source = BzrPatchSource()
323
s = shelf.Shelf(source.base)
326
if args_list is None:
328
return cmd.run_argv_aliases(args_list)
330
def _get_cmd_object(self, cmd_name):
331
for cmd_class in self.subcommands:
332
for alias in cmd_class.aliases:
333
if alias == cmd_name:
335
raise CommandError("Unknown shelf subcommand '%s'" % cmd_name)
338
text = ["%s\n\nSubcommands:\n" % self.__doc__]
340
for cmd_class in self.subcommands:
341
text.extend(self.sub_help(cmd_class) + ['\n'])
345
def sub_help(self, cmd_class):
347
cmd_obj = cmd_class()
350
usage = cmd_obj._usage()
351
usage = usage.replace('bzr shelf-', '')
352
text.append('%s%s\n' % (indent, usage))
354
text.append('%s%s\n' % (2 * indent, cmd_class.__doc__))
356
# Somewhat copied from bzrlib.help.help_on_command_options
358
for option_name, option in sorted(cmd_obj.options().items()):
359
if option_name == 'help':
361
option_help.append('%s--%s' % (3 * indent, option_name))
362
if option.type is not None:
363
option_help.append(' %s' % option.argname.upper())
364
if option.short_name():
365
option_help.append(', -%s' % option.short_name())
366
option_help.append('%s%s\n' % (2 * indent, option.help))
368
if len(option_help) > 0:
369
text.append('%soptions:\n' % (2 * indent))
370
text.extend(option_help)
133
375
class cmd_unshelve(bzrlib.commands.Command):
134
"""Restore previously-shelved changes to the current tree.
376
"""Restore shelved changes.
378
By default the most recently shelved changes are restored. However if you
379
specify a patch by name those changes will be restored instead.
381
See 'shelve' for more information.
384
Option('all', help='Unshelve all changes without prompting'),
385
Option('force', help='Force unshelving even if errors occur'),
386
Option('no-color', help='Never display changes in color')
388
takes_args = ['patch?']
389
def run(self, patch=None, all=False, force=False, no_color=False):
390
source = BzrPatchSource()
391
s = shelf.Shelf(source.base)
392
s.unshelve(source, patch, all, force, no_color)
141
396
class cmd_shell(bzrlib.commands.Command):
397
"""Begin an interactive shell tailored for bzr.
398
Bzr commands can be used without typing bzr first, and will be run natively
399
when possible. Tab completion is tailored for bzr. The shell prompt shows
400
the branch nick, revno, and path.
402
If it encounters any moderately complicated shell command, it will punt to
407
bzr bzrtools:287/> status
410
bzr bzrtools:287/> status --[TAB][TAB]
411
--all --help --revision --show-ids
412
bzr bzrtools:287/> status --
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)
416
return shell.run_shell()
419
class cmd_branch_history(bzrlib.commands.Command):
421
Display the development history of a branch.
423
Each different committer or branch nick is considered a different line of
424
development. Committers are treated as the same if they have the same
425
name, or if they have the same email address.
427
takes_args = ["branch?"]
428
def run(self, branch=None):
429
from branchhistory import branch_history
430
return branch_history(branch)
433
class cmd_zap(bzrlib.commands.Command):
435
Remove a lightweight checkout, if it can be done safely.
437
This command will remove a lightweight checkout without losing data. That
438
means it only removes lightweight checkouts, and only if they have no
441
If --branch is specified, the branch will be deleted too, but only if the
442
the branch has no new commits (relative to its parent).
444
takes_options = [Option("branch", help="Remove associtated branch from"
446
takes_args = ["checkout"]
447
def run(self, checkout, branch=False):
449
return zap(checkout, remove_branch=branch)
452
class cmd_cbranch(bzrlib.commands.Command):
454
Create a new checkout, associated with a new repository branch.
456
When you cbranch, bzr looks up a target location in locations.conf, and
457
creates the branch there.
459
In your locations.conf, add the following lines:
460
[/working_directory_root]
461
cbranch_target = /branch_root
462
cbranch_target:policy = appendpath
464
This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
465
working directory root, the branch will be created in
466
"/branch_root/foo/baz"
468
NOTE: cbranch also supports "cbranch_root", but that behaviour is
471
takes_options = [Option("lightweight",
472
help="Create a lightweight checkout"), 'revision']
473
takes_args = ["source", "target?"]
474
def run(self, source, target=None, lightweight=False, revision=None):
475
from cbranch import cbranch
476
return cbranch(source, target, lightweight=lightweight,
480
class cmd_branches(bzrlib.commands.Command):
481
"""Scan a location for branches"""
482
takes_args = ["location?"]
483
def run(self, location=None):
484
from branches import branches
485
return branches(location)
488
class cmd_multi_pull(bzrlib.commands.Command):
489
"""Pull all the branches under a location, e.g. a repository.
491
Both branches present in the directory and the branches of checkouts are
494
takes_args = ["location?"]
495
def run(self, location=None):
496
from bzrlib.branch import Branch
497
from bzrlib.transport import get_transport
498
from bzrtools import iter_branch_tree
501
t = get_transport(location)
503
print "Can't list this type of location."
505
for branch, wt in iter_branch_tree(t):
510
parent = branch.get_parent()
517
if base.startswith(t.base):
518
relpath = base[len(t.base):].rstrip('/')
521
print "Pulling %s from %s" % (relpath, parent)
523
pullable.pull(Branch.open(parent))
528
class cmd_branch_mark(bzrlib.commands.Command):
530
Add, view or list branch markers <EXPERIMENTAL>
532
To add a mark, do 'bzr branch-mark MARK'.
533
To list marks, do 'bzr branch-mark' (this lists all marks for the branch's
535
To delete a mark, do 'bzr branch-mark --delete MARK'
537
These marks can be used to track a branch's status.
539
takes_args = ['mark?', 'branch?']
540
takes_options = [Option('delete', help='Delete this mark')]
541
def run(self, mark=None, branch=None, delete=False):
542
from branch_mark import branch_mark
543
branch_mark(mark, branch, delete)
546
class cmd_import(bzrlib.commands.Command):
547
"""Import sources from a directory, tarball or zip file
549
This command will import a directory, tarball or zip file into a bzr
550
tree, replacing any versioned files already present. If a directory is
551
specified, it is used as the target. If the directory does not exist, or
552
is not versioned, it is created.
554
Tarballs may be gzip or bzip2 compressed. This is autodetected.
556
If the tarball or zip has a single root directory, that directory is
557
stripped when extracting the tarball. This is not done for directories.
560
takes_args = ['source', 'tree?']
561
def run(self, source, tree=None):
562
from upstream_import import do_import
563
do_import(source, tree)
566
class cmd_cdiff(bzrlib.commands.Command):
567
"""A color version of bzr's diff"""
568
takes_args = property(lambda x: get_cmd_object('diff').takes_args)
570
def _takes_options(self):
571
options = list(get_cmd_object('diff').takes_options)
572
options.append(Option('check-style',
573
help='Warn if trailing whitespace or spurious changes have been'
577
takes_options = property(_takes_options)
579
def run(self, check_style=False, *args, **kwargs):
580
from colordiff import colordiff
581
colordiff(check_style, *args, **kwargs)
584
class cmd_baz_import(bzrlib.commands.Command):
585
"""Import an Arch or Baz archive into a bzr repository.
587
This command should be used on local archives (or mirrors) only. It is
588
quite slow on remote archives.
590
reuse_history allows you to specify any previous imports you
591
have done of different archives, which this archive has branches
592
tagged from. This will dramatically reduce the time to convert
593
the archive as it will not have to convert the history already
594
converted in that other branch.
596
If you specify prefixes, only branches whose names start with that prefix
597
will be imported. Skipped branches will be listed, so you can import any
598
branches you missed by accident. Here's an example of doing a partial
599
import from thelove@canonical.com:
600
bzr baz-import thelove thelove@canonical.com --prefixes dists:talloc-except
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'
609
takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
610
takes_options = ['verbose', Option('prefixes', type=str,
611
help="Prefixes of branches to import, colon-separated"),
612
Option('encoding', type=str,
613
help='Force encoding to specified value. See WARNING.')]
615
def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
616
reuse_history_list=[], prefixes=None):
617
from errors import NoPyBaz
620
baz_import.baz_import(to_root_dir, from_archive, encoding,
621
verbose, reuse_history_list, prefixes)
623
print "This command is disabled. Please install PyBaz."
626
class cmd_baz_import_branch(bzrlib.commands.Command):
627
"""Import an Arch or Baz branch into a bzr branch.
629
WARNING: Encoding should not be specified unless necessary, because if you
630
specify an encoding, your converted branch will not interoperate with
631
independently-converted branches, unless the other branches were converted
632
with exactly the same encoding. Any encoding recognized by Python may
633
be specified. Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
636
takes_args = ['to_location', 'from_branch?', 'reuse_history*']
637
takes_options = ['verbose', Option('max-count', type=int),
638
Option('encoding', type=str,
639
help='Force encoding to specified value. See WARNING.')]
641
def run(self, to_location, from_branch=None, fast=False, max_count=None,
642
encoding=None, verbose=False, dry_run=False,
643
reuse_history_list=[]):
644
from errors import NoPyBaz
647
baz_import.baz_import_branch(to_location, from_branch, fast,
648
max_count, verbose, encoding, dry_run,
651
print "This command is disabled. Please install PyBaz."
654
class cmd_rspush(bzrlib.commands.Command):
655
"""Upload this branch to another location using rsync.
657
If no location is specified, the last-used location will be used. To
658
prevent dirty trees from being uploaded, rspush will error out if there are
659
unknown files or local changes. It will also error out if the upstream
660
directory is non-empty and not an earlier version of the branch.
662
takes_args = ['location?']
663
takes_options = [Option('overwrite', help='Ignore differences between'
664
' branches and overwrite unconditionally'),
665
Option('no-tree', help='Do not push the working tree,'
668
def run(self, location=None, overwrite=False, no_tree=False):
669
from bzrlib import workingtree
671
cur_branch = workingtree.WorkingTree.open_containing(".")[0]
672
bzrtools.rspush(cur_branch, location, overwrite=overwrite,
673
working_tree=not no_tree)
676
class cmd_switch(bzrlib.commands.Command):
677
"""Set the branch of a lightweight checkout and update."""
679
takes_args = ['to_location']
681
def run(self, to_location):
682
from switch import cmd_switch
683
cmd_switch().run(to_location)
688
cmd_baz_import_branch,
169
710
if hasattr(bzrlib.commands, 'register_command'):
170
711
for command in commands:
171
712
bzrlib.commands.register_command(command)
173
715
def test_suite():
174
from doctest import DocTestSuite
175
from unittest import TestSuite, TestLoader
716
from bzrlib.tests.TestUtil import TestLoader
718
from doctest import DocTestSuite, ELLIPSIS
719
from unittest import TestSuite
721
import tests.clean_tree
722
import tests.is_clean
723
import tests.upstream_import
725
import tests.blackbox
726
import tests.shelf_tests
179
727
result = TestSuite()
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())
728
result.addTest(DocTestSuite(bzrtools, optionflags=ELLIPSIS))
729
result.addTest(tests.clean_tree.test_suite())
732
result.addTest(DocTestSuite(baz_import))
735
result.addTest(tests.test_suite())
736
result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))
737
result.addTest(tests.blackbox.test_suite())
738
result.addTest(tests.upstream_import.test_suite())
739
result.addTest(zap.test_suite())
740
result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))
added
removed
__init__.py
