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             pass
  33             #import sys, traceback
  34             #traceback.print_exc(file=sys.stdout)
  35
  36     return all
  37
  38 def get_modules_scipt_implementations(script_name, impl_class):
  39     scripts = get_modules_script(script_name)
  40     all_impls = {}
  41
  42     for script in scripts:
  43         all_impls.update(dict([
  44             (n, i) for (n, i) in [(n, getattr(script, n)) for n in dir(script)]
  45             if isinstance(i, impl_class)
  46         ]))
  47
  48     return all_impls
  49
  50 def get_modules_script_classes(script_name, base_class):
  51     scripts = get_modules_script(script_name)
  52     all_classes = {}
  53
  54     for script in scripts:
  55         all_classes.update(dict([
  56             (n, c) for (n, c) in [(n, getattr(script, n)) for n in dir(script)]
  57             if isinstance(c, (type, types.ClassType)) and issubclass(c, base_class)
  58         ]))
  59
  60     return all_classes
  61
  62 def get_all_handlers(name):
  63      handler_files = get_modules_script('handlers')
  64
  65      return [
  66         h for h in [
  67             getattr(f, name) for f in handler_files
  68             if hasattr(f, name)
  69         ]
  70
  71         if callable(h)
  72      ]
  73
  74 def call_all_handlers(name, *args, **kwargs):
  75     all = get_all_handlers(name)
  76
  77     ret = []
  78
  79     for handler in all:
  80         ret.append(handler(*args, **kwargs))
  81
  82     return ret
  83
  84 def get_handler(name, default):
  85     all = get_all_handlers(name)
  86     return len(all) and all[0] or default
  87
  88 module_template_re = re.compile('^modules\/(\w+)\/(.*)$')
  89
  90 def module_templates_loader(name, dirs=None):
  91     result = module_template_re.search(name)
  92
  93     if result is not None:
  94         file_name = os.path.join(MODULES_FOLDER, result.group(1), 'templates', result.group(2))
  95
  96         if os.path.exists(file_name):
  97             try:
  98                 f = open(file_name, 'r')
  99                 source = f.read()
 100                 f.close()
 101                 return (source, file_name)
 102             except:
 103                 pass
 104
 105     raise TemplateDoesNotExist, name
 106
 107 module_templates_loader.is_usable = True