Salome HOME
cht version
[tools/eficas.git] / convert / convert_pyth.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2007-2021   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     Ce module contient le plugin convertisseur de fichier
22     au format python pour EFICAS.
23
24     Un plugin convertisseur doit fournir deux attributs de classe :
25     extensions et formats et deux methodes : readfile,convert.
26
27     L'attribut de classe extensions est une liste d'extensions
28     de fichiers preconisees pour ce type de format. Cette information
29     est seulement indicative.
30
31     L'attribut de classe formats est une liste de formats de sortie
32     supportes par le convertisseur. Les formats possibles sont :
33     eval, dict ou exec.
34     Le format eval est un texte source Python qui peut etre evalue. Le
35     resultat de l'evaluation est un objet Python quelconque.
36     Le format dict est un dictionnaire Python.
37     Le format exec est un texte source Python qui peut etre execute.
38
39     La methode readfile a pour fonction de lire un fichier dont le
40     nom est passe en argument de la fonction.
41        - convertisseur.readfile(nom_fichier)
42
43     La methode convert a pour fonction de convertir le fichier
44     prealablement lu dans un objet du format passe en argument.
45        - objet=convertisseur.convert(outformat)
46
47     Ce convertisseur supporte le format de sortie dict
48
49 """
50 from __future__ import absolute_import
51 try :
52     from builtins import str
53     from builtins import object
54 except :
55     pass
56
57 import sys,traceback
58
59 from Noyau import N_CR
60 from Extensions.i18n import tr
61 from Extensions.eficas_exception import EficasException
62
63 def entryPoint():
64     """
65         Retourne les informations necessaires pour le chargeur de plugins
66         Ces informations sont retournees dans un dictionnaire
67     """
68     return {
69          # Le nom du plugin
70            'name' : 'pyth',
71          # La factory pour creer une instance du plugin
72            'factory' : Pythparser,
73            }
74
75
76 class Pythparser(object):
77     """
78         Ce convertisseur lit un fichier au format pyth avec la
79         methode readfile : convertisseur.readfile(nom_fichier)
80         et retourne le texte au format outformat avec la
81         methode convertisseur.convert(outformat)
82
83         Ses caracteristiques principales sont exposees dans 2 attributs
84         de classe :
85           - extensions : qui donne une liste d'extensions de fichier preconisees
86           - formats : qui donne une liste de formats de sortie supportes
87     """
88     # Les extensions de fichier preconisees
89     extensions=('.pyth',)
90     # Les formats de sortie supportes (eval dict ou exec)
91     formats=('dict',)
92
93     def __init__(self,cr=None):
94         # Si l'objet compte-rendu n'est pas fourni, on utilise le compte-rendu standard
95         if cr :
96             self.cr=cr
97         else:
98             self.cr=N_CR.CR(debut='CR convertisseur format pyth',
99                             fin='fin CR format pyth')
100         self.g={}
101
102     def readfile(self,filename):
103         self.filename=filename
104         try:
105             with open(filename) as fd :
106                 self.text=fd.read()
107         except:
108             self.cr.fatal(tr("Impossible d'ouvrir le fichier : %s",str( filename)))
109             return
110         self.g={}
111         try:
112             exec(self.text, self.g)
113         except EficasException as e:
114             l=traceback.format_exception(sys.exc_info()[0],sys.exc_info()[1],sys.exc_info()[2])
115             s= ''.join(l[2:])
116             s= s.replace('"<string>"','"<%s>"'%self.filename)
117             self.cr.fatal(tr("Erreur a l'evaluation :\n %s", s))
118
119     def convert(self,outformat,appliEficas=None):
120         if outformat == 'dict':
121             return self.getdict()
122         else:
123             raise EficasException(tr("Format de sortie : %s, non supporte", outformat))
124
125     def getdict(self):
126         d={}
127         for k,v in list(self.g.items()):
128             if k[0] != '_':d[k]=v
129         return d