Editeur/comploader.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2007-2017   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="QT"):
  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     listfich=glob.glob(os.path.join(repertoire, "compo*.py"))
  55     for fichier in listfich:
  56         m= os.path.basename(fichier)[:-3]
  57         module=__import__(package,globals(),locals(),[m])
  58         module = getattr(module, m)
  59         composants[module.objet]=module.treeitem
  60     return composants
  61
  62 def gettreeitem(object):
  63     """
  64       Cette fonction retourne la classe item associee a l'objet object.
  65       Cette classe item depend bien sur de la nature de object, d'ou
  66       l'interrogation du dictionnaire composants
  67     """
  68     # Si la definition de l'objet a un attribut itemeditor, il indique
  69     # la classe a utiliser pour l'item
  70     try:
  71        return object.definition.itemeditor
  72     except:
  73        pass
  74
  75     # On cherche ensuite dans les composants (plugins)
  76     try:
  77        itemtype= composants[object.__class__]
  78        return itemtype
  79     except:
  80        pass
  81
  82     # Puis une eventuelle classe heritee (aleatoire car sans ordre)
  83     for e in list(composants.keys()):
  84         if e and isinstance(object,e):
  85            itemtype= composants[e]
  86            return itemtype
  87
  88     # Si on n'a rien trouve dans les composants on utilise l'objet par defaut
  89     itemtype=composants[None]
  90     return itemtype
  91
  92 def make_objecttreeitem(appli,labeltext, object, setfunction=None):
  93     """
  94        Cette fonction permet de construire et de retourner un objet
  95        de type item associe a l'object passe en argument.
  96     """
  97     c = gettreeitem(object)
  98     #print appli,labeltext, object, setfunction
  99     return c(appli,labeltext, object, setfunction)
 100