Salome HOME
except and print
[tools/eficas.git] / Editeur / utils.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     Ce module contient des utilitaires divers
22 """
23 import os,re
24 import glob
25 import traceback
26 import codecs,types
27
28 from Extensions.i18n import tr
29
30 def substract_list(liste1,liste2):
31   """ 
32       Enleve tous les elements de liste2 presents dans liste1 et retourne liste1
33   """
34   for item in liste2:
35     try:
36       liste1.remove(item)
37     except:
38       pass
39   return liste1
40
41 def get_rep_user(dir):
42   """
43       Determine sur quelle plate-forme s'execute Eficas et recherche
44       le repertoire de l'utilisateur /$home/Eficas_install
45   """
46
47   rep_user_eficas= os.path.join(os.environ['HOME'],dir)
48   if os.path.exists(rep_user_eficas):
49     if os.path.isfile(rep_user_eficas) :
50       print (tr("Un fichier de nom %s existe deja : impossible de creer un repertoire de meme nom", rep_user_eficas))
51       rep_user_eficas=None
52   else :
53     try:
54       os.mkdir(rep_user_eficas)
55     except:
56       print (tr("Creation du repertoire %s impossible\n Verifiez vos droits d'acces", rep_user_eficas))
57   return rep_user_eficas
58
59 def read_file(file):
60   """
61       ouvre le fichier file et retourne son contenu
62       si pbe retourne None
63   """
64   try :
65     f=open(file)
66     text=f.read()
67     f.close()
68     return text
69   except:
70     return None
71
72 def save_in_file(file,text,dir=None):
73   """
74       cree le fichier file (ou l'ecrase s'il existe) et ecrit text dedans
75       retourne 1 si OK 0 sinon
76   """
77   try :
78       import string
79       #file=string.split(file,"/")[-1]
80       if dir != None:
81          os.chdir(dir)
82       f=open(file,'w')
83       f.write(text)
84       f.close()
85       return 1
86   except:
87       return 0
88
89 def extension_fichier(pathAndFile):
90     """ Return ext if path/filename.ext is given """
91     return os.path.splitext(pathAndFile)[1][1:]
92
93 def stripPath(pathAndFile):
94     """ Return filename.ext if path/filename.ext is given """
95     return os.path.split(pathAndFile)[1]
96
97 def init_rep_cata_dev(fic_cata,rep_goal):
98   """ 
99       Initialise le repertoire des catalogues developpeurs (chemin d'acces donne
100       dans le fichier eficas.ini cad :
101         - le cree s'il n'existe pas encore
102         - copie dedans les 3 fichiers necessaires :
103           * __init__.py (pour que ce repertoire puisse etre interprete comme un package)
104           * entete.py (pour realiser les import necessaires a l'interpretation des catalogues)
105           * declaration_concepts.py (idem)
106         - cree le fichier cata_developpeur.py qui sera par la suite importe
107   """
108   try :
109     if not os.path.isdir(rep_goal) :
110       os.mkdir(rep_goal)
111     #texte_entete = get_entete_cata(fic_cata)
112     texte_entete=""
113     # rep_goal doit contenir les catalogues du developpeur sous la forme *.capy
114     # il faut creer le catalogue developpeur par concatenation de entete,declaration_concepts
115     # et de tous ces fichiers
116     cur_dir = os.getcwd()
117     os.chdir(rep_goal)
118     l_cata_dev = glob.glob('*.capy')
119     if os.path.isfile('cata_developpeur.py') : os.remove('cata_developpeur.py')
120     if len(l_cata_dev) :
121       # des catalogues developpeurs sont effectivement presents : on cree cata_dev.py dans rep_goal
122       str = ''
123       str = str + texte_entete+'\n'
124       for file in l_cata_dev :
125         str = str + open(file,'r').read() +'\n'
126       open('cata_developpeur.py','w+').write(str)
127     os.chdir(cur_dir)
128   except:
129     traceback.print_exc()
130     print ( tr("Impossible de transferer les fichiers requis dans : %s", str(rep_goal)))
131
132 def get_entete_cata(fic_cata):
133   """ Retrouve l'entete du catalogue """
134   l_lignes = open(fic_cata,'r').readlines()
135   txt = ''
136   flag = 0
137   for ligne in l_lignes :
138     if re.match(u"# debut entete",ligne) : flag = 1
139     if re.match(u"# fin entete",ligne) : break
140     if not flag : continue
141     txt = txt + ligne
142   return txt
143