forum/modules/__init__.py

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