242
217
class ExternalCommand(Command):
243
218
"""Class to wrap external commands.
245
We cheat a little here, when get_cmd_class() calls us we actually
246
give it back an object we construct that has the appropriate path,
247
help, options etc for the specified command.
249
When run_bzr() tries to instantiate that 'class' it gets caught by
250
the __call__ method, which we override to call the Command.__init__
251
method. That then calls our run method which is pretty straight
254
The only wrinkle is that we have to map bzr's dictionary of options
255
and arguments back into command line options and arguments for the
220
We cheat a little here, when get_cmd_class() calls us we actually give it back
221
an object we construct that has the appropriate path, help, options etc for the
224
When run_bzr() tries to instantiate that 'class' it gets caught by the __call__
225
method, which we override to call the Command.__init__ method. That then calls
226
our run method which is pretty straight forward.
228
The only wrinkle is that we have to map bzr's dictionary of options and arguments
229
back into command line options and arguments for the script.
259
232
def find_command(cls, cmd):
539
class cmd_mv(Command):
540
"""Move or rename a file.
543
bzr mv OLDNAME NEWNAME
544
bzr mv SOURCE... DESTINATION
546
If the last argument is a versioned directory, all the other names
547
are moved into it. Otherwise, there must be exactly two arguments
548
and the file is changed to a new name, which must not already exist.
550
Files cannot be moved between branches.
552
takes_args = ['names*']
553
def run(self, names_list):
554
if len(names_list) < 2:
555
raise BzrCommandError("missing file argument")
556
b = find_branch(names_list[0])
558
rel_names = [b.relpath(x) for x in names_list]
560
if os.path.isdir(names_list[-1]):
561
# move into existing directory
562
b.move(rel_names[:-1], rel_names[-1])
564
if len(names_list) != 2:
565
raise BzrCommandError('to mv multiple files the destination '
566
'must be a versioned directory')
567
b.move(rel_names[0], rel_names[1])
572
510
class cmd_pull(Command):
640
578
takes_args = ['from_location', 'to_location?']
641
579
takes_options = ['revision']
642
aliases = ['get', 'clone']
644
581
def run(self, from_location, to_location=None, revision=None):
646
583
from bzrlib.merge import merge
647
from bzrlib.branch import DivergedBranches, \
584
from bzrlib.branch import DivergedBranches, NoSuchRevision, \
648
585
find_cached_branch, Branch
649
586
from shutil import rmtree
650
587
from meta_store import CachedStore
652
589
cache_root = tempfile.mkdtemp()
656
elif len(revision) > 1:
657
raise BzrCommandError('bzr branch --revision takes exactly 1 revision value')
591
if revision is not None:
592
if len(revision) > 1:
593
raise BzrCommandError('bzr branch --revision takes exactly 1 revision value')
661
596
br_from = find_cached_branch(from_location, cache_root)
683
618
br_to = Branch(to_location, init=True)
685
br_to.set_root_id(br_from.get_root_id())
688
if revision[0] is None:
689
revno = br_from.revno()
691
revno, rev_id = br_from.get_revision_info(revision[0])
693
br_to.update_revisions(br_from, stop_revision=revno)
694
except bzrlib.errors.NoSuchRevision:
696
msg = "The branch %s has no revision %d." % (from_location,
698
raise BzrCommandError(msg)
620
revno = br_to.lookup_revision(revision[0])
622
br_to.update_revisions(br_from, stop_revision=revno)
623
except NoSuchRevision:
625
msg = "The branch %s has no revision %d." % (from_location,
627
raise BzrCommandError(msg)
700
628
merge((to_location, -1), (to_location, 0), this_dir=to_location,
701
629
check_clean=False, ignore_zero=True)
702
630
from_location = pull_loc(br_from)
1455
1353
--force is given.
1457
1355
takes_args = ['other_spec', 'base_spec?']
1458
takes_options = ['force', 'merge-type']
1356
takes_options = ['force']
1460
def run(self, other_spec, base_spec=None, force=False, merge_type=None):
1358
def run(self, other_spec, base_spec=None, force=False):
1461
1359
from bzrlib.merge import merge
1462
from bzrlib.merge_core import ApplyMerge3
1463
if merge_type is None:
1464
merge_type = ApplyMerge3
1465
1360
merge(parse_spec(other_spec), parse_spec(base_spec),
1466
check_clean=(not force), merge_type=merge_type)
1361
check_clean=(not force))
1469
1365
class cmd_revert(Command):
1366
"""Restore selected files from a previous revision.
1368
takes_args = ['file+']
1369
def run(self, file_list):
1370
from bzrlib.branch import find_branch
1375
b = find_branch(file_list[0])
1377
b.revert([b.relpath(f) for f in file_list])
1380
class cmd_merge_revert(Command):
1470
1381
"""Reverse all changes since the last commit.
1472
Only versioned files are affected. Specify filenames to revert only
1473
those files. By default, any files that are changed will be backed up
1474
first. Backup files have a '~' appended to their name.
1383
Only versioned files are affected.
1385
TODO: Store backups of any files that will be reverted, so
1386
that the revert can be undone.
1476
takes_options = ['revision', 'no-backup']
1477
takes_args = ['file*']
1478
aliases = ['merge-revert']
1388
takes_options = ['revision']
1480
def run(self, revision=None, no_backup=False, file_list=None):
1390
def run(self, revision=None):
1481
1391
from bzrlib.merge import merge
1482
if file_list is not None:
1483
if len(file_list) == 0:
1484
raise BzrCommandError("No files specified")
1485
1392
if revision is None:
1487
1394
elif len(revision) != 1:
1488
raise BzrCommandError('bzr revert --revision takes exactly 1 argument')
1395
raise BzrCommandError('bzr merge-revert --revision takes exactly 1 argument')
1489
1396
merge(('.', revision[0]), parse_spec('.'),
1490
1397
check_clean=False,
1492
backup_files=not no_backup,
1493
file_list=file_list)
1496
1401
class cmd_assert_fail(Command):
1611
def _parse_master_args(argv):
1612
"""Parse the arguments that always go with the original command.
1613
These are things like bzr --no-plugins, etc.
1615
There are now 2 types of option flags. Ones that come *before* the command,
1616
and ones that come *after* the command.
1617
Ones coming *before* the command are applied against all possible commands.
1618
And are generally applied before plugins are loaded.
1620
The current list are:
1621
--builtin Allow plugins to load, but don't let them override builtin commands,
1622
they will still be allowed if they do not override a builtin.
1623
--no-plugins Don't load any plugins. This lets you get back to official source
1625
--profile Enable the hotspot profile before running the command.
1626
For backwards compatibility, this is also a non-master option.
1627
--version Spit out the version of bzr that is running and exit.
1628
This is also a non-master option.
1629
--help Run help and exit, also a non-master option (I think that should stay, though)
1631
>>> argv, opts = _parse_master_args(['bzr', '--test'])
1632
Traceback (most recent call last):
1634
BzrCommandError: Invalid master option: 'test'
1635
>>> argv, opts = _parse_master_args(['bzr', '--version', 'command'])
1638
>>> print opts['version']
1640
>>> argv, opts = _parse_master_args(['bzr', '--profile', 'command', '--more-options'])
1642
['command', '--more-options']
1643
>>> print opts['profile']
1645
>>> argv, opts = _parse_master_args(['bzr', '--no-plugins', 'command'])
1648
>>> print opts['no-plugins']
1650
>>> print opts['profile']
1652
>>> argv, opts = _parse_master_args(['bzr', 'command', '--profile'])
1654
['command', '--profile']
1655
>>> print opts['profile']
1658
master_opts = {'builtin':False,
1665
# This is the point where we could hook into argv[0] to determine
1666
# what front-end is supposed to be run
1667
# For now, we are just ignoring it.
1668
cmd_name = argv.pop(0)
1670
if arg[:2] != '--': # at the first non-option, we return the rest
1672
arg = arg[2:] # Remove '--'
1673
if arg not in master_opts:
1674
# We could say that this is not an error, that we should
1675
# just let it be handled by the main section instead
1676
raise BzrCommandError('Invalid master option: %r' % arg)
1677
argv.pop(0) # We are consuming this entry
1678
master_opts[arg] = True
1679
return argv, master_opts
1716
1683
def run_bzr(argv):
1717
1684
"""Execute a command.
1719
1686
This is similar to main(), but without all the trappings for
1720
1687
logging and error handling.
1723
The command-line arguments, without the program name from argv[0]
1725
Returns a command status or raises an exception.
1727
Special master options: these must come before the command because
1728
they control how the command is interpreted.
1731
Do not load plugin modules at all
1734
Only use builtin commands. (Plugins are still allowed to change
1738
Run under the Python profiler.
1741
1689
argv = [a.decode(bzrlib.user_encoding) for a in argv]
1743
opt_profile = opt_no_plugins = opt_builtin = False
1745
# --no-plugins is handled specially at a very early stage. We need
1746
# to load plugins before doing other command parsing so that they
1747
# can override commands, but this needs to happen first.
1750
if a == '--profile':
1752
elif a == '--no-plugins':
1753
opt_no_plugins = True
1754
elif a == '--builtin':
1760
if not opt_no_plugins:
1761
from bzrlib.plugin import load_plugins
1764
args, opts = parse_args(argv)
1767
from bzrlib.help import help
1774
if 'version' in opts:
1779
print >>sys.stderr, "please try 'bzr help' for help"
1692
# some options like --builtin and --no-plugins have special effects
1693
argv, master_opts = _parse_master_args(argv)
1694
if not master_opts['no-plugins']:
1695
from bzrlib.plugin import load_plugins
1698
args, opts = parse_args(argv)
1700
if master_opts['help']:
1701
from bzrlib.help import help
1709
from bzrlib.help import help
1715
elif 'version' in opts:
1718
elif args and args[0] == 'builtin':
1719
include_plugins=False
1721
cmd = str(args.pop(0))
1782
cmd = str(args.pop(0))
1784
canonical_cmd, cmd_class = \
1785
get_cmd_class(cmd, plugins_override=not opt_builtin)
1728
plugins_override = not (master_opts['builtin'])
1729
canonical_cmd, cmd_class = get_cmd_class(cmd, plugins_override=plugins_override)
1731
profile = master_opts['profile']
1732
# For backwards compatibility, I would rather stick with --profile being a
1733
# master/global option
1734
if 'profile' in opts:
1787
1738
# check options are reasonable
1788
1739
allowed = cmd_class.takes_options