~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: Aaron Bentley
  • Date: 2007-02-06 14:52:16 UTC
  • mfrom: (2266 +trunk)
  • mto: This revision was merged to the branch mainline in revision 2268.
  • Revision ID: abentley@panoramicfeedback.com-20070206145216-fcpi8o3ufvuzwbp9
Merge bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
43
43
    errors,
44
44
    option,
45
45
    osutils,
46
 
    registry,
47
46
    trace,
48
 
    win32utils,
49
47
    )
50
48
""")
51
49
 
52
50
from bzrlib.symbol_versioning import (
53
51
    deprecated_function,
54
52
    deprecated_method,
 
53
    zero_eight,
 
54
    zero_eleven,
55
55
    )
56
56
# Compatibility
57
57
from bzrlib.option import Option
86
86
    else:
87
87
        trace.log_error('Two plugins defined the same command: %r' % k)
88
88
        trace.log_error('Not loading the one in %r' % sys.modules[cmd.__module__])
89
 
        trace.log_error('Previously this command was registered from %r' %
90
 
                        sys.modules[plugin_cmds[k_unsquished].__module__])
91
89
 
92
90
 
93
91
def _squish_command_name(cmd):
95
93
 
96
94
 
97
95
def _unsquish_command_name(cmd):
 
96
    assert cmd.startswith("cmd_")
98
97
    return cmd[4:].replace('_','-')
99
98
 
100
99
 
138
137
    plugins_override
139
138
        If true, plugin commands can override builtins.
140
139
    """
141
 
    try:
142
 
        return _get_cmd_object(cmd_name, plugins_override)
143
 
    except KeyError:
144
 
        raise errors.BzrCommandError('unknown command "%s"' % cmd_name)
145
 
 
146
 
 
147
 
def _get_cmd_object(cmd_name, plugins_override=True):
148
 
    """Worker for get_cmd_object which raises KeyError rather than BzrCommandError."""
149
140
    from bzrlib.externalcommand import ExternalCommand
150
141
 
151
142
    # We want only 'ascii' command names, but the user may have typed
169
160
    if cmd_obj:
170
161
        return cmd_obj
171
162
 
172
 
    # look for plugins that provide this command but aren't installed
173
 
    for provider in command_providers_registry:
174
 
        try:
175
 
            plugin_metadata = provider.plugin_for_command(cmd_name)
176
 
        except errors.NoPluginAvailable:
177
 
            pass
178
 
        else:
179
 
            raise errors.CommandAvailableInPlugin(cmd_name, 
180
 
                                                  plugin_metadata, provider)
181
 
 
182
 
    raise KeyError
 
163
    raise errors.BzrCommandError('unknown command "%s"' % cmd_name)
183
164
 
184
165
 
185
166
class Command(object):
251
232
        """Construct an instance of this command."""
252
233
        if self.__doc__ == Command.__doc__:
253
234
            warn("No help message set for %r" % self)
254
 
        # List of standard options directly supported
255
 
        self.supported_std_options = []
256
 
 
257
 
    def _maybe_expand_globs(self, file_list):
258
 
        """Glob expand file_list if the platform does not do that itself.
259
 
        
260
 
        :return: A possibly empty list of unicode paths.
261
 
 
262
 
        Introduced in bzrlib 0.18.
263
 
        """
264
 
        if not file_list:
265
 
            file_list = []
266
 
        if sys.platform == 'win32':
267
 
            file_list = win32utils.glob_expand(file_list)
268
 
        return list(file_list)
269
 
 
270
 
    def _usage(self):
271
 
        """Return single-line grammar for this command.
272
 
 
273
 
        Only describes arguments, not options.
274
 
        """
275
 
        s = 'bzr ' + self.name() + ' '
276
 
        for aname in self.takes_args:
277
 
            aname = aname.upper()
278
 
            if aname[-1] in ['$', '+']:
279
 
                aname = aname[:-1] + '...'
280
 
            elif aname[-1] == '?':
281
 
                aname = '[' + aname[:-1] + ']'
282
 
            elif aname[-1] == '*':
283
 
                aname = '[' + aname[:-1] + '...]'
284
 
            s += aname + ' '
285
 
        s = s[:-1]      # remove last space
286
 
        return s
287
 
 
288
 
    def get_help_text(self, additional_see_also=None, plain=True,
289
 
                      see_also_as_links=False):
290
 
        """Return a text string with help for this command.
291
 
        
292
 
        :param additional_see_also: Additional help topics to be
293
 
            cross-referenced.
294
 
        :param plain: if False, raw help (reStructuredText) is
295
 
            returned instead of plain text.
296
 
        :param see_also_as_links: if True, convert items in 'See also'
297
 
            list to internal links (used by bzr_man rstx generator)
298
 
        """
299
 
        doc = self.help()
300
 
        if doc is None:
301
 
            raise NotImplementedError("sorry, no detailed help yet for %r" % self.name())
302
 
 
303
 
        # Extract the summary (purpose) and sections out from the text
304
 
        purpose,sections = self._get_help_parts(doc)
305
 
 
306
 
        # If a custom usage section was provided, use it
307
 
        if sections.has_key('Usage'):
308
 
            usage = sections.pop('Usage')
309
 
        else:
310
 
            usage = self._usage()
311
 
 
312
 
        # The header is the purpose and usage
313
 
        result = ""
314
 
        result += ':Purpose: %s\n' % purpose
315
 
        if usage.find('\n') >= 0:
316
 
            result += ':Usage:\n%s\n' % usage
317
 
        else:
318
 
            result += ':Usage:   %s\n' % usage
319
 
        result += '\n'
320
 
 
321
 
        # Add the options
322
 
        options = option.get_optparser(self.options()).format_option_help()
323
 
        if options.startswith('Options:'):
324
 
            result += ':' + options
325
 
        elif options.startswith('options:'):
326
 
            # Python 2.4 version of optparse
327
 
            result += ':Options:' + options[len('options:'):]
328
 
        else:
329
 
            result += options
330
 
        result += '\n'
331
 
 
332
 
        # Add the description, indenting it 2 spaces
333
 
        # to match the indentation of the options
334
 
        if sections.has_key(None):
335
 
            text = sections.pop(None)
336
 
            text = '\n  '.join(text.splitlines())
337
 
            result += ':%s:\n  %s\n\n' % ('Description',text)
338
 
 
339
 
        # Add the custom sections (e.g. Examples). Note that there's no need
340
 
        # to indent these as they must be indented already in the source.
341
 
        if sections:
342
 
            labels = sorted(sections.keys())
343
 
            for label in labels:
344
 
                result += ':%s:\n%s\n\n' % (label,sections[label])
345
 
 
346
 
        # Add the aliases, source (plug-in) and see also links, if any
347
 
        if self.aliases:
348
 
            result += ':Aliases:  '
349
 
            result += ', '.join(self.aliases) + '\n'
350
 
        plugin_name = self.plugin_name()
351
 
        if plugin_name is not None:
352
 
            result += ':From:     plugin "%s"\n' % plugin_name
353
 
        see_also = self.get_see_also(additional_see_also)
354
 
        if see_also:
355
 
            if not plain and see_also_as_links:
356
 
                see_also_links = []
357
 
                for item in see_also:
358
 
                    if item == 'topics':
359
 
                        # topics doesn't have an independent section
360
 
                        # so don't create a real link
361
 
                        see_also_links.append(item)
362
 
                    else:
363
 
                        # Use a reST link for this entry
364
 
                        see_also_links.append("`%s`_" % (item,))
365
 
                see_also = see_also_links
366
 
            result += ':See also: '
367
 
            result += ', '.join(see_also) + '\n'
368
 
 
369
 
        # If this will be rendered as plan text, convert it
370
 
        if plain:
371
 
            import bzrlib.help_topics
372
 
            result = bzrlib.help_topics.help_as_plain_text(result)
373
 
        return result
374
 
 
375
 
    @staticmethod
376
 
    def _get_help_parts(text):
377
 
        """Split help text into a summary and named sections.
378
 
 
379
 
        :return: (summary,sections) where summary is the top line and
380
 
            sections is a dictionary of the rest indexed by section name.
381
 
            A section starts with a heading line of the form ":xxx:".
382
 
            Indented text on following lines is the section value.
383
 
            All text found outside a named section is assigned to the
384
 
            default section which is given the key of None.
385
 
        """
386
 
        def save_section(sections, label, section):
387
 
            if len(section) > 0:
388
 
                if sections.has_key(label):
389
 
                    sections[label] += '\n' + section
390
 
                else:
391
 
                    sections[label] = section
392
 
            
393
 
        lines = text.rstrip().splitlines()
394
 
        summary = lines.pop(0)
395
 
        sections = {}
396
 
        label,section = None,''
397
 
        for line in lines:
398
 
            if line.startswith(':') and line.endswith(':') and len(line) > 2:
399
 
                save_section(sections, label, section)
400
 
                label,section = line[1:-1],''
401
 
            elif (label is not None) and len(line) > 1 and not line[0].isspace():
402
 
                save_section(sections, label, section)
403
 
                label,section = None,line
404
 
            else:
405
 
                if len(section) > 0:
406
 
                    section += '\n' + line
407
 
                else:
408
 
                    section = line
409
 
        save_section(sections, label, section)
410
 
        return summary, sections
411
 
 
412
 
    def get_help_topic(self):
413
 
        """Return the commands help topic - its name."""
414
 
        return self.name()
415
 
 
416
 
    def get_see_also(self, additional_terms=None):
417
 
        """Return a list of help topics that are related to this command.
418
 
        
419
 
        The list is derived from the content of the _see_also attribute. Any
420
 
        duplicates are removed and the result is in lexical order.
421
 
        :param additional_terms: Additional help topics to cross-reference.
422
 
        :return: A list of help topics.
423
 
        """
424
 
        see_also = set(getattr(self, '_see_also', []))
425
 
        if additional_terms:
426
 
            see_also.update(additional_terms)
427
 
        return sorted(see_also)
428
235
 
429
236
    def options(self):
430
237
        """Return dict of valid options for this command.
431
238
 
432
239
        Maps from long option name to option object."""
433
 
        r = Option.STD_OPTIONS.copy()
434
 
        std_names = r.keys()
 
240
        r = dict()
 
241
        r['help'] = option.Option.OPTIONS['help']
435
242
        for o in self.takes_options:
436
243
            if isinstance(o, basestring):
437
244
                o = option.Option.OPTIONS[o]
438
245
            r[o.name] = o
439
 
            if o.name in std_names:
440
 
                self.supported_std_options.append(o.name)
441
246
        return r
442
247
 
443
248
    def _setup_outf(self):
444
249
        """Return a file linked to stdout, which has proper encoding."""
 
250
        assert self.encoding_type in ['strict', 'exact', 'replace']
 
251
 
445
252
        # Originally I was using self.stdout, but that looks
446
253
        # *way* too much like sys.stdout
447
254
        if self.encoding_type == 'exact':
456
263
 
457
264
        output_encoding = osutils.get_terminal_encoding()
458
265
 
459
 
        self.outf = codecs.getwriter(output_encoding)(sys.stdout,
460
 
                        errors=self.encoding_type)
 
266
        # use 'replace' so that we don't abort if trying to write out
 
267
        # in e.g. the default C locale.
 
268
        self.outf = codecs.getwriter(output_encoding)(sys.stdout, errors=self.encoding_type)
461
269
        # For whatever reason codecs.getwriter() does not advertise its encoding
462
270
        # it just returns the encoding of the wrapped file, which is completely
463
271
        # bogus. So set the attribute, so we can find the correct encoding later.
464
272
        self.outf.encoding = output_encoding
465
273
 
 
274
    @deprecated_method(zero_eight)
 
275
    def run_argv(self, argv):
 
276
        """Parse command line and run.
 
277
        
 
278
        See run_argv_aliases for the 0.8 and beyond api.
 
279
        """
 
280
        return self.run_argv_aliases(argv)
 
281
 
466
282
    def run_argv_aliases(self, argv, alias_argv=None):
467
283
        """Parse the command line and run with extra aliases in alias_argv."""
468
284
        if argv is None:
470
286
                 DeprecationWarning, stacklevel=2)
471
287
            argv = []
472
288
        args, opts = parse_args(self, argv, alias_argv)
473
 
 
474
 
        # Process the standard options
475
289
        if 'help' in opts:  # e.g. bzr add --help
476
 
            sys.stdout.write(self.get_help_text())
 
290
            from bzrlib.help import help_on_command
 
291
            help_on_command(self.name())
477
292
            return 0
478
 
        trace.set_verbosity_level(option._verbosity_level)
479
 
        if 'verbose' in self.supported_std_options:
480
 
            opts['verbose'] = trace.is_verbose()
481
 
        elif opts.has_key('verbose'):
482
 
            del opts['verbose']
483
 
        if 'quiet' in self.supported_std_options:
484
 
            opts['quiet'] = trace.is_quiet()
485
 
        elif opts.has_key('quiet'):
486
 
            del opts['quiet']
487
 
 
488
293
        # mix arguments and options into one dictionary
489
294
        cmdargs = _match_argform(self.name(), self.takes_args, args)
490
295
        cmdopts = {}
533
338
            return None
534
339
 
535
340
 
 
341
# Technically, this function hasn't been use in a *really* long time
 
342
# but we are only deprecating it now.
 
343
@deprecated_function(zero_eleven)
 
344
def parse_spec(spec):
 
345
    """
 
346
    >>> parse_spec(None)
 
347
    [None, None]
 
348
    >>> parse_spec("./")
 
349
    ['./', None]
 
350
    >>> parse_spec("../@")
 
351
    ['..', -1]
 
352
    >>> parse_spec("../f/@35")
 
353
    ['../f', 35]
 
354
    >>> parse_spec('./@revid:john@arbash-meinel.com-20050711044610-3ca0327c6a222f67')
 
355
    ['.', 'revid:john@arbash-meinel.com-20050711044610-3ca0327c6a222f67']
 
356
    """
 
357
    if spec is None:
 
358
        return [None, None]
 
359
    if '/@' in spec:
 
360
        parsed = spec.split('/@')
 
361
        assert len(parsed) == 2
 
362
        if parsed[1] == "":
 
363
            parsed[1] = -1
 
364
        else:
 
365
            try:
 
366
                parsed[1] = int(parsed[1])
 
367
            except ValueError:
 
368
                pass # We can allow stuff like ./@revid:blahblahblah
 
369
            else:
 
370
                assert parsed[1] >=0
 
371
    else:
 
372
        parsed = [spec, None]
 
373
    return parsed
 
374
 
536
375
def parse_args(command, argv, alias_argv=None):
537
376
    """Parse command line.
538
377
    
597
436
 
598
437
    return argdict
599
438
 
600
 
def apply_coveraged(dirname, the_callable, *args, **kwargs):
601
 
    # Cannot use "import trace", as that would import bzrlib.trace instead of
602
 
    # the standard library's trace.
603
 
    trace = __import__('trace')
604
 
 
605
 
    tracer = trace.Trace(count=1, trace=0)
606
 
    sys.settrace(tracer.globaltrace)
607
 
 
608
 
    ret = the_callable(*args, **kwargs)
609
 
 
610
 
    sys.settrace(None)
611
 
    results = tracer.results()
612
 
    results.write_results(show_missing=1, summary=False,
613
 
                          coverdir=dirname)
614
439
 
615
440
 
616
441
def apply_profiled(the_callable, *args, **kwargs):
638
463
 
639
464
def apply_lsprofiled(filename, the_callable, *args, **kwargs):
640
465
    from bzrlib.lsprof import profile
 
466
    import cPickle
641
467
    ret, stats = profile(the_callable, *args, **kwargs)
642
468
    stats.sort()
643
469
    if filename is None:
644
470
        stats.pprint()
645
471
    else:
646
 
        stats.save(filename)
647
 
        trace.note('Profile data written to "%s".', filename)
 
472
        stats.freeze()
 
473
        cPickle.dump(stats, open(filename, 'w'), 2)
 
474
        print 'Profile data written to %r.' % filename
648
475
    return ret
649
476
 
650
477
 
651
 
def shlex_split_unicode(unsplit):
652
 
    import shlex
653
 
    return [u.decode('utf-8') for u in shlex.split(unsplit.encode('utf-8'))]
654
 
 
655
 
 
656
478
def get_alias(cmd, config=None):
657
479
    """Return an expanded alias, or None if no alias exists.
658
480
 
668
490
        config = bzrlib.config.GlobalConfig()
669
491
    alias = config.get_alias(cmd)
670
492
    if (alias):
671
 
        return shlex_split_unicode(alias)
 
493
        import shlex
 
494
        return [a.decode('utf-8') for a in shlex.split(alias.encode('utf-8'))]
672
495
    return None
673
496
 
674
497
 
703
526
 
704
527
    --lsprof
705
528
        Run under the Python lsprof profiler.
706
 
 
707
 
    --coverage
708
 
        Generate line coverage report in the specified directory.
709
529
    """
710
530
    argv = list(argv)
711
531
    trace.mutter("bzr arguments: %r", argv)
712
532
 
713
533
    opt_lsprof = opt_profile = opt_no_plugins = opt_builtin =  \
714
534
                opt_no_aliases = False
715
 
    opt_lsprof_file = opt_coverage_dir = None
 
535
    opt_lsprof_file = None
716
536
 
717
537
    # --no-plugins is handled specially at a very early stage. We need
718
538
    # to load plugins before doing other command parsing so that they
736
556
            opt_no_aliases = True
737
557
        elif a == '--builtin':
738
558
            opt_builtin = True
739
 
        elif a == '--coverage':
740
 
            opt_coverage_dir = argv[i + 1]
741
 
            i += 1
 
559
        elif a in ('--quiet', '-q'):
 
560
            trace.be_quiet()
742
561
        elif a.startswith('-D'):
743
562
            debug.debug_flags.add(a[2:])
744
563
        else:
752
571
        return 0
753
572
 
754
573
    if argv[0] == '--version':
755
 
        from bzrlib.builtins import cmd_version
756
 
        cmd_version().run_argv_aliases([])
 
574
        from bzrlib.version import show_version
 
575
        show_version()
757
576
        return 0
758
577
        
759
578
    if not opt_no_plugins:
777
596
    # 'command not found' error later.
778
597
 
779
598
    cmd_obj = get_cmd_object(cmd, plugins_override=not opt_builtin)
780
 
    run = cmd_obj.run_argv_aliases
781
 
    run_argv = [argv, alias_argv]
 
599
    if not getattr(cmd_obj.run_argv, 'is_deprecated', False):
 
600
        run = cmd_obj.run_argv
 
601
        run_argv = [argv]
 
602
    else:
 
603
        run = cmd_obj.run_argv_aliases
 
604
        run_argv = [argv, alias_argv]
782
605
 
783
606
    try:
784
607
        if opt_lsprof:
785
 
            if opt_coverage_dir:
786
 
                trace.warning(
787
 
                    '--coverage ignored, because --lsprof is in use.')
788
608
            ret = apply_lsprofiled(opt_lsprof_file, run, *run_argv)
789
609
        elif opt_profile:
790
 
            if opt_coverage_dir:
791
 
                trace.warning(
792
 
                    '--coverage ignored, because --profile is in use.')
793
610
            ret = apply_profiled(run, *run_argv)
794
 
        elif opt_coverage_dir:
795
 
            ret = apply_coveraged(opt_coverage_dir, run, *run_argv)
796
611
        else:
797
612
            ret = run(*run_argv)
798
 
        if 'memory' in debug.debug_flags:
799
 
            try:
800
 
                status_file = file('/proc/%s/status' % os.getpid(), 'rb')
801
 
            except IOError:
802
 
                pass
803
 
            else:
804
 
                status = status_file.read()
805
 
                status_file.close()
806
 
                trace.note("Process status after command:")
807
 
                for line in status.splitlines():
808
 
                    trace.note(line)
809
613
        return ret or 0
810
614
    finally:
811
615
        # reset, in case we may do other commands later within the same process
812
 
        option._verbosity_level = 0
 
616
        trace.be_quiet(False)
813
617
 
814
618
def display_command(func):
815
619
    """Decorator that suppresses pipe/interrupt errors."""
823
627
                raise
824
628
            if e.errno != errno.EPIPE:
825
629
                # Win32 raises IOError with errno=0 on a broken pipe
826
 
                if sys.platform != 'win32' or (e.errno not in (0, errno.EINVAL)):
 
630
                if sys.platform != 'win32' or e.errno != 0:
827
631
                    raise
828
632
            pass
829
633
        except KeyboardInterrupt:
835
639
    import bzrlib.ui
836
640
    from bzrlib.ui.text import TextUIFactory
837
641
    bzrlib.ui.ui_factory = TextUIFactory()
838
 
     
839
 
    # Is this a final release version? If so, we should suppress warnings
840
 
    if bzrlib.version_info[3] == 'final':
841
 
        from bzrlib import symbol_versioning
842
 
        symbol_versioning.suppress_deprecation_warnings(override=False)
843
 
    try:
844
 
        argv = [a.decode(bzrlib.user_encoding) for a in argv[1:]]
845
 
    except UnicodeDecodeError:
846
 
        raise errors.BzrError(("Parameter '%r' is unsupported by the current "
847
 
                                                            "encoding." % a))
 
642
    argv = [a.decode(bzrlib.user_encoding) for a in argv[1:]]
848
643
    ret = run_bzr_catch_errors(argv)
849
644
    trace.mutter("return code %d", ret)
850
645
    return ret
851
646
 
852
647
 
853
648
def run_bzr_catch_errors(argv):
854
 
    # Note: The except clause logic below should be kept in sync with the
855
 
    # profile() routine in lsprof.py.
856
649
    try:
857
650
        return run_bzr(argv)
 
651
        # do this here inside the exception wrappers to catch EPIPE
 
652
        sys.stdout.flush()
858
653
    except (KeyboardInterrupt, Exception), e:
859
654
        # used to handle AssertionError and KeyboardInterrupt
860
655
        # specially here, but hopefully they're handled ok by the logger now
861
 
        exitcode = trace.report_exception(sys.exc_info(), sys.stderr)
 
656
        trace.report_exception(sys.exc_info(), sys.stderr)
862
657
        if os.environ.get('BZR_PDB'):
863
658
            print '**** entering debugger'
864
659
            import pdb
865
660
            pdb.post_mortem(sys.exc_traceback)
866
 
        return exitcode
867
 
 
868
 
 
869
 
def run_bzr_catch_user_errors(argv):
870
 
    """Run bzr and report user errors, but let internal errors propagate.
871
 
 
872
 
    This is used for the test suite, and might be useful for other programs
873
 
    that want to wrap the commandline interface.
874
 
    """
875
 
    try:
876
 
        return run_bzr(argv)
877
 
    except Exception, e:
878
 
        if (isinstance(e, (OSError, IOError))
879
 
            or not getattr(e, 'internal_error', True)):
880
 
            trace.report_exception(sys.exc_info(), sys.stderr)
881
 
            return 3
882
 
        else:
883
 
            raise
884
 
 
885
 
 
886
 
class HelpCommandIndex(object):
887
 
    """A index for bzr help that returns commands."""
888
 
 
889
 
    def __init__(self):
890
 
        self.prefix = 'commands/'
891
 
 
892
 
    def get_topics(self, topic):
893
 
        """Search for topic amongst commands.
894
 
 
895
 
        :param topic: A topic to search for.
896
 
        :return: A list which is either empty or contains a single
897
 
            Command entry.
898
 
        """
899
 
        if topic and topic.startswith(self.prefix):
900
 
            topic = topic[len(self.prefix):]
901
 
        try:
902
 
            cmd = _get_cmd_object(topic)
903
 
        except KeyError:
904
 
            return []
905
 
        else:
906
 
            return [cmd]
907
 
 
908
 
 
909
 
class Provider(object):
910
 
    '''Generic class to be overriden by plugins'''
911
 
 
912
 
    def plugin_for_command(self, cmd_name):
913
 
        '''Takes a command and returns the information for that plugin
914
 
        
915
 
        :return: A dictionary with all the available information 
916
 
        for the requested plugin
917
 
        '''
918
 
        raise NotImplementedError
919
 
 
920
 
 
921
 
class ProvidersRegistry(registry.Registry):
922
 
    '''This registry exists to allow other providers to exist'''
923
 
 
924
 
    def __iter__(self):
925
 
        for key, provider in self.iteritems():
926
 
            yield provider
927
 
 
928
 
command_providers_registry = ProvidersRegistry()
929
 
 
 
661
        return 3
930
662
 
931
663
if __name__ == '__main__':
932
664
    sys.exit(main(sys.argv))