Salome HOME
PN
[tools/eficas.git] / Editeur / utils.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 des utilitaires divers
23 """
24 import os,re
25 import glob
26 import traceback
27 import codecs,types
28
29 def substract_list(liste1,liste2):
30   """ 
31       Enlève tous les éléments de liste2 présents dans liste1 et retourne liste1
32   """
33   for item in liste2:
34     try:
35       liste1.remove(item)
36     except:
37       pass
38   return liste1
39
40 def get_rep_user():
41   """
42       Détermine sur quelle plate-forme s'exécute Eficas et recherche
43       le répertoire de l'utilisateur /$home/Eficas_install
44   """
45   if os.name not in ('posix','nt'):
46     print "Système non reconnu par Eficas"
47     print "Prévenir la maintenance"
48     sys.exit(0)
49   if os.name == 'nt':
50     try:
51       drive = os.environ['HOMEDRIVE']
52       nom_user = os.environ['USERNAME']
53       rep_user_eficas = drive+'\\'+nom_user+'\\'+'Eficas_install'
54     except:
55       rep_user_eficas = os.path.join('C:','Eficas_install')
56   else :
57     rep_user_eficas = os.path.join(os.environ['HOME'],'Eficas_install')
58   if os.path.exists(rep_user_eficas):
59     if os.path.isfile(rep_user_eficas) :
60       print "Un fichier de nom %s existe déjà : impossible de créer un répertoire de même nom" %rep_user_eficas
61       sys.exit(0)
62   else :
63     try:
64       os.mkdir(rep_user_eficas)
65     except:
66       print "Création du répertoire %s impossible\n Vérifiez vos droits d'accès" %rep_user_eficas
67   return rep_user_eficas
68
69 def read_file(file):
70   """
71       ouvre le fichier file et retourne son contenu
72       si pbe retourne None
73   """
74   try :
75     f=open(file)
76     text=f.read()
77     f.close()
78     return text
79   except:
80     return None
81
82 def save_in_file(file,text,dir=None):
83   """
84       crée le fichier file (ou l'écrase s'il existe) et écrit text dedans
85       retourne 1 si OK 0 sinon
86   """
87   try :
88       import string
89       #file=string.split(file,"/")[-1]
90       if dir != None:
91          os.chdir(dir)
92       f=open(file,'w')
93       f.write(text)
94       f.close()
95       return 1
96   except:
97       return 0
98
99 def extension_fichier(pathAndFile):
100     """ Return ext if path/filename.ext is given """
101     return os.path.splitext(pathAndFile)[1][1:]
102
103 def stripPath(pathAndFile):
104     """ Return filename.ext if path/filename.ext is given """
105     return os.path.split(pathAndFile)[1]
106
107 def init_rep_cata_dev(fic_cata,rep_goal):
108   """ 
109       Initialise le répertoire des catalogues développeurs (chemin d'accès donné
110       dans le fichier eficas.ini cad :
111         - le crée s'il n'existe pas encore
112         - copie dedans les 3 fichiers nécessaires :
113           * __init__.py (pour que ce répertoire puisse être interprété comme un package)
114           * entete.py (pour réaliser les import nécessaires à l'interprétation des catalogues)
115           * declaration_concepts.py (idem)
116         - crée le fichier cata_developpeur.py qui sera par la suite importé
117   """
118   try :
119     if not os.path.isdir(rep_goal) :
120       os.mkdir(rep_goal)
121     #texte_entete = get_entete_cata(fic_cata)
122     texte_entete=""
123     # rep_goal doit contenir les catalogues du développeur sous la forme *.capy
124     # il faut créer le catalogue développeur par concaténation de entete,declaration_concepts
125     # et de tous ces fichiers
126     cur_dir = os.getcwd()
127     os.chdir(rep_goal)
128     l_cata_dev = glob.glob('*.capy')
129     if os.path.isfile('cata_developpeur.py') : os.remove('cata_developpeur.py')
130     if len(l_cata_dev) :
131       # des catalogues développeurs sont effectivement présents : on crée cata_dev.py dans rep_goal
132       str = ''
133       str = str + texte_entete+'\n'
134       for file in l_cata_dev :
135         str = str + open(file,'r').read() +'\n'
136       open('cata_developpeur.py','w+').write(str)
137     os.chdir(cur_dir)
138   except:
139     traceback.print_exc()
140     print "Impossible de transférer les fichiers requis dans :",rep_goal
141
142 def get_entete_cata(fic_cata):
143   """ Retrouve l'entete du catalogue """
144   l_lignes = open(fic_cata,'r').readlines()
145   txt = ''
146   flag = 0
147   for ligne in l_lignes :
148     if re.match("# debut entete",ligne) : flag = 1
149     if re.match("# fin entete",ligne) : break
150     if not flag : continue
151     txt = txt + ligne
152   return txt
153