← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Robert Collins
  • Date: 2005-08-23 06:52:09 UTC
  • mto: (974.1.50) (1185.1.10) (1092.3.1)
  • mto: This revision was merged to the branch mainline in revision 1139.
  • Revision ID: robertc@robertcollins.net-20050823065209-81cd5962c401751b
move io redirection into each test case from the global runner

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugin.py
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
18
 
"""bzr python plugin support
19
 
 
20
 
Any python module in $BZR_PLUGIN_PATH will be imported upon initialization of
21
 
bzrlib. The module will be imported as 'bzrlib.plugins.$BASENAME(PLUGIN)'.
22
 
In the plugin's main body, it should update any bzrlib registries it wants to
23
 
extend; for example, to add new commands, import bzrlib.commands and add your
24
 
new command to the plugin_cmds variable.
25
 
"""
26
 
 
27
 
# TODO: Refactor this to make it more testable.  The main problem at the
28
 
# moment is that loading plugins affects the global process state -- for bzr
29
 
# in general use it's a reasonable assumption that all plugins are loaded at
30
 
# startup and then stay loaded, but this is less good for testing.
31
 
# 
32
 
# Several specific issues:
33
 
#  - plugins can't be unloaded and will continue to effect later tests
34
 
#  - load_plugins does nothing if called a second time
35
 
#  - plugin hooks can't be removed
36
 
#
37
 
# Our options are either to remove these restrictions, or work around them by
38
 
# loading the plugins into a different space than the one running the tests.
39
 
# That could be either a separate Python interpreter or perhaps a new
40
 
# namespace inside this interpreter.
41
 
 
42
 
import imp
 
18
# This module implements plug-in support.
 
19
# Any python module in $BZR_PLUGIN_PATH will be imported upon initialization
 
20
# of bzrlib (and then forgotten about).  In the plugin's main body, it should
 
21
# update any bzrlib registries it wants to extend; for example, to add new
 
22
# commands, import bzrlib.commands and add your new command to the
 
23
# plugin_cmds variable.
 
24
 
 
25
 
43
26
import os
44
 
import sys
45
 
 
46
 
import bzrlib
47
 
from bzrlib.config import config_dir
48
 
from bzrlib.trace import log_error, mutter, log_exception, warning, \
49
 
        log_exception_quietly
50
 
from bzrlib.errors import BzrError
51
 
from bzrlib import plugins
52
 
from bzrlib.osutils import pathjoin
53
 
 
54
 
DEFAULT_PLUGIN_PATH = pathjoin(config_dir(), 'plugins')
 
27
from bzrlib.osutils import config_dir
 
28
DEFAULT_PLUGIN_PATH = os.path.join(config_dir(), 'plugins')
55
29
 
56
30
all_plugins = []
57
31
_loaded = False
58
32
 
59
33
 
60
34
def load_plugins():
61
 
    """Find all python plugins and load them.
 
35
    """
 
36
    Find all python plugins and load them.
62
37
 
63
38
    Loading a plugin means importing it into the python interpreter.
64
39
    The plugin is expected to make calls to register commands when
80
55
        #raise BzrError("plugins already initialized")
81
56
    _loaded = True
82
57
 
83
 
    dirs = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH).split(os.pathsep)
 
58
    import sys, os, imp
 
59
    
 
60
    from bzrlib.trace import log_error, mutter, log_exception
 
61
    from bzrlib.errors import BzrError
 
62
    from bzrlib import plugins
 
63
 
 
64
    dirs = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH).split(":")
84
65
    dirs.insert(0, os.path.dirname(plugins.__file__))
85
66
 
86
67
    # The problem with imp.get_suffixes() is that it doesn't include
94
75
    for d in dirs:
95
76
        # going through them one by one allows different plugins with the same
96
77
        # filename in different directories in the path
97
 
        mutter('looking for plugins in %s', d)
 
78
        mutter('looking for plugins in %s' % d)
98
79
        if not d:
99
80
            continue
100
81
        plugin_names = set()
101
82
        if not os.path.isdir(d):
102
83
            continue
103
84
        for f in os.listdir(d):
104
 
            path = pathjoin(d, f)
 
85
            path = os.path.join(d, f)
105
86
            if os.path.isdir(path):
106
87
                for entry in package_entries:
107
88
                    # This directory should be a package, and thus added to
108
89
                    # the list
109
 
                    if os.path.isfile(pathjoin(path, entry)):
 
90
                    if os.path.isfile(os.path.join(path, entry)):
110
91
                        break
111
92
                else: # This directory is not a package
112
93
                    continue
119
100
                        break
120
101
                else:
121
102
                    continue
122
 
            mutter('add plugin name %s', f)
 
103
            mutter('add plugin name %s' % f)
123
104
            plugin_names.add(f)
124
105
 
125
106
        plugin_names = list(plugin_names)
127
108
        for name in plugin_names:
128
109
            try:
129
110
                plugin_info = imp.find_module(name, [d])
130
 
                mutter('load plugin %r', plugin_info)
 
111
                mutter('load plugin %r' % (plugin_info,))
131
112
                try:
132
 
                    plugin = imp.load_module('bzrlib.plugins.' + name,
 
113
                    plugin = imp.load_module('bzrlib.plugin.' + name,
133
114
                                             *plugin_info)
134
115
                    all_plugins.append(plugin)
135
 
                    setattr(bzrlib.plugins, name, plugin)
136
116
                finally:
137
117
                    if plugin_info[0] is not None:
138
118
                        plugin_info[0].close()
139
 
 
140
 
                mutter('loaded succesfully')
141
 
            except KeyboardInterrupt:
142
 
                raise
143
119
            except Exception, e:
144
 
                ## import pdb; pdb.set_trace()
145
 
                warning('Unable to load plugin %r from %r' % (name, d))
146
 
                log_exception_quietly()
 
120
                log_error('Unable to load plugin %r from %r' % (name, d))
 
121
                log_error(str(e))
 
122
                log_exception()
 
123