← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: Patch Queue Manager
  • Date: 2016-02-01 19:13:13 UTC
  • mfrom: (6614.2.2 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20160201191313-wdfvmfff1djde6oq
(vila) Release 2.7.0 (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/commands.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
17
18
 
18
19
# TODO: Define arguments by objects, rather than just using names.
19
20
# Those objects can specify the expected type of the argument, which
32
33
 
33
34
import bzrlib
34
35
from bzrlib import (
 
36
    config,
35
37
    cleanup,
36
38
    cmdline,
37
39
    debug,
50
52
from bzrlib.option import Option
51
53
from bzrlib.plugin import disable_plugins, load_plugins
52
54
from bzrlib import registry
53
 
from bzrlib.symbol_versioning import (
54
 
    deprecated_function,
55
 
    deprecated_in,
56
 
    deprecated_method,
57
 
    )
58
55
 
59
56
 
60
57
class CommandInfo(object):
229
226
    try:
230
227
        return _get_cmd_object(cmd_name, plugins_override)
231
228
    except KeyError:
232
 
        raise errors.BzrCommandError('unknown command "%s"' % cmd_name)
 
229
        raise errors.BzrCommandError(gettext('unknown command "%s"') % cmd_name)
233
230
 
234
231
 
235
232
def _get_cmd_object(cmd_name, plugins_override=True, check_missing=True):
465
462
            usage help (e.g. Purpose, Usage, Options) with a
466
463
            message explaining how to obtain full help.
467
464
        """
468
 
        if self.l10n and not i18n.installed():
 
465
        if self.l10n:
469
466
            i18n.install()  # Install i18n only for get_help_text for now.
470
467
        doc = self.help()
471
468
        if doc:
554
551
                        see_also_links.append(item)
555
552
                    else:
556
553
                        # Use a Sphinx link for this entry
557
 
                        link_text = gettext(":doc:`%s <%s-help>`") % (item, item)
 
554
                        link_text = gettext(":doc:`{0} <{1}-help>`").format(
 
555
                                                                    item, item)
558
556
                        see_also_links.append(link_text)
559
557
                see_also = see_also_links
560
558
            result += gettext(':See also: %s') % ', '.join(see_also) + '\n'
662
660
            opts['quiet'] = trace.is_quiet()
663
661
        elif opts.has_key('quiet'):
664
662
            del opts['quiet']
665
 
 
666
663
        # mix arguments and options into one dictionary
667
664
        cmdargs = _match_argform(self.name(), self.takes_args, args)
668
665
        cmdopts = {}
693
690
        """
694
691
        class_run = self.run
695
692
        def run(*args, **kwargs):
 
693
            for hook in Command.hooks['pre_command']:
 
694
                hook(self)
696
695
            self._operation = cleanup.OperationWithCleanups(class_run)
697
696
            try:
698
697
                return self._operation.run_simple(*args, **kwargs)
699
698
            finally:
700
699
                del self._operation
 
700
                for hook in Command.hooks['post_command']:
 
701
                    hook(self)
701
702
        self.run = run
702
703
 
703
704
    def run(self):
791
792
            " is safe to mutate - e.g. to remove a command. "
792
793
            "list_commands should return the updated set of command names.",
793
794
            (1, 17))
 
795
        self.add_hook('pre_command',
 
796
            "Called prior to executing a command. Called with the command "
 
797
            "object.", (2, 6))
 
798
        self.add_hook('post_command',
 
799
            "Called after executing a command. Called with the command "
 
800
            "object.", (2, 6))
794
801
 
795
802
Command.hooks = CommandHooks()
796
803
 
815
822
    try:
816
823
        options, args = parser.parse_args(args)
817
824
    except UnicodeEncodeError,e:
818
 
        raise errors.BzrCommandError('Only ASCII permitted in option names')
 
825
        raise errors.BzrCommandError(
 
826
            gettext('Only ASCII permitted in option names'))
819
827
 
820
828
    opts = dict([(k, v) for k, v in options.__dict__.iteritems() if
821
829
                 v is not option.OptionParser.DEFAULT_VALUE])
839
847
                argdict[argname + '_list'] = None
840
848
        elif ap[-1] == '+':
841
849
            if not args:
842
 
                raise errors.BzrCommandError("command %r needs one or more %s"
843
 
                                             % (cmd, argname.upper()))
 
850
                raise errors.BzrCommandError(gettext(
 
851
                      "command {0!r} needs one or more {1}").format(
 
852
                      cmd, argname.upper()))
844
853
            else:
845
854
                argdict[argname + '_list'] = args[:]
846
855
                args = []
847
856
        elif ap[-1] == '$': # all but one
848
857
            if len(args) < 2:
849
 
                raise errors.BzrCommandError("command %r needs one or more %s"
850
 
                                             % (cmd, argname.upper()))
 
858
                raise errors.BzrCommandError(
 
859
                      gettext("command {0!r} needs one or more {1}").format(
 
860
                                             cmd, argname.upper()))
851
861
            argdict[argname + '_list'] = args[:-1]
852
862
            args[:-1] = []
853
863
        else:
854
864
            # just a plain arg
855
865
            argname = ap
856
866
            if not args:
857
 
                raise errors.BzrCommandError("command %r requires argument %s"
858
 
                               % (cmd, argname.upper()))
 
867
                raise errors.BzrCommandError(
 
868
                     gettext("command {0!r} requires argument {1}").format(
 
869
                               cmd, argname.upper()))
859
870
            else:
860
871
                argdict[argname] = args.pop(0)
861
872
 
862
873
    if args:
863
 
        raise errors.BzrCommandError("extra argument to command %s: %s"
864
 
                                     % (cmd, args[0]))
 
874
        raise errors.BzrCommandError( gettext(
 
875
                              "extra argument to command {0}: {1}").format(
 
876
                                       cmd, args[0]) )
865
877
 
866
878
    return argdict
867
879
 
923
935
        exitcode = trace.report_exception(exc_info, sys.stderr)
924
936
        if os.environ.get('BZR_PDB'):
925
937
            print '**** entering debugger'
926
 
            tb = exc_info[2]
927
938
            import pdb
928
 
            if sys.version_info[:2] < (2, 6):
929
 
                # XXX: we want to do
930
 
                #    pdb.post_mortem(tb)
931
 
                # but because pdb.post_mortem gives bad results for tracebacks
932
 
                # from inside generators, we do it manually.
933
 
                # (http://bugs.python.org/issue4150, fixed in Python 2.6)
934
 
 
935
 
                # Setup pdb on the traceback
936
 
                p = pdb.Pdb()
937
 
                p.reset()
938
 
                p.setup(tb.tb_frame, tb)
939
 
                # Point the debugger at the deepest frame of the stack
940
 
                p.curindex = len(p.stack) - 1
941
 
                p.curframe = p.stack[p.curindex][0]
942
 
                # Start the pdb prompt.
943
 
                p.print_stack_entry(p.stack[p.curindex])
944
 
                p.execRcLines()
945
 
                p.cmdloop()
946
 
            else:
947
 
                pdb.post_mortem(tb)
 
939
            pdb.post_mortem(exc_info[2])
948
940
        return exitcode
949
941
 
950
942
 
957
949
        stats.pprint()
958
950
    else:
959
951
        stats.save(filename)
960
 
        trace.note('Profile data written to "%s".', filename)
 
952
        trace.note(gettext('Profile data written to "%s".'), filename)
961
953
    return ret
962
954
 
963
955
 
1035
1027
 
1036
1028
    argv_copy = []
1037
1029
    i = 0
 
1030
    override_config = []
1038
1031
    while i < len(argv):
1039
1032
        a = argv[i]
1040
1033
        if a == '--profile':
1063
1056
            pass # already handled in startup script Bug #588277
1064
1057
        elif a.startswith('-D'):
1065
1058
            debug.debug_flags.add(a[2:])
 
1059
        elif a.startswith('-O'):
 
1060
            override_config.append(a[2:])
1066
1061
        else:
1067
1062
            argv_copy.append(a)
1068
1063
        i += 1
1069
1064
 
 
1065
    if bzrlib.global_state is None:
 
1066
        # FIXME: Workaround for users that imported bzrlib but didn't call
 
1067
        # bzrlib.initialize -- vila 2012-01-19
 
1068
        cmdline_overrides = config.CommandLineStore()
 
1069
    else:
 
1070
        cmdline_overrides = bzrlib.global_state.cmdline_overrides
 
1071
    cmdline_overrides._from_cmdline(override_config)
 
1072
 
1070
1073
    debug.set_debug_flags_from_config()
1071
1074
 
1072
1075
    if not opt_no_plugins:
1124
1127
        if 'memory' in debug.debug_flags:
1125
1128
            trace.debug_memory('Process status after command:', short=False)
1126
1129
        option._verbosity_level = saved_verbosity_level
 
1130
        # Reset the overrides 
 
1131
        cmdline_overrides._reset()
1127
1132
 
1128
1133
 
1129
1134
def display_command(func):
1158
1163
        "bzr plugin commands")
1159
1164
    Command.hooks.install_named_hook("get_command", _get_external_command,
1160
1165
        "bzr external command lookup")
1161
 
    Command.hooks.install_named_hook("get_missing_command", _try_plugin_provider,
1162
 
        "bzr plugin-provider-db check")
 
1166
    Command.hooks.install_named_hook("get_missing_command",
 
1167
                                     _try_plugin_provider,
 
1168
                                     "bzr plugin-provider-db check")
1163
1169
 
1164
1170
 
1165
1171