forum/modules/__init__.py

   1 import os
   2 import types
   3 import re
   4 import logging
   5
   6 from django.template import Template, TemplateDoesNotExist
   7 from django.conf import settings
   8
   9 MODULES_PACKAGE = 'forum_modules'
  10
  11 MODULES_FOLDER = os.path.join(os.path.dirname(__file__), '../../' + MODULES_PACKAGE)
  12
  13 DISABLED_MODULES = getattr(settings, 'DISABLED_MODULES', [])
  14
  15 MODULE_LIST = filter(lambda m: getattr(m, 'CAN_USE', True), [
  16         __import__('forum_modules.%s' % f, globals(), locals(), ['forum_modules'])
  17         for f in os.listdir(MODULES_FOLDER)
  18         if os.path.isdir(os.path.join(MODULES_FOLDER, f)) and
  19            os.path.exists(os.path.join(MODULES_FOLDER, "%s/__init__.py" % f)) and
  20            not f in DISABLED_MODULES
  21 ])
  22
  23 def get_modules_script(script_name):
  24     all = []
  25
  26     for m in MODULE_LIST:
  27         try:
  28             all.append(__import__('%s.%s' % (m.__name__, script_name), globals(), locals(), [m.__name__]))
  29         except ImportError, e:
  30             #print repr(type(e)) + m.__name__ + ":" + str(e)
  31             pass
  32         except:
  33             import traceback
  34             msg = "Error importing %s from module %s: \n %s" % (
  35                 script_name, m.__name__, traceback.format_exc()
  36             )
  37             logging.error(msg)
  38
  39     return all
  40
  41 def get_modules_script_implementations(script_name, impl_class):
  42     scripts = get_modules_script(script_name)
  43     all_impls = {}
  44
  45     for script in scripts:
  46         all_impls.update(dict([
  47             (n, i) for (n, i) in [(n, getattr(script, n)) for n in dir(script)]
  48             if isinstance(i, impl_class)
  49         ]))
  50
  51     return all_impls
  52
  53 def get_modules_script_classes(script_name, base_class):
  54     scripts = get_modules_script(script_name)
  55     all_classes = {}
  56
  57     for script in scripts:
  58         all_classes.update(dict([
  59             (n, c) for (n, c) in [(n, getattr(script, n)) for n in dir(script)]
  60             if isinstance(c, (type, types.ClassType)) and issubclass(c, base_class)
  61         ]))
  62
  63     return all_classes
  64
  65 def get_all_handlers(name):
  66      handler_files = get_modules_script('handlers')
  67
  68      return [
  69         h for h in [
  70             getattr(f, name) for f in handler_files
  71             if hasattr(f, name)
  72         ]
  73
  74         if callable(h)
  75      ]
  76
  77 def call_all_handlers(name, *args, **kwargs):
  78     all = get_all_handlers(name)
  79
  80     ret = []
  81
  82     for handler in all:
  83         ret.append(handler(*args, **kwargs))
  84
  85     return ret
  86
  87 def get_handler(name, default):
  88     all = get_all_handlers(name)
  89     return len(all) and all[0] or default
  90
  91 module_template_re = re.compile('^modules\/(\w+)\/(.*)$')
  92
  93 def module_templates_loader(name, dirs=None):
  94     result = module_template_re.search(name)
  95
  96     if result is not None:
  97         file_name = os.path.join(MODULES_FOLDER, result.group(1), 'templates', result.group(2))
  98
  99         if os.path.exists(file_name):
 100             try:
 101                 f = open(file_name, 'r')
 102                 source = f.read()
 103                 f.close()
 104                 return (source, file_name)
 105             except:
 106                 pass
 107
 108     raise TemplateDoesNotExist, name
 109
 110 module_templates_loader.is_usable = True
 111
 112 from decorators import decorate, ReturnImediatelyException