/brz/remove-bazaar

# Copyright (C) 2004, 2005 by Canonical Ltd

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

# This module implements plug-in support.

# Any python module in $BZR_PLUGIN_PATH will be imported upon initialization

# of bzrlib (and then forgotten about).  In the plugin's main body, it should

# update any bzrlib registries it wants to extend; for example, to add new

# commands, import bzrlib.commands and add your new command to the

# plugin_cmds variable.

import sys, os, imp

try:

    set

except NameError:

    from sets import Set as set

from bzrlib.trace import log_error

def load_plugins():

    """Find all python files which are plugins, and load them

    The environment variable BZR_PLUGIN_PATH is considered a delimited set of

    paths to look through. Each entry is searched for *.py files (and whatever

    other extensions are used in the platform, such as *.pyd).

    """

    bzrpath = os.environ.get('BZR_PLUGIN_PATH', os.path.expanduser('~/.bzr/plugins'))

    # The problem with imp.get_suffixes() is that it doesn't include

    # .pyo which is technically valid

    # It also means that "testmodule.so" will show up as both test and testmodule

    # though it is only valid as 'test'

    # but you should be careful, because "testmodule.py" loads as testmodule.

    suffixes = imp.get_suffixes()

    suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))

    package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']

    for d in bzrpath.split(os.pathsep):

        # going trough them one by one allows different plugins with the same

        # filename in different directories in the path

        if not d:

            continue

        plugin_names = set()

        if not os.path.isdir(d):

            continue

        for f in os.listdir(d):

            path = os.path.join(d, f)

            if os.path.isdir(path):

                for entry in package_entries:

                    # This directory should be a package, and thus added to

                    # the list

                    if os.path.isfile(os.path.join(path, entry)):

                        break

                else: # This directory is not a package

                    continue

            else:

                for suffix_info in suffixes:

                    if f.endswith(suffix_info[0]):

                        f = f[:-len(suffix_info[0])]

                        if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):

                            f = f[:-len('module')]

                        break

                else:

                    continue

            plugin_names.add(f)

        plugin_names = list(plugin_names)

        plugin_names.sort()

        for name in plugin_names:

            try:

                plugin_info = imp.find_module(name, [d])

                try:

                    plugin = imp.load_module('bzrlib.plugin.' + name,

                                             *plugin_info)

                finally:

                    if plugin_info[0] is not None:

                        plugin_info[0].close()

            except Exception, e:

                log_error('Unable to load plugin: %r from %r\n%s' % (name, d, e))