~doctormo/gnome-wallchanger/trunk

« back to all changes in this revision

Viewing changes to gwc/loader.py

  • Committer: Martin Owens (DoctorMO)
  • Date: 2009-10-11 19:32:33 UTC
  • Revision ID: doctormo@gmail.com-20091011193233-15p70kim7zsjyhqx
Add new modules for ver 3

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#
 
2
# Copyright 2009, Martin Owens.
 
3
#
 
4
# This program is free software: you can redistribute it and/or modify
 
5
# it under the terms of the GNU General Public License as published by
 
6
# the Free Software Foundation, either version 3 of the License, or
 
7
# (at your option) any later version.
 
8
#
 
9
# This program is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU General Public License
 
15
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
16
#
 
17
"""
 
18
Loader for plugins in python.
 
19
"""
 
20
 
 
21
import os
 
22
import sys
 
23
import logging
 
24
 
 
25
def load_modules( module=None, deep=False ):
 
26
    """Load any modules available to us"""
 
27
    result_modules = {}
 
28
    path     = './'
 
29
    parent   = None
 
30
    fromlist = None
 
31
    if module:
 
32
        path = module.__path__[0]
 
33
        parent = module.__name__
 
34
        fromlist = parent.split('.')
 
35
 
 
36
    for filename in os.listdir(path):
 
37
        if os.path.isfile(os.path.join(path, filename)):
 
38
            if filename[-3:] == '.py': #and '__' not in filename:
 
39
                name = filename[:-3]
 
40
                if parent:
 
41
                    name = "%s.%s" % (parent, filename[:-3])
 
42
                logging.debug("Importing Module: %s" % name)
 
43
                if sys.modules.has_key(name):
 
44
                     result_modules[name] = sys.modules[name]
 
45
                else:
 
46
                    result_modules[name] = __import__(name,
 
47
                        fromlist=fromlist, level=0)
 
48
        elif os.path.isdir(os.path.join(path, filename)) and deep:
 
49
            if not os.path.islink(os.path.join(path, filename)):
 
50
                name = filename
 
51
                if parent:
 
52
                    name = "%s.%s" % (parent, filename)
 
53
                try:
 
54
                    module_name = __import__(name, fromlist=fromlist)
 
55
                    logging.debug("Found module directory: %s" % filename)
 
56
                    sub_modules = load_modules(module_name)
 
57
                    result_modules.update(sub_modules)
 
58
                except ImportError, msg:
 
59
                    if 'No module' not in msg:
 
60
                        raise
 
61
    return result_modules
 
62
 
 
63