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
10
from switch import cmd_switch
71
import bzrlib.builtins
72
import bzrlib.commands
73
from bzrlib.commands import get_cmd_object
13
from bzrlib.option import Option
74
15
from bzrlib.errors import BzrCommandError
75
from bzrlib.help import command_usage
77
from bzrlib.option import Option
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
19
from bzrlib.help import command_usage
22
DEFAULT_IGNORE.append('./.shelf')
23
DEFAULT_IGNORE.append('./.bzr-shelf*')
86
26
class cmd_clean_tree(bzrlib.commands.Command):
87
"""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.
27
"""Remove unwanted files from working tree. <BZRTOOLS>
28
Normally, ignored files are left alone.
101
30
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)'),
31
Option('detritus', help='delete conflict files merge'
32
' backups, and failed selftest dirs. (*.THIS, '
33
'*.BASE, *.OTHER, *~, *.tmp)'),
106
34
Option('dry-run', help='show files to delete instead of'
107
35
' deleting them.')]
108
def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
36
def run(self, ignored=False, detritus=False, dry_run=False):
109
37
from clean_tree import clean_tree
110
if not (unknown or ignored or detritus):
112
clean_tree('.', unknown=unknown, ignored=ignored, detritus=detritus,
38
clean_tree('.', ignored=ignored, detritus=detritus, dry_run=dry_run)
116
40
class cmd_graph_ancestry(bzrlib.commands.Command):
117
"""Produce ancestry graphs using dot.
41
"""Produce ancestry graphs using dot. <BZRTOOLS>
119
43
Output format is detected according to file extension. Some of the more
120
44
common output formats are html, png, gif, svg, ps. An extension of '.dot'
544
450
from upstream_import import do_import
545
451
do_import(source, tree)
548
class cmd_cdiff(bzrlib.commands.Command):
549
"""A color version of bzr's diff"""
550
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):
553
from colordiff import colordiff
554
colordiff(*args, **kwargs)
557
class cmd_baz_import(bzrlib.commands.Command):
558
"""Import an Arch or Baz archive into a bzr repository.
560
This command should be used on local archives (or mirrors) only. It is
561
quite slow on remote archives.
453
class cmd_shove(bzrlib.commands.Command):
454
"""Apply uncommitted changes to another tree <BZRTOOLS>
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.
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
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'
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.')]
588
def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
589
reuse_history_list=[], prefixes=None):
590
from errors import NoPyBaz
593
baz_import.baz_import(to_root_dir, from_archive, encoding,
594
verbose, reuse_history_list, prefixes)
596
print "This command is disabled. Please install PyBaz."
599
class cmd_baz_import_branch(bzrlib.commands.Command):
600
"""Import an Arch or Baz branch into a bzr branch.
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_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.')]
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
620
baz_import.baz_import_branch(to_location, from_branch, fast,
621
max_count, verbose, encoding, dry_run,
624
print "This command is disabled. Please install PyBaz."
627
class cmd_rspush(bzrlib.commands.Command):
628
"""Upload this branch to another location using rsync.
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.
635
takes_args = ['location?']
636
takes_options = [Option('overwrite', help='Ignore differences between'
637
' branches and overwrite unconditionally'),
638
Option('no-tree', help='Do not push the working tree,'
641
def run(self, location=None, overwrite=False, no_tree=False):
642
from bzrlib import workingtree
644
cur_branch = workingtree.WorkingTree.open_containing(".")[0]
645
bzrtools.rspush(cur_branch, location, overwrite=overwrite,
646
working_tree=not no_tree)
649
class cmd_switch(bzrlib.commands.Command):
650
"""Set the branch of a lightweight checkout and update."""
652
takes_args = ['to_location']
654
def run(self, to_location):
655
from switch import cmd_switch
656
cmd_switch().run(to_location)
661
cmd_baz_import_branch,
456
This is useful when you start to make changes in one tree, then realize
457
they should really be done in a different tree.
459
Shove is implemented using merge, so:
460
- All changes, including renames and adds, will be applied.
461
- No changes that have already been applied will be applied.
462
- If the target is significantly different from the source, conflicts may
466
takes_args = ['target', 'source?']
467
def run(self, target, source='.'):
468
from shove import do_shove
469
do_shove(source, target)
472
commands = [cmd_shelve, cmd_unshelve, cmd_shelf, cmd_clean_tree,
473
cmd_graph_ancestry, cmd_fetch_ghosts, cmd_patch, cmd_shell,
474
cmd_branch_history, cmd_zap, cmd_cbranch, cmd_branches,
475
cmd_multi_pull, cmd_switch, cmd_branch_mark, cmd_import, cmd_shove]
478
import bzrlib.builtins
479
commands.append(rspush.cmd_rspush)
481
from errors import NoPyBaz
484
commands.append(baz_import.cmd_baz_import_branch)
485
commands.append(baz_import.cmd_baz_import)
488
class cmd_baz_import_branch(bzrlib.commands.Command):
489
"""Disabled. (Requires PyBaz) <BZRTOOLS>"""
490
takes_args = ['to_location?', 'from_branch?', 'reuse_history*']
491
takes_options = ['verbose', Option('max-count', type=int)]
492
def run(self, to_location=None, from_branch=None, fast=False,
493
max_count=None, verbose=False, dry_run=False,
494
reuse_history_list=[]):
495
print "This command is disabled. Please install PyBaz."
498
class cmd_baz_import(bzrlib.commands.Command):
499
"""Disabled. (Requires PyBaz) <BZRTOOLS>"""
500
takes_args = ['to_root_dir?', 'from_archive?', 'reuse_history*']
501
takes_options = ['verbose', Option('prefixes', type=str,
502
help="Prefixes of branches to import")]
503
def run(self, to_root_dir=None, from_archive=None, verbose=False,
504
reuse_history_list=[], prefixes=None):
505
print "This command is disabled. Please install PyBaz."
506
commands.extend((cmd_baz_import_branch, cmd_baz_import))
683
509
if hasattr(bzrlib.commands, 'register_command'):