Salome HOME
PN : pour les clefs documentaires
[tools/eficas.git] / Editeur / utils.py
1 #            CONFIGURATION MANAGEMENT OF EDF VERSION
2 # ======================================================================
3 # COPYRIGHT (C) 1991 - 2002  EDF R&D                  WWW.CODE-ASTER.ORG
4 # THIS PROGRAM IS FREE SOFTWARE; YOU CAN REDISTRIBUTE IT AND/OR MODIFY
5 # IT UNDER THE TERMS OF THE GNU GENERAL PUBLIC LICENSE AS PUBLISHED BY
6 # THE FREE SOFTWARE FOUNDATION; EITHER VERSION 2 OF THE LICENSE, OR
7 # (AT YOUR OPTION) ANY LATER VERSION.
8 #
9 # THIS PROGRAM IS DISTRIBUTED IN THE HOPE THAT IT WILL BE USEFUL, BUT
10 # WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED WARRANTY OF
11 # MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. SEE THE GNU
12 # GENERAL PUBLIC LICENSE FOR MORE DETAILS.
13 #
14 # YOU SHOULD HAVE RECEIVED A COPY OF THE GNU GENERAL PUBLIC LICENSE
15 # ALONG WITH THIS PROGRAM; IF NOT, WRITE TO EDF R&D CODE_ASTER,
16 #    1 AVENUE DU GENERAL DE GAULLE, 92141 CLAMART CEDEX, FRANCE.
17 #
18 #
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 def substract_list(liste1,liste2):
29   """ 
30       Enlève tous les éléments de liste2 présents dans liste1 et retourne liste1
31   """
32   for item in liste2:
33     try:
34       liste1.remove(item)
35     except:
36       pass
37   return liste1
38
39 def get_rep_user():
40   """
41       Détermine sur quelle plate-forme s'exécute Eficas et recherche
42       le répertoire de l'utilisateur /$home/Eficas_install
43   """
44   if os.name not in ('posix','nt'):
45     print "Système non reconnu par Eficas"
46     print "Prévenir la maintenance"
47     sys.exit(0)
48   if os.name == 'nt':
49     try:
50       drive = os.environ['HOMEDRIVE']
51       nom_user = os.environ['USERNAME']
52       rep_user_eficas = drive+'\\'+nom_user+'\\'+'Eficas_install'
53     except:
54       rep_user_eficas = os.path.join('C:','Eficas_install')
55   else :
56     rep_user_eficas = os.path.join(os.environ['HOME'],'Eficas_install')
57   if os.path.exists(rep_user_eficas):
58     if os.path.isfile(rep_user_eficas) :
59       print "Un fichier de nom %s existe déjà : impossible de créer un répertoire de même nom" %rep_user_eficas
60       sys.exit(0)
61   else :
62     try:
63       os.mkdir(rep_user_eficas)
64     except:
65       print "Création du répertoire %s impossible\n Vérifiez vos droits d'accès" %rep_user_eficas
66   return rep_user_eficas
67
68 def read_file(file):
69   """
70       ouvre le fichier file et retourne son contenu
71       si pbe retourne None
72   """
73   try :
74     f=open(file)
75     text=f.read()
76     f.close()
77     return text
78   except:
79     return None
80
81 def save_in_file(file,text):
82   """
83       crée le fichier file (ou l'écrase s'il existe) et écrit text dedans
84       retourne 1 si OK 0 sinon
85   """
86   try :
87       f=open(file,'w')
88       f.write(text)
89       f.close()
90       return 1
91   except:
92       return 0
93
94 def extension_fichier(pathAndFile):
95     """ Return ext if path/filename.ext is given """
96     return os.path.splitext(pathAndFile)[1][1:]
97
98 def stripPath(pathAndFile):
99     """ Return filename.ext if path/filename.ext is given """
100     return os.path.split(pathAndFile)[1]
101
102 def init_rep_cata_dev(fic_cata,rep_goal):
103   """ 
104       Initialise le répertoire des catalogues développeurs (chemin d'accès donné
105       dans le fichier eficas.ini cad :
106         - le crée s'il n'existe pas encore
107         - copie dedans les 3 fichiers nécessaires :
108           * __init__.py (pour que ce répertoire puisse être interprété comme un package)
109           * entete.py (pour réaliser les import nécessaires à l'interprétation des catalogues)
110           * declaration_concepts.py (idem)
111         - crée le fichier cata_developpeur.py qui sera par la suite importé
112   """
113   try :
114     if not os.path.isdir(rep_goal) :
115       os.mkdir(rep_goal)
116     texte_entete = get_entete_cata(fic_cata)
117     # rep_goal doit contenir les catalogues du développeur sous la forme *.capy
118     # il faut créer le catalogue développeur par concaténation de entete,declaration_concepts
119     # et de tous ces fichiers
120     cur_dir = os.getcwd()
121     os.chdir(rep_goal)
122     l_cata_dev = glob.glob('*.capy')
123     if os.path.isfile('cata_developpeur.py') : os.remove('cata_developpeur.py')
124     if len(l_cata_dev) :
125       # des catalogues développeurs sont effectivement présents : on crée cata_dev.py dans rep_goal
126       str = ''
127       str = str + texte_entete+'\n'
128       for file in l_cata_dev :
129         str = str + open(file,'r').read() +'\n'
130       open('cata_developpeur.py','w+').write(str)
131     os.chdir(cur_dir)
132   except:
133     traceback.print_exc()
134     print "Impossible de transférer les fichiers requis dans :",rep_goal
135
136 def get_entete_cata(fic_cata):
137   """ Retrouve l'entete du catalogue """
138   l_lignes = open(fic_cata,'r').readlines()
139   txt = ''
140   flag = 0
141   for ligne in l_lignes :
142     if re.match("# debut entete",ligne) : flag = 1
143     if re.match("# fin entete",ligne) : break
144     if not flag : continue
145     txt = txt + ligne
146   return txt
147