/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/commands.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
27
27
 
28
28
from bzrlib.lazy_import import lazy_import
29
29
lazy_import(globals(), """
 
30
import codecs
30
31
import errno
31
32
import threading
 
33
from warnings import warn
32
34
 
33
35
import bzrlib
34
36
from bzrlib import (
40
42
    osutils,
41
43
    trace,
42
44
    ui,
 
45
    win32utils,
43
46
    )
44
47
""")
45
48
 
46
 
from bzrlib.hooks import Hooks
 
49
from bzrlib.hooks import HookPoint, Hooks
47
50
# Compatibility - Option used to be in commands.
48
51
from bzrlib.option import Option
49
52
from bzrlib.plugin import disable_plugins, load_plugins
219
222
    Use of all_command_names() is encouraged rather than builtin_command_names
220
223
    and/or plugin_command_names.
221
224
    """
222
 
    _register_builtin_commands()
223
225
    return builtin_command_registry.keys()
224
226
 
225
227
 
395
397
            will not mangled.
396
398
 
397
399
    :cvar hooks: An instance of CommandHooks.
398
 
    :ivar __doc__: The help shown by 'bzr help command' for this command.
399
 
        This is set by assigning explicitly to __doc__ so that -OO can
400
 
        be used::
401
 
 
402
 
        class Foo(Command):
403
 
            __doc__ = "My help goes here"
404
400
    """
405
401
    aliases = []
406
402
    takes_args = []
411
407
 
412
408
    def __init__(self):
413
409
        """Construct an instance of this command."""
 
410
        if self.__doc__ == Command.__doc__:
 
411
            warn("No help message set for %r" % self)
414
412
        # List of standard options directly supported
415
413
        self.supported_std_options = []
416
414
        self._setup_run()
484
482
            message explaining how to obtain full help.
485
483
        """
486
484
        doc = self.help()
487
 
        if not doc:
488
 
            doc = "No help for this command."
 
485
        if doc is None:
 
486
            raise NotImplementedError("sorry, no detailed help yet for %r" % self.name())
489
487
 
490
488
        # Extract the summary (purpose) and sections out from the text
491
489
        purpose,sections,order = self._get_help_parts(doc)
684
682
 
685
683
        self._setup_outf()
686
684
 
687
 
        try:
688
 
            return self.run(**all_cmd_args)
689
 
        finally:
690
 
            # reset it, so that other commands run in the same process won't
691
 
            # inherit state. Before we reset it, log any activity, so that it
692
 
            # gets properly tracked.
693
 
            ui.ui_factory.log_transport_activity(
694
 
                display=('bytes' in debug.debug_flags))
695
 
            trace.set_verbosity_level(0)
 
685
        return self.run(**all_cmd_args)
696
686
 
697
687
    def _setup_run(self):
698
688
        """Wrap the defined run method on self with a cleanup.
772
762
        These are all empty initially, because by default nothing should get
773
763
        notified.
774
764
        """
775
 
        Hooks.__init__(self, "bzrlib.commands", "Command.hooks")
776
 
        self.add_hook('extend_command',
 
765
        Hooks.__init__(self)
 
766
        self.create_hook(HookPoint('extend_command',
777
767
            "Called after creating a command object to allow modifications "
778
768
            "such as adding or removing options, docs etc. Called with the "
779
 
            "new bzrlib.commands.Command object.", (1, 13))
780
 
        self.add_hook('get_command',
 
769
            "new bzrlib.commands.Command object.", (1, 13), None))
 
770
        self.create_hook(HookPoint('get_command',
781
771
            "Called when creating a single command. Called with "
782
772
            "(cmd_or_None, command_name). get_command should either return "
783
773
            "the cmd_or_None parameter, or a replacement Command object that "
784
774
            "should be used for the command. Note that the Command.hooks "
785
775
            "hooks are core infrastructure. Many users will prefer to use "
786
776
            "bzrlib.commands.register_command or plugin_cmds.register_lazy.",
787
 
            (1, 17))
788
 
        self.add_hook('get_missing_command',
 
777
            (1, 17), None))
 
778
        self.create_hook(HookPoint('get_missing_command',
789
779
            "Called when creating a single command if no command could be "
790
780
            "found. Called with (command_name). get_missing_command should "
791
781
            "either return None, or a Command object to be used for the "
792
 
            "command.", (1, 17))
793
 
        self.add_hook('list_commands',
 
782
            "command.", (1, 17), None))
 
783
        self.create_hook(HookPoint('list_commands',
794
784
            "Called when enumerating commands. Called with a set of "
795
785
            "cmd_name strings for all the commands found so far. This set "
796
786
            " is safe to mutate - e.g. to remove a command. "
797
787
            "list_commands should return the updated set of command names.",
798
 
            (1, 17))
 
788
            (1, 17), None))
799
789
 
800
790
Command.hooks = CommandHooks()
801
791
 
815
805
    else:
816
806
        args = argv
817
807
 
818
 
    # for python 2.5 and later, optparse raises this exception if a non-ascii
819
 
    # option name is given.  See http://bugs.python.org/issue2931
820
 
    try:
821
 
        options, args = parser.parse_args(args)
822
 
    except UnicodeEncodeError,e:
823
 
        raise errors.BzrCommandError('Only ASCII permitted in option names')
824
 
 
 
808
    options, args = parser.parse_args(args)
825
809
    opts = dict([(k, v) for k, v in options.__dict__.iteritems() if
826
810
                 v is not option.OptionParser.DEFAULT_VALUE])
827
811
    return args, opts
1066
1050
        elif a == '--coverage':
1067
1051
            opt_coverage_dir = argv[i + 1]
1068
1052
            i += 1
1069
 
        elif a == '--profile-imports':
1070
 
            pass # already handled in startup script Bug #588277
1071
1053
        elif a.startswith('-D'):
1072
1054
            debug.debug_flags.add(a[2:])
1073
1055
        else:
1095
1077
    if not opt_no_aliases:
1096
1078
        alias_argv = get_alias(argv[0])
1097
1079
        if alias_argv:
 
1080
            user_encoding = osutils.get_user_encoding()
 
1081
            alias_argv = [a.decode(user_encoding) for a in alias_argv]
1098
1082
            argv[0] = alias_argv.pop(0)
1099
1083
 
1100
1084
    cmd = argv.pop(0)
 
1085
    # We want only 'ascii' command names, but the user may have typed
 
1086
    # in a Unicode name. In that case, they should just get a
 
1087
    # 'command not found' error later.
 
1088
 
1101
1089
    cmd_obj = get_cmd_object(cmd, plugins_override=not opt_builtin)
1102
1090
    run = cmd_obj.run_argv_aliases
1103
1091
    run_argv = [argv, alias_argv]
1202
1190
    argv = _specified_or_unicode_argv(argv)
1203
1191
    _register_builtin_commands()
1204
1192
    ret = run_bzr_catch_errors(argv)
 
1193
    bzrlib.ui.ui_factory.log_transport_activity(
 
1194
        display=('bytes' in debug.debug_flags))
1205
1195
    trace.mutter("return code %d", ret)
1206
1196
    return ret
1207
1197