← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2007-12-20 16:16:34 UTC
  • mfrom: (3123.5.18 hardlinks)
  • Revision ID: pqm@pqm.ubuntu.com-20071220161634-2kcjb650o21ydko4
Accelerate build_tree using similar workingtrees (abentley)

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,
47
46
    trace,
48
47
    win32utils,
49
48
    )
95
94
 
96
95
 
97
96
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
 
 
182
171
    raise KeyError
183
172
 
184
173
 
282
271
            elif aname[-1] == '*':
283
272
                aname = '[' + aname[:-1] + '...]'
284
273
            s += aname + ' '
285
 
        s = s[:-1]      # remove last space
 
274
                
 
275
        assert s[-1] == ' '
 
276
        s = s[:-1]
286
277
        return s
287
278
 
288
279
    def get_help_text(self, additional_see_also=None, plain=True,
398
389
            if line.startswith(':') and line.endswith(':') and len(line) > 2:
399
390
                save_section(sections, label, section)
400
391
                label,section = line[1:-1],''
401
 
            elif (label is not None) and len(line) > 1 and not line[0].isspace():
 
392
            elif label != None and len(line) > 1 and not line[0].isspace():
402
393
                save_section(sections, label, section)
403
394
                label,section = None,line
404
395
            else:
442
433
 
443
434
    def _setup_outf(self):
444
435
        """Return a file linked to stdout, which has proper encoding."""
 
436
        assert self.encoding_type in ['strict', 'exact', 'replace']
 
437
 
445
438
        # Originally I was using self.stdout, but that looks
446
439
        # *way* too much like sys.stdout
447
440
        if self.encoding_type == 'exact':
597
590
 
598
591
    return argdict
599
592
 
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
593
 
615
594
 
616
595
def apply_profiled(the_callable, *args, **kwargs):
648
627
    return ret
649
628
 
650
629
 
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
630
def get_alias(cmd, config=None):
657
631
    """Return an expanded alias, or None if no alias exists.
658
632
 
668
642
        config = bzrlib.config.GlobalConfig()
669
643
    alias = config.get_alias(cmd)
670
644
    if (alias):
671
 
        return shlex_split_unicode(alias)
 
645
        import shlex
 
646
        return [a.decode('utf-8') for a in shlex.split(alias.encode('utf-8'))]
672
647
    return None
673
648
 
674
649
 
703
678
 
704
679
    --lsprof
705
680
        Run under the Python lsprof profiler.
706
 
 
707
 
    --coverage
708
 
        Generate line coverage report in the specified directory.
709
681
    """
710
682
    argv = list(argv)
711
683
    trace.mutter("bzr arguments: %r", argv)
712
684
 
713
685
    opt_lsprof = opt_profile = opt_no_plugins = opt_builtin =  \
714
686
                opt_no_aliases = False
715
 
    opt_lsprof_file = opt_coverage_dir = None
 
687
    opt_lsprof_file = None
716
688
 
717
689
    # --no-plugins is handled specially at a very early stage. We need
718
690
    # to load plugins before doing other command parsing so that they
736
708
            opt_no_aliases = True
737
709
        elif a == '--builtin':
738
710
            opt_builtin = True
739
 
        elif a == '--coverage':
740
 
            opt_coverage_dir = argv[i + 1]
741
 
            i += 1
742
711
        elif a.startswith('-D'):
743
712
            debug.debug_flags.add(a[2:])
744
713
        else:
782
751
 
783
752
    try:
784
753
        if opt_lsprof:
785
 
            if opt_coverage_dir:
786
 
                trace.warning(
787
 
                    '--coverage ignored, because --lsprof is in use.')
788
754
            ret = apply_lsprofiled(opt_lsprof_file, run, *run_argv)
789
755
        elif opt_profile:
790
 
            if opt_coverage_dir:
791
 
                trace.warning(
792
 
                    '--coverage ignored, because --profile is in use.')
793
756
            ret = apply_profiled(run, *run_argv)
794
 
        elif opt_coverage_dir:
795
 
            ret = apply_coveraged(opt_coverage_dir, run, *run_argv)
796
757
        else:
797
758
            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
759
        return ret or 0
810
760
    finally:
811
761
        # reset, in case we may do other commands later within the same process
835
785
    import bzrlib.ui
836
786
    from bzrlib.ui.text import TextUIFactory
837
787
    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
788
    try:
844
789
        argv = [a.decode(bzrlib.user_encoding) for a in argv[1:]]
845
790
    except UnicodeDecodeError:
906
851
            return [cmd]
907
852
 
908
853
 
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
 
 
931
854
if __name__ == '__main__':
932
855
    sys.exit(main(sys.argv))