1
# Copyright (C) 2005, 2006 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
20
3
Various useful plugins for working with bzr.
26
__version__ = '0.14.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:
45
from bzrlib.trace import warning
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)
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'
62
check_bzrlib_version(version_info[:2])
65
from errors import CommandError, NoPyBaz
7
from errors import CommandError
66
8
from patchsource import BzrPatchSource
67
9
from shelf import Shelf
71
import bzrlib.builtins
72
import bzrlib.commands
73
from bzrlib.commands import get_cmd_object
12
from bzrlib.option import Option
74
14
from bzrlib.errors import BzrCommandError
75
from bzrlib.help import command_usage
77
from bzrlib.option import Option
15
from reweave_inventory import cmd_fix
78
16
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
83
bzrlib.ignores.add_runtime_ignores(['./.shelf'])
18
from bzrlib import DEFAULT_IGNORE
21
DEFAULT_IGNORE.append('./.shelf')
22
DEFAULT_IGNORE.append('./.bzr-shelf*')
25
Option.OPTIONS['ignored'] = Option('ignored',
26
help='delete all ignored files.')
27
Option.OPTIONS['detritus'] = Option('detritus',
28
help='delete conflict files merge backups, and failed selftest dirs.' +
29
'(*.THIS, *.BASE, *.OTHER, *~, *.tmp')
30
Option.OPTIONS['dry-run'] = Option('dry-run',
31
help='show files to delete instead of deleting them.')
86
33
class cmd_clean_tree(bzrlib.commands.Command):
87
34
"""Remove unwanted files from working tree.
89
By default, only unknown files, not ignored files, are deleted. Versioned
90
files are never deleted.
92
Another class is 'detritus', which includes files emitted by bzr during
93
normal operations and selftests. (The value of these files decreases with
96
If no options are specified, unknown files are deleted. Otherwise, option
97
flags are respected, and may be combined.
99
To check what clean-tree will do, use --dry-run.
35
Normally, ignored files are left alone.
101
takes_options = [Option('ignored', help='delete all ignored files.'),
102
Option('detritus', help='delete conflict files, merge'
103
' backups, and failed selftest dirs.'),
105
help='delete files unknown to bzr. (default)'),
106
Option('dry-run', help='show files to delete instead of'
108
def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
37
takes_options = ['ignored', 'detritus', 'dry-run']
38
def run(self, ignored=False, detritus=False, dry_run=False):
109
39
from clean_tree import clean_tree
110
if not (unknown or ignored or detritus):
112
clean_tree('.', unknown=unknown, ignored=ignored, detritus=detritus,
40
clean_tree('.', ignored=ignored, detritus=detritus, dry_run=dry_run)
42
Option.OPTIONS['merge-branch'] = Option('merge-branch',type=str)
116
44
class cmd_graph_ancestry(bzrlib.commands.Command):
117
45
"""Produce ancestry graphs using dot.
242
163
source = BzrPatchSource(revision, file_list)
243
164
s = Shelf(source.base)
244
s.shelve(source, all, message, no_color)
165
s.shelve(source, all, message)
248
# The following classes are only used as subcommands for 'shelf', they're
249
# not to be registered directly with bzr.
251
class cmd_shelf_list(bzrlib.commands.Command):
252
"""List the patches on the current shelf."""
253
aliases = ['list', 'ls']
258
class cmd_shelf_delete(bzrlib.commands.Command):
259
"""Delete the patch from the current shelf."""
260
aliases = ['delete', 'del']
261
takes_args = ['patch']
262
def run(self, patch):
263
self.shelf.delete(patch)
266
class cmd_shelf_switch(bzrlib.commands.Command):
267
"""Switch to the other shelf, create it if necessary."""
269
takes_args = ['othershelf']
270
def run(self, othershelf):
271
s = Shelf(self.shelf.base, othershelf)
275
class cmd_shelf_show(bzrlib.commands.Command):
276
"""Show the contents of the specified or topmost patch."""
277
aliases = ['show', 'cat', 'display']
278
takes_args = ['patch?']
279
def run(self, patch=None):
280
self.shelf.display(patch)
283
class cmd_shelf_upgrade(bzrlib.commands.Command):
284
"""Upgrade old format shelves."""
285
aliases = ['upgrade']
290
168
class cmd_shelf(bzrlib.commands.Command):
291
"""Perform various operations on your shelved patches. See also shelve."""
169
"""Perform various operations on your shelved patches. See also shelve.
172
list (ls) List the patches on the current shelf.
173
delete (del) <patch> Delete a patch from the current shelf.
174
switch <shelf> Switch to the named shelf, create it if necessary.
175
show <patch> Show the contents of the specified patch.
176
upgrade Upgrade old format shelves.
292
178
takes_args = ['subcommand', 'args*']
294
subcommands = [cmd_shelf_list, cmd_shelf_delete, cmd_shelf_switch,
295
cmd_shelf_show, cmd_shelf_upgrade]
297
180
def run(self, subcommand, args_list):
300
if args_list is None:
302
cmd = self._get_cmd_object(subcommand)
303
183
source = BzrPatchSource()
304
184
s = Shelf(source.base)
306
return cmd.run_argv_aliases(args_list)
308
def _get_cmd_object(self, cmd_name):
309
for cmd_class in self.subcommands:
310
for alias in cmd_class.aliases:
311
if alias == cmd_name:
313
raise CommandError("Unknown shelf subcommand '%s'" % cmd_name)
316
text = ["%s\n\nSubcommands:\n" % self.__doc__]
318
for cmd_class in self.subcommands:
319
text.extend(self.sub_help(cmd_class) + ['\n'])
323
def sub_help(self, cmd_class):
325
cmd_obj = cmd_class()
328
usage = command_usage(cmd_obj)
329
usage = usage.replace('bzr shelf-', '')
330
text.append('%s%s\n' % (indent, usage))
332
text.append('%s%s\n' % (2 * indent, cmd_class.__doc__))
334
# Somewhat copied from bzrlib.help.help_on_command_options
336
for option_name, option in sorted(cmd_obj.options().items()):
337
if option_name == 'help':
339
option_help.append('%s--%s' % (3 * indent, option_name))
340
if option.type is not None:
341
option_help.append(' %s' % option.argname.upper())
342
if option.short_name():
343
option_help.append(', -%s' % option.short_name())
344
option_help.append('%s%s\n' % (2 * indent, option.help))
346
if len(option_help) > 0:
347
text.append('%soptions:\n' % (2 * indent))
348
text.extend(option_help)
186
if subcommand == 'ls' or subcommand == 'list':
187
self.__check_no_args(args_list, "shelf list takes no arguments!")
189
elif subcommand == 'delete' or subcommand == 'del':
190
self.__check_one_arg(args_list, "shelf delete takes one argument!")
191
s.delete(args_list[0])
192
elif subcommand == 'switch':
193
self.__check_one_arg(args_list, "shelf switch takes one argument!")
194
s = Shelf(source.base, args_list[0])
196
elif subcommand == 'show':
197
self.__check_one_arg(args_list, "shelf show takes one argument!")
198
s.display(args_list[0])
199
elif subcommand == 'upgrade':
200
self.__check_no_args(args_list, "shelf upgrade takes no arguments!")
203
print subcommand, args_list
204
print >>sys.stderr, "Unknown shelf subcommand '%s'" % subcommand
206
def __check_one_arg(self, args, msg):
207
if args is None or len(args) != 1:
208
raise CommandError(msg)
210
def __check_no_args(self, args, msg):
212
raise CommandError(msg)
353
215
class cmd_unshelve(bzrlib.commands.Command):
354
"""Restore shelved changes.
356
By default the most recently shelved changes are restored. However if you
357
specify a patch by name those changes will be restored instead.
216
"""Restore the most recently shelved changes to the current tree.
359
217
See 'shelve' for more information.
361
219
takes_options = [
362
220
Option('all', help='Unshelve all changes without prompting'),
363
221
Option('force', help='Force unshelving even if errors occur'),
364
Option('no-color', help='Never display changes in color')
366
takes_args = ['patch?']
367
def run(self, patch=None, all=False, force=False, no_color=False):
223
def run(self, all=False, force=False):
368
224
source = BzrPatchSource()
369
225
s = Shelf(source.base)
370
s.unshelve(source, patch, all, force, no_color)
226
s.unshelve(source, all, force)
407
262
from branchhistory import branch_history
408
263
return branch_history(branch)
411
class cmd_zap(bzrlib.commands.Command):
413
Remove a lightweight checkout, if it can be done safely.
415
This command will remove a lightweight checkout without losing data. That
416
means it only removes lightweight checkouts, and only if they have no
419
If --branch is specified, the branch will be deleted too, but only if the
420
the branch has no new commits (relative to its parent).
422
takes_options = [Option("branch", help="Remove associtated branch from"
424
takes_args = ["checkout"]
425
def run(self, checkout, branch=False):
427
return zap(checkout, remove_branch=branch)
430
class cmd_cbranch(bzrlib.commands.Command):
432
Create a new checkout, associated with a new repository branch.
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.
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:
442
[/home/jrandom/bigproject]
443
cbranch_root = /home/jrandom/bigproject/repository
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).
449
takes_options = [Option("lightweight",
450
help="Create a lightweight checkout"), 'revision']
451
takes_args = ["source", "target?"]
452
def run(self, source, target=None, lightweight=False, revision=None):
453
from cbranch import cbranch
454
return cbranch(source, target, lightweight=lightweight,
458
class cmd_branches(bzrlib.commands.Command):
459
"""Scan a location for branches"""
460
takes_args = ["location?"]
461
def run(self, location=None):
462
from branches import branches
463
return branches(location)
466
class cmd_multi_pull(bzrlib.commands.Command):
467
"""Pull all the branches under a location, e.g. a repository.
469
Both branches present in the directory and the branches of checkouts are
472
takes_args = ["location?"]
473
def run(self, location=None):
474
from bzrlib.branch import Branch
475
from bzrlib.transport import get_transport
476
from bzrtools import iter_branch_tree
479
t = get_transport(location)
481
print "Can't list this type of location."
483
for branch, wt in iter_branch_tree(t):
488
parent = branch.get_parent()
495
if base.startswith(t.base):
496
relpath = base[len(t.base):].rstrip('/')
499
print "Pulling %s from %s" % (relpath, parent)
501
pullable.pull(Branch.open(parent))
506
class cmd_branch_mark(bzrlib.commands.Command):
508
Add, view or list branch markers <EXPERIMENTAL>
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
513
To delete a mark, do 'bzr branch-mark --delete MARK'
515
These marks can be used to track a branch's status.
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)
524
class cmd_import(bzrlib.commands.Command):
525
"""Import sources from a tarball
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
532
Tarballs may be gzip or bzip2 compressed. This is autodetected.
534
If the tarball has a single root directory, that directory is stripped
535
when extracting the tarball.
538
takes_args = ['source', 'tree?']
539
def run(self, source, tree=None):
540
from upstream_import import do_import
541
do_import(source, tree)
544
class cmd_cdiff(bzrlib.commands.Command):
545
"""A color version of bzr's diff"""
546
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):
549
from colordiff import colordiff
550
colordiff(*args, **kwargs)
553
class cmd_baz_import(bzrlib.commands.Command):
554
"""Import an Arch or Baz archive into a bzr repository.
556
This command should be used on local archives (or mirrors) only. It is
557
quite slow on remote archives.
559
reuse_history allows you to specify any previous imports you
560
have done of different archives, which this archive has branches
561
tagged from. This will dramatically reduce the time to convert
562
the archive as it will not have to convert the history already
563
converted in that other branch.
565
If you specify prefixes, only branches whose names start with that prefix
566
will be imported. Skipped branches will be listed, so you can import any
567
branches you missed by accident. Here's an example of doing a partial
568
import from thelove@canonical.com:
569
bzr baz-import thelove thelove@canonical.com --prefixes dists:talloc-except
571
WARNING: Encoding should not be specified unless necessary, because if you
572
specify an encoding, your converted branch will not interoperate with
573
independently-converted branches, unless the other branches were converted
574
with exactly the same encoding. Any encoding recognized by Python may
575
be specified. Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
578
takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
579
takes_options = ['verbose', Option('prefixes', type=str,
580
help="Prefixes of branches to import, colon-separated"),
581
Option('encoding', type=str,
582
help='Force encoding to specified value. See WARNING.')]
584
def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
585
reuse_history_list=[], prefixes=None):
586
from errors import NoPyBaz
589
baz_import.baz_import(to_root_dir, from_archive, encoding,
590
verbose, reuse_history_list, prefixes)
592
print "This command is disabled. Please install PyBaz."
595
class cmd_baz_import_branch(bzrlib.commands.Command):
596
"""Import an Arch or Baz branch into a bzr branch.
598
WARNING: Encoding should not be specified unless necessary, because if you
599
specify an encoding, your converted branch will not interoperate with
600
independently-converted branches, unless the other branches were converted
601
with exactly the same encoding. Any encoding recognized by Python may
602
be specified. Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
605
takes_args = ['to_location', 'from_branch?', 'reuse_history*']
606
takes_options = ['verbose', Option('max-count', type=int),
607
Option('encoding', type=str,
608
help='Force encoding to specified value. See WARNING.')]
610
def run(self, to_location, from_branch=None, fast=False, max_count=None,
611
encoding=None, verbose=False, dry_run=False,
612
reuse_history_list=[]):
613
from errors import NoPyBaz
616
baz_import.baz_import_branch(to_location, from_branch, fast,
617
max_count, verbose, encoding, dry_run,
620
print "This command is disabled. Please install PyBaz."
623
class cmd_rspush(bzrlib.commands.Command):
624
"""Upload this branch to another location using rsync.
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.
631
takes_args = ['location?']
632
takes_options = [Option('overwrite', help='Ignore differences between'
633
' branches and overwrite unconditionally'),
634
Option('no-tree', help='Do not push the working tree,'
637
def run(self, location=None, overwrite=False, no_tree=False):
638
from bzrlib import workingtree
640
cur_branch = workingtree.WorkingTree.open_containing(".")[0]
641
bzrtools.rspush(cur_branch, location, overwrite=overwrite,
642
working_tree=not no_tree)
645
class cmd_switch(bzrlib.commands.Command):
646
"""Set the branch of a lightweight checkout and update."""
648
takes_args = ['to_location']
650
def run(self, to_location):
651
from switch import cmd_switch
652
cmd_switch().run(to_location)
657
cmd_baz_import_branch,
265
commands = [cmd_shelve, cmd_unshelve, cmd_shelf, cmd_clean_tree,
266
cmd_graph_ancestry, cmd_fetch_ghosts, cmd_patch, cmd_shell,
267
cmd_fix, cmd_branch_history]
269
command_decorators = []
271
command_decorators = []
273
import bzrlib.builtins
274
if not hasattr(bzrlib.builtins, "cmd_push"):
275
commands.append(push.cmd_push)
277
command_decorators.append(push.cmd_push)
279
from errors import NoPyBaz
282
commands.append(baz_import.cmd_baz_import_branch)
283
commands.append(baz_import.cmd_baz_import)
286
class cmd_baz_import_branch(bzrlib.commands.Command):
287
"""Disabled. (Requires PyBaz)"""
288
takes_args = ['to_location?', 'from_branch?', 'reuse_history*']
289
takes_options = ['verbose', Option('max-count', type=int)]
290
def run(self, to_location=None, from_branch=None, fast=False,
291
max_count=None, verbose=False, dry_run=False,
292
reuse_history_list=[]):
293
print "This command is disabled. Please install PyBaz."
296
class cmd_baz_import(bzrlib.commands.Command):
297
"""Disabled. (Requires PyBaz)"""
298
takes_args = ['to_root_dir?', 'from_archive?', 'reuse_history*']
299
takes_options = ['verbose', Option('prefixes', type=str,
300
help="Prefixes of branches to import")]
301
def run(self, to_root_dir=None, from_archive=None, verbose=False,
302
reuse_history_list=[], prefixes=None):
303
print "This command is disabled. Please install PyBaz."
304
commands.extend((cmd_baz_import_branch, cmd_baz_import))
679
307
if hasattr(bzrlib.commands, 'register_command'):
680
308
for command in commands:
681
309
bzrlib.commands.register_command(command)
310
for command in command_decorators:
311
command._original_command = bzrlib.commands.register_command(
684
315
def test_suite():
685
317
from bzrlib.tests.TestUtil import TestLoader
687
319
from doctest import DocTestSuite, ELLIPSIS
688
320
from unittest import TestSuite
690
import tests.clean_tree
691
import upstream_import
693
322
import tests.blackbox
694
323
import tests.shelf_tests
695
324
result = TestSuite()
696
325
result.addTest(DocTestSuite(bzrtools, optionflags=ELLIPSIS))
697
result.addTest(tests.clean_tree.test_suite())
700
result.addTest(DocTestSuite(baz_import))
326
result.addTest(clean_tree.test_suite())
327
result.addTest(DocTestSuite(baz_import))
703
328
result.addTest(tests.test_suite())
704
329
result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))
705
330
result.addTest(tests.blackbox.test_suite())
706
result.addTest(upstream_import.test_suite())
707
result.addTest(zap.test_suite())