/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Martin Pool
  • Date: 2005-06-28 03:02:31 UTC
  • Revision ID: mbp@sourcefrog.net-20050628030231-d311e4ebcd467ef4
Merge John's import-speedup branch:

                                                                                         
  777 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:20:32 -0500
      revision-id: john@arbash-meinel.com-20050627032031-e82a50db3863b18e
      bzr selftest was not using the correct bzr

  776 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:20:22 -0500
      revision-id: john@arbash-meinel.com-20050627032021-c9f21fde989ddaee
      Add was using an old mutter

  775 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:02:33 -0500
      revision-id: john@arbash-meinel.com-20050627030233-9165cfe98fc63298
      Cleaned up to be less different

  774 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:54:53 -0500
      revision-id: john@arbash-meinel.com-20050627025452-4260d0e744edef43
      Allow BZR_PLUGIN_PATH='' to negate plugin loading.

  773 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:49:34 -0500
      revision-id: john@arbash-meinel.com-20050627024933-b7158f67b7b9eae5
      Finished the previous cleanup (allowing load_plugins to be called twice)

  772 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:45:08 -0500
      revision-id: john@arbash-meinel.com-20050627024508-723b1df510d196fc
      Work on making the tests pass. versioning.py is calling run_cmd directly, but plugins have been loaded.

  771 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:32:29 -0500
      revision-id: john@arbash-meinel.com-20050627023228-79972744d7c53e15
      Got it down a little bit more by removing import of tree and inventory.

  770 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:26:05 -0500
      revision-id: john@arbash-meinel.com-20050627022604-350b9773ef622f95
      Reducing the number of import from bzrlib/__init__.py and bzrlib/branch.py

  769 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:32:25 -0500
      revision-id: john@arbash-meinel.com-20050627013225-32dd044f10d23948
      Updated revision.py and xml.py to include SubElement.

  768 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:03:56 -0500
      revision-id: john@arbash-meinel.com-20050627010356-ee66919e1c377faf
      Minor typo

  767 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:03:13 -0500
      revision-id: john@arbash-meinel.com-20050627010312-40d024007eb85051
      Caching the import

  766 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:51:47 -0500
      revision-id: john@arbash-meinel.com-20050627005147-5281c99e48ed1834
      Created wrapper functions for lazy import of ElementTree

  765 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:46:37 -0500
      revision-id: john@arbash-meinel.com-20050627004636-bf432902004a94c5
      Removed all of the test imports of cElementTree

  764 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:43:59 -0500
      revision-id: john@arbash-meinel.com-20050627004358-d137fbe9570dd71b
      Trying to make bzr startup faster.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugin.py
 
1
# Copyright (C) 2004, 2005 by Canonical Ltd
 
2
 
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
 
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
 
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
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
 
 
26
import os
 
27
from bzrlib.osutils import config_dir
 
28
DEFAULT_PLUGIN_PATH = os.path.join(config_dir(), 'plugins')
 
29
 
 
30
all_plugins = []
 
31
_loaded = False
 
32
 
 
33
 
 
34
def load_plugins():
 
35
    """
 
36
    Find all python plugins and load them.
 
37
 
 
38
    Loading a plugin means importing it into the python interpreter.
 
39
    The plugin is expected to make calls to register commands when
 
40
    it's loaded (or perhaps access other hooks in future.)
 
41
 
 
42
    A list of plugs is stored in bzrlib.plugin.all_plugins for future
 
43
    reference.
 
44
 
 
45
    The environment variable BZR_PLUGIN_PATH is considered a delimited
 
46
    set of paths to look through. Each entry is searched for *.py
 
47
    files (and whatever other extensions are used in the platform,
 
48
    such as *.pyd).
 
49
    """
 
50
 
 
51
    global all_plugins, _loaded
 
52
    if _loaded:
 
53
        # People can make sure plugins are loaded, they just won't be twice
 
54
        return
 
55
        #raise BzrError("plugins already initialized")
 
56
    _loaded = True
 
57
 
 
58
    import sys, os, imp
 
59
    try:
 
60
        set
 
61
    except NameError:
 
62
        from sets import Set as set         # python2.3
 
63
 
 
64
    from bzrlib.trace import log_error, mutter, log_exception
 
65
    from bzrlib.errors import BzrError
 
66
 
 
67
    bzrpath = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH)
 
68
 
 
69
    # The problem with imp.get_suffixes() is that it doesn't include
 
70
    # .pyo which is technically valid
 
71
    # It also means that "testmodule.so" will show up as both test and testmodule
 
72
    # though it is only valid as 'test'
 
73
    # but you should be careful, because "testmodule.py" loads as testmodule.
 
74
    suffixes = imp.get_suffixes()
 
75
    suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
 
76
    package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
 
77
    for d in bzrpath.split(os.pathsep):
 
78
        # going through them one by one allows different plugins with the same
 
79
        # filename in different directories in the path
 
80
        mutter('looking for plugins in %s' % d)
 
81
        if not d:
 
82
            continue
 
83
        plugin_names = set()
 
84
        if not os.path.isdir(d):
 
85
            continue
 
86
        for f in os.listdir(d):
 
87
            path = os.path.join(d, f)
 
88
            if os.path.isdir(path):
 
89
                for entry in package_entries:
 
90
                    # This directory should be a package, and thus added to
 
91
                    # the list
 
92
                    if os.path.isfile(os.path.join(path, entry)):
 
93
                        break
 
94
                else: # This directory is not a package
 
95
                    continue
 
96
            else:
 
97
                for suffix_info in suffixes:
 
98
                    if f.endswith(suffix_info[0]):
 
99
                        f = f[:-len(suffix_info[0])]
 
100
                        if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):
 
101
                            f = f[:-len('module')]
 
102
                        break
 
103
                else:
 
104
                    continue
 
105
            mutter('add plugin name %s' % f)
 
106
            plugin_names.add(f)
 
107
 
 
108
        plugin_names = list(plugin_names)
 
109
        plugin_names.sort()
 
110
        for name in plugin_names:
 
111
            try:
 
112
                plugin_info = imp.find_module(name, [d])
 
113
                mutter('load plugin %r' % (plugin_info,))
 
114
                try:
 
115
                    plugin = imp.load_module('bzrlib.plugin.' + name,
 
116
                                             *plugin_info)
 
117
                    all_plugins.append(plugin_info)
 
118
                finally:
 
119
                    if plugin_info[0] is not None:
 
120
                        plugin_info[0].close()
 
121
            except Exception, e:
 
122
                log_error('Unable to load plugin %r from %r' % (name, d))
 
123
                log_error(str(e))
 
124
                log_exception()
 
125