Editeur/comploader.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2007-2013   EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License.
   8 #
   9 # This library 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 GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 """
  21    Module de chargement des composants et de mapping des objets du noyau
  22    Accas vers les items d'EFICAS
  23
  24      - composants : dictionnaire de stockage des relations entre types
  25        d'objet du noyau et types d'item
  26      - charger_composants() : fonction de chargement des composants. Retourne
  27        le dictionnaire composants.
  28      - gettreeitem(object) -> type d'item : fonction qui retourne un type
  29        d'item correspondant au type de l'objet noyau fourni.
  30      - make_objecttreeitem(appli,labeltext, object, setfunction=None) -> item : fonction qui retourne un item
  31        correspondant a l'objet noyau fourni.
  32 """
  33 # import generaux
  34 from __future__ import absolute_import
  35 import os,glob,types
  36
  37 # Dictionnaire {object : item} permettant d'associer un item a un object
  38 # Ce dictionnaire est renseigne par la methode charger_composants
  39 composants = {}
  40
  41 def charger_composants(Ihm="TK"):
  42     """
  43         Cette fonction a pour but de charger tous les modules composants graphiques
  44         (fichiers compo*.py dans le meme repertoire que ce module )
  45         et de remplir le dictionnaire composants utilise par make_objecttreeitem
  46     """
  47     reper=os.path.dirname(__file__)
  48     if Ihm == "TK" :
  49        repertoire=reper+"/../InterfaceTK"
  50        package="InterfaceTK"
  51     else :
  52        repertoire=reper+"/../InterfaceQT4"
  53        package="InterfaceQT4"
  54        #repertoire=reper+"/../InterfaceQT"
  55        #package="InterfaceQT"
  56     listfich=glob.glob(os.path.join(repertoire, "compo*.py"))
  57     for fichier in listfich:
  58         m= os.path.basename(fichier)[:-3]
  59         module=__import__(package,globals(),locals(),[m])
  60         module = getattr(module, m)
  61         composants[module.objet]=module.treeitem
  62     return composants
  63
  64 def gettreeitem(object):
  65     """
  66       Cette fonction retourne la classe item associee a l'objet object.
  67       Cette classe item depend bien sur de la nature de object, d'ou
  68       l'interrogation du dictionnaire composants
  69     """
  70     # Si la definition de l'objet a un attribut itemeditor, il indique
  71     # la classe a utiliser pour l'item
  72     try:
  73        return object.definition.itemeditor
  74     except:
  75        pass
  76
  77     # On cherche ensuite dans les composants (plugins)
  78     try:
  79        itemtype= composants[object.__class__]
  80        return itemtype
  81     except:
  82        pass
  83
  84     # Puis une eventuelle classe heritee (aleatoire car sans ordre)
  85     for e in list(composants.keys()):
  86         if e and isinstance(object,e):
  87            itemtype= composants[e]
  88            return itemtype
  89
  90     # Si on n'a rien trouve dans les composants on utilise l'objet par defaut
  91     itemtype=composants[None]
  92     return itemtype
  93
  94 def make_objecttreeitem(appli,labeltext, object, setfunction=None):
  95     """
  96        Cette fonction permet de construire et de retourner un objet
  97        de type item associe a l'object passe en argument.
  98     """
  99     c = gettreeitem(object)
 100     #print appli,labeltext, object, setfunction
 101     return c(appli,labeltext, object, setfunction)
 102