← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: John Arbash Meinel
  • Author(s): Mark Hammond
  • Date: 2008-09-09 17:02:21 UTC
  • mto: This revision was merged to the branch mainline in revision 3697.
  • Revision ID: john@arbash-meinel.com-20080909170221-svim3jw2mrz0amp3
An updated transparent icon for bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/commands.py
43
43
    errors,
44
44
    option,
45
45
    osutils,
 
46
    registry,
46
47
    trace,
47
48
    win32utils,
48
49
    )
94
95
 
95
96
 
96
97
def _unsquish_command_name(cmd):
97
 
    assert cmd.startswith("cmd_")
98
98
    return cmd[4:].replace('_','-')
99
99
 
100
100
 
168
168
    cmd_obj = ExternalCommand.find_command(cmd_name)
169
169
    if cmd_obj:
170
170
        return cmd_obj
 
171
 
 
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
 
171
182
    raise KeyError
172
183
 
173
184
 
271
282
            elif aname[-1] == '*':
272
283
                aname = '[' + aname[:-1] + '...]'
273
284
            s += aname + ' '
274
 
                
275
 
        assert s[-1] == ' '
276
 
        s = s[:-1]
 
285
        s = s[:-1]      # remove last space
277
286
        return s
278
287
 
279
288
    def get_help_text(self, additional_see_also=None, plain=True,
357
366
            result += ':See also: '
358
367
            result += ', '.join(see_also) + '\n'
359
368
 
360
 
        # If this will be rendered as plan text, convert it
 
369
        # If this will be rendered as plain text, convert it
361
370
        if plain:
362
371
            import bzrlib.help_topics
363
372
            result = bzrlib.help_topics.help_as_plain_text(result)
380
389
                    sections[label] += '\n' + section
381
390
                else:
382
391
                    sections[label] = section
383
 
            
 
392
 
384
393
        lines = text.rstrip().splitlines()
385
394
        summary = lines.pop(0)
386
395
        sections = {}
389
398
            if line.startswith(':') and line.endswith(':') and len(line) > 2:
390
399
                save_section(sections, label, section)
391
400
                label,section = line[1:-1],''
392
 
            elif label != None and len(line) > 1 and not line[0].isspace():
 
401
            elif (label is not None) and len(line) > 1 and not line[0].isspace():
393
402
                save_section(sections, label, section)
394
403
                label,section = None,line
395
404
            else:
433
442
 
434
443
    def _setup_outf(self):
435
444
        """Return a file linked to stdout, which has proper encoding."""
436
 
        assert self.encoding_type in ['strict', 'exact', 'replace']
437
 
 
438
445
        # Originally I was using self.stdout, but that looks
439
446
        # *way* too much like sys.stdout
440
447
        if self.encoding_type == 'exact':
490
497
        self._setup_outf()
491
498
 
492
499
        return self.run(**all_cmd_args)
493
 
    
 
500
 
494
501
    def run(self):
495
502
        """Actually run the command.
496
503
 
590
597
 
591
598
    return argdict
592
599
 
 
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)
593
614
 
594
615
 
595
616
def apply_profiled(the_callable, *args, **kwargs):
627
648
    return ret
628
649
 
629
650
 
 
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
 
630
656
def get_alias(cmd, config=None):
631
657
    """Return an expanded alias, or None if no alias exists.
632
658
 
642
668
        config = bzrlib.config.GlobalConfig()
643
669
    alias = config.get_alias(cmd)
644
670
    if (alias):
645
 
        import shlex
646
 
        return [a.decode('utf-8') for a in shlex.split(alias.encode('utf-8'))]
 
671
        return shlex_split_unicode(alias)
647
672
    return None
648
673
 
649
674
 
678
703
 
679
704
    --lsprof
680
705
        Run under the Python lsprof profiler.
 
706
 
 
707
    --coverage
 
708
        Generate line coverage report in the specified directory.
681
709
    """
682
710
    argv = list(argv)
683
711
    trace.mutter("bzr arguments: %r", argv)
684
712
 
685
713
    opt_lsprof = opt_profile = opt_no_plugins = opt_builtin =  \
686
714
                opt_no_aliases = False
687
 
    opt_lsprof_file = None
 
715
    opt_lsprof_file = opt_coverage_dir = None
688
716
 
689
717
    # --no-plugins is handled specially at a very early stage. We need
690
718
    # to load plugins before doing other command parsing so that they
708
736
            opt_no_aliases = True
709
737
        elif a == '--builtin':
710
738
            opt_builtin = True
 
739
        elif a == '--coverage':
 
740
            opt_coverage_dir = argv[i + 1]
 
741
            i += 1
711
742
        elif a.startswith('-D'):
712
743
            debug.debug_flags.add(a[2:])
713
744
        else:
751
782
 
752
783
    try:
753
784
        if opt_lsprof:
 
785
            if opt_coverage_dir:
 
786
                trace.warning(
 
787
                    '--coverage ignored, because --lsprof is in use.')
754
788
            ret = apply_lsprofiled(opt_lsprof_file, run, *run_argv)
755
789
        elif opt_profile:
 
790
            if opt_coverage_dir:
 
791
                trace.warning(
 
792
                    '--coverage ignored, because --profile is in use.')
756
793
            ret = apply_profiled(run, *run_argv)
 
794
        elif opt_coverage_dir:
 
795
            ret = apply_coveraged(opt_coverage_dir, run, *run_argv)
757
796
        else:
758
797
            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)
759
809
        return ret or 0
760
810
    finally:
761
811
        # reset, in case we may do other commands later within the same process
785
835
    import bzrlib.ui
786
836
    from bzrlib.ui.text import TextUIFactory
787
837
    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)
788
843
    try:
789
844
        argv = [a.decode(bzrlib.user_encoding) for a in argv[1:]]
790
845
    except UnicodeDecodeError:
851
906
            return [cmd]
852
907
 
853
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
 
 
930
 
854
931
if __name__ == '__main__':
855
932
    sys.exit(main(sys.argv))