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
25
from bzrlib.lazy_import import lazy_import
69
26
lazy_import(globals(), """
70
from bzrlib import help
27
from bzrlib import help, urlutils
74
from errors import CommandError, NoPyBaz
31
from version import version_info, __version__
32
from command import BzrToolsCommand
33
from errors import CommandError
75
34
from patchsource import BzrPatchSource
79
38
import bzrlib.builtins
80
39
import bzrlib.commands
40
from bzrlib.branch import Branch
41
from bzrlib.bzrdir import BzrDir
81
42
from bzrlib.commands import get_cmd_object
82
43
from bzrlib.errors import BzrCommandError
83
44
import bzrlib.ignores
45
from bzrlib.trace import note
84
46
from bzrlib.option import Option
85
47
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
51
from command import BzrToolsCommand
90
53
bzrlib.ignores.add_runtime_ignores(['./.shelf'])
93
class cmd_clean_tree(bzrlib.commands.Command):
56
class cmd_clean_tree(BzrToolsCommand):
94
57
"""Remove unwanted files from working tree.
96
59
By default, only unknown files, not ignored files, are deleted. Versioned
106
69
To check what clean-tree will do, use --dry-run.
108
takes_options = [Option('ignored', help='delete all ignored files.'),
109
Option('detritus', help='delete conflict files, merge'
71
takes_options = [Option('ignored', help='Delete all ignored files.'),
72
Option('detritus', help='Delete conflict files, merge'
110
73
' backups, and failed selftest dirs.'),
112
help='delete files unknown to bzr. (default)'),
113
Option('dry-run', help='show files to delete instead of'
75
help='Delete files unknown to bzr (default).'),
76
Option('dry-run', help='Show files to delete instead of'
114
77
' deleting them.')]
115
78
def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
116
79
from clean_tree import clean_tree
155
118
If available, rsvg is used to antialias PNG and JPEG output, but this can
156
119
be disabled with --no-antialias.
158
takes_args = ['branch', 'file']
159
takes_options = [Option('no-collapse', help="Do not skip simple nodes"),
121
takes_args = ['file', 'merge_branch?']
122
takes_options = [Option('no-collapse', help="Do not skip simple nodes."),
160
123
Option('no-antialias',
161
help="Do not use rsvg to produce antialiased output"),
124
help="Do not use rsvg to produce antialiased output."),
162
125
Option('merge-branch', type=str,
163
help="Use this branch to calcuate a merge base"),
126
help="Use this branch to calcuate a merge base."),
164
127
Option('cluster', help="Use clustered output."),
165
Option('max-distance', help="Show no nodes farther than this",
167
def run(self, branch, file, no_collapse=False, no_antialias=False,
168
merge_branch=None, cluster=False, max_distance=None):
128
Option('max-distance',
129
help="Show no nodes farther than this.", type=int),
131
help='Source branch to use (default is current'
136
def run(self, file, merge_branch=None, no_collapse=False,
137
no_antialias=False, cluster=False, max_distance=100,
139
if max_distance == -1:
171
143
ranking = "cluster"
173
145
ranking = "forced"
174
graph.write_ancestry_file(branch, file, not no_collapse,
146
graph.write_ancestry_file(directory, file, not no_collapse,
175
147
not no_antialias, merge_branch, ranking,
176
148
max_distance=max_distance)
179
class cmd_fetch_ghosts(bzrlib.commands.Command):
151
class cmd_fetch_ghosts(BzrToolsCommand):
180
152
"""Attempt to retrieve ghosts from another branch.
181
153
If the other branch is not supplied, the last-pulled branch is used.
183
155
aliases = ['fetch-missing']
184
156
takes_args = ['branch?']
185
takes_options = [Option('no-fix')]
157
takes_options = [Option('no-fix', help="Skip additional synchonization.")]
186
158
def run(self, branch=None, no_fix=False):
187
159
from fetch_ghosts import fetch_ghosts
188
160
fetch_ghosts(branch, no_fix)
239
211
takes_args = ['file*']
240
takes_options = ['message', 'revision',
241
Option('all', help='Shelve all changes without prompting'),
242
Option('no-color', help='Never display changes in color')]
212
takes_options = [Option('message',
213
help='A message to associate with the shelved changes.',
214
short_name='m', type=unicode),
216
Option('all', help='Shelve all changes without prompting.'),
217
Option('no-color', help='Never display changes in color.')]
244
219
def run(self, all=False, file_list=None, message=None, revision=None,
433
408
If --branch is specified, the branch will be deleted too, but only if the
434
409
the branch has no new commits (relative to its parent).
436
takes_options = [Option("branch", help="Remove associtated branch from"
411
takes_options = [Option("branch", help="Remove associated branch from"
413
Option('force', help='Delete tree even if contents are'
438
415
takes_args = ["checkout"]
439
def run(self, checkout, branch=False):
416
def run(self, checkout, branch=False, force=False):
440
417
from zap import zap
441
return zap(checkout, remove_branch=branch)
444
class cmd_cbranch(bzrlib.commands.Command):
418
return zap(checkout, remove_branch=branch, allow_modified=force)
421
class cmd_cbranch(BzrToolsCommand):
446
423
Create a new checkout, associated with a new repository branch.
463
440
takes_options = [Option("lightweight",
464
help="Create a lightweight checkout"), 'revision']
441
help="Create a lightweight checkout."), 'revision',
442
Option('files-from', type=unicode,
443
help='Accelerate checkout using files from this'
446
help='Hard-link files from source/files-from tree'
465
448
takes_args = ["source", "target?"]
466
def run(self, source, target=None, lightweight=False, revision=None):
449
def run(self, source, target=None, lightweight=False, revision=None,
450
files_from=None, hardlink=False):
467
451
from cbranch import cbranch
468
452
return cbranch(source, target, lightweight=lightweight,
472
class cmd_branches(bzrlib.commands.Command):
453
revision=revision, files_from=files_from,
457
class cmd_branches(BzrToolsCommand):
473
458
"""Scan a location for branches"""
474
459
takes_args = ["location?"]
475
460
def run(self, location=None):
476
461
from branches import branches
477
462
return branches(location)
464
class cmd_trees(BzrToolsCommand):
465
"""Scan a location for trees"""
466
takes_args = ['location?']
467
def run(self, location='.'):
468
from bzrlib.workingtree import WorkingTree
469
from bzrlib.transport import get_transport
470
t = get_transport(location)
471
for tree in WorkingTree.find_trees(location):
472
self.outf.write('%s\n' % t.relpath(
473
tree.bzrdir.root_transport.base))
480
class cmd_multi_pull(bzrlib.commands.Command):
475
class cmd_multi_pull(BzrToolsCommand):
481
476
"""Pull all the branches under a location, e.g. a repository.
483
478
Both branches present in the directory and the branches of checkouts are
513
508
print "Pulling %s from %s" % (relpath, parent)
515
pullable.pull(Branch.open(parent))
510
branch_t = get_transport(parent, possible_transports)
511
pullable.pull(Branch.open_from_transport(branch_t))
516
512
except Exception, e:
520
class cmd_branch_mark(bzrlib.commands.Command):
522
Add, view or list branch markers <EXPERIMENTAL>
524
To add a mark, do 'bzr branch-mark MARK'.
525
To list marks, do 'bzr branch-mark' (this lists all marks for the branch's
527
To delete a mark, do 'bzr branch-mark --delete MARK'
529
These marks can be used to track a branch's status.
531
takes_args = ['mark?', 'branch?']
532
takes_options = [Option('delete', help='Delete this mark')]
533
def run(self, mark=None, branch=None, delete=False):
534
from branch_mark import branch_mark
535
branch_mark(mark, branch, delete)
538
class cmd_import(bzrlib.commands.Command):
517
class cmd_import(BzrToolsCommand):
539
518
"""Import sources from a directory, tarball or zip file
541
520
This command will import a directory, tarball or zip file into a bzr
555
534
do_import(source, tree)
558
class cmd_cdiff(bzrlib.commands.Command):
537
class cmd_cdiff(BzrToolsCommand):
559
538
"""A color version of bzr's diff"""
560
539
takes_args = property(lambda x: get_cmd_object('diff').takes_args)
562
def _takes_options(self):
563
options = list(get_cmd_object('diff').takes_options)
564
options.append(Option('check-style',
540
takes_options = list(get_cmd_object('diff').takes_options) + [
541
Option('check-style',
565
542
help='Warn if trailing whitespace or spurious changes have been'
569
takes_options = property(_takes_options)
571
545
def run(self, check_style=False, *args, **kwargs):
572
546
from colordiff import colordiff
573
547
colordiff(check_style, *args, **kwargs)
576
class cmd_baz_import(bzrlib.commands.Command):
577
"""Import an Arch or Baz archive into a bzr repository.
579
This command should be used on local archives (or mirrors) only. It is
580
quite slow on remote archives.
582
reuse_history allows you to specify any previous imports you
583
have done of different archives, which this archive has branches
584
tagged from. This will dramatically reduce the time to convert
585
the archive as it will not have to convert the history already
586
converted in that other branch.
588
If you specify prefixes, only branches whose names start with that prefix
589
will be imported. Skipped branches will be listed, so you can import any
590
branches you missed by accident. Here's an example of doing a partial
591
import from thelove@canonical.com:
592
bzr baz-import thelove thelove@canonical.com --prefixes dists:talloc-except
594
WARNING: Encoding should not be specified unless necessary, because if you
595
specify an encoding, your converted branch will not interoperate with
596
independently-converted branches, unless the other branches were converted
597
with exactly the same encoding. Any encoding recognized by Python may
598
be specified. Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
601
takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
602
takes_options = ['verbose', Option('prefixes', type=str,
603
help="Prefixes of branches to import, colon-separated"),
604
Option('encoding', type=str,
605
help='Force encoding to specified value. See WARNING.')]
607
def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
608
reuse_history_list=[], prefixes=None):
609
from errors import NoPyBaz
612
baz_import.baz_import(to_root_dir, from_archive, encoding,
613
verbose, reuse_history_list, prefixes)
615
print "This command is disabled. Please install PyBaz."
618
class cmd_baz_import_branch(bzrlib.commands.Command):
619
"""Import an Arch or Baz branch into a bzr branch.
621
WARNING: Encoding should not be specified unless necessary, because if you
622
specify an encoding, your converted branch will not interoperate with
623
independently-converted branches, unless the other branches were converted
624
with exactly the same encoding. Any encoding recognized by Python may
625
be specified. Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
628
takes_args = ['to_location', 'from_branch?', 'reuse_history*']
629
takes_options = ['verbose', Option('max-count', type=int),
630
Option('encoding', type=str,
631
help='Force encoding to specified value. See WARNING.')]
633
def run(self, to_location, from_branch=None, fast=False, max_count=None,
634
encoding=None, verbose=False, dry_run=False,
635
reuse_history_list=[]):
636
from errors import NoPyBaz
639
baz_import.baz_import_branch(to_location, from_branch, fast,
640
max_count, verbose, encoding, dry_run,
643
print "This command is disabled. Please install PyBaz."
646
class cmd_rspush(bzrlib.commands.Command):
550
class cmd_rspush(BzrToolsCommand):
647
551
"""Upload this branch to another location using rsync.
649
553
If no location is specified, the last-used location will be used. To
665
569
working_tree=not no_tree)
668
class cmd_switch(bzrlib.commands.Command):
669
"""Set the branch of a lightweight checkout and update."""
671
takes_args = ['to_location']
673
def run(self, to_location):
674
from switch import cmd_switch
675
cmd_switch().run(to_location)
572
class cmd_link_tree(BzrToolsCommand):
573
"""Hardlink matching files to another tree.
575
Only files with identical content and execute bit will be linked.
577
takes_args = ['location']
579
def run(self, location):
580
from bzrlib import workingtree
581
from bzrlib.plugins.bzrtools.link_tree import link_tree
582
target_tree = workingtree.WorkingTree.open_containing(".")[0]
583
source_tree = workingtree.WorkingTree.open(location)
584
target_tree.lock_write()
586
source_tree.lock_read()
588
link_tree(target_tree, source_tree)
594
from heads import cmd_heads
680
cmd_baz_import_branch,
682
597
cmd_branch_history,
687
601
cmd_fetch_ghosts,
688
602
cmd_graph_ancestry,