Extensions/pluginloader.py

   1 # -*- coding: utf-8 -*-
   2 #            CONFIGURATION MANAGEMENT OF EDF VERSION
   3 # ======================================================================
   4 # COPYRIGHT (C) 1991 - 2002  EDF R&D                  WWW.CODE-ASTER.ORG
   5 # THIS PROGRAM IS FREE SOFTWARE; YOU CAN REDISTRIBUTE IT AND/OR MODIFY
   6 # IT UNDER THE TERMS OF THE GNU GENERAL PUBLIC LICENSE AS PUBLISHED BY
   7 # THE FREE SOFTWARE FOUNDATION; EITHER VERSION 2 OF THE LICENSE, OR
   8 # (AT YOUR OPTION) ANY LATER VERSION.
   9 #
  10 # THIS PROGRAM IS DISTRIBUTED IN THE HOPE THAT IT WILL BE USEFUL, BUT
  11 # WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED WARRANTY OF
  12 # MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. SEE THE GNU
  13 # GENERAL PUBLIC LICENSE FOR MORE DETAILS.
  14 #
  15 # YOU SHOULD HAVE RECEIVED A COPY OF THE GNU GENERAL PUBLIC LICENSE
  16 # ALONG WITH THIS PROGRAM; IF NOT, WRITE TO EDF R&D CODE_ASTER,
  17 #    1 AVENUE DU GENERAL DE GAULLE, 92141 CLAMART CEDEX, FRANCE.
  18 #
  19 #
  20 # ======================================================================
  21 """
  22     Ce module contient le chargeur dynamique de plugins (emprunté à HappyDoc)
  23 """
  24
  25 import glob,os,sys,traceback
  26 import UserDict
  27
  28 class PluginLoader(UserDict.UserDict):
  29    def __init__(self,module):
  30       UserDict.UserDict.__init__(self)
  31       self.plugin_dir=module.__path__[0]
  32       self.plugin_set_name=module.__name__
  33       _module_list = glob.glob( os.path.join( self.plugin_dir,
  34                               '%s*py' % self.plugin_set_name,
  35                                            )
  36                               )
  37       _module_list.sort()
  38
  39       for _module_name in _module_list:
  40
  41         _module_name = os.path.basename(_module_name)[:-3]
  42         _import_name = '%s.%s' % ( self.plugin_set_name,
  43                                    _module_name )
  44
  45         try:
  46           _module = __import__( _import_name )
  47         except:
  48           sys.stderr.write('\n--- Plugin Module Error ---\n')
  49           traceback.print_exc()
  50           sys.stderr.write('---------------------------\n\n')
  51           continue
  52         try:
  53           _module = getattr(_module, _module_name)
  54         except AttributeError:
  55           sys.stderr.write('ERROR: Could not retrieve %s\n' % _import_name)
  56
  57         try:
  58           info = _module.entryPoint()
  59         except AttributeError:
  60           pass
  61         else:
  62           self.addEntryPoint(info)
  63
  64
  65    def addEntryPoint(self,infoDict):
  66       name=infoDict['name']
  67       factory=infoDict['factory']
  68       self[name]=factory
  69