]> SALOME platform Git repositories - modules/kernel.git/blob - bin/setenv.py
Salome HOME
Python3 porting: change python executable to 'python3'.
[modules/kernel.git] / bin / setenv.py
1 #!/usr/bin/env python3
2 #  -*- coding: iso-8859-1 -*-
3 # Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
4 #
5 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
6 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
7 #
8 # This library is free software; you can redistribute it and/or
9 # modify it under the terms of the GNU Lesser General Public
10 # License as published by the Free Software Foundation; either
11 # version 2.1 of the License, or (at your option) any later version.
12 #
13 # This library is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16 # Lesser General Public License for more details.
17 #
18 # You should have received a copy of the GNU Lesser General Public
19 # License along with this library; if not, write to the Free Software
20 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
21 #
22 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #
24
25 import sys, os, string, glob, time, pickle
26 import orbmodule
27 from launchConfigureParser import verbose
28
29 # this file is extraction of set_env from runSalome.py
30 # for reusage in automated tests
31
32 # salome_subdir variable is used for composing paths like $KERNEL_ROOT_DIR/share/salome/resources, etc.
33 # before moving to SUIT-based gui, instead of salome_subdir there was args['appname'] used.
34 # but after - 'appname'  = "SalomeApp", so using it in making the subdirectory is an error.
35 salome_subdir = "salome"
36
37
38 # -----------------------------------------------------------------------------
39
40 def add_path(directory, variable_name):
41     """Function helper to add environment variables"""
42     if sys.platform == "win32":
43       splitsym = ";"
44     else:
45       splitsym = ":"
46     if variable_name not in os.environ:
47         os.environ[variable_name] = ""
48         pass
49     if os.path.exists(directory):
50         newpath=[]
51         for _dir in os.environ[variable_name].split(splitsym):
52             if os.path.exists(_dir):
53                 if sys.platform != "win32":
54                     if not os.path.samefile(_dir, directory):
55                         newpath.append(_dir)
56                 else:
57                     newpath.append(_dir)
58             else:
59                 if os.path.abspath(_dir) != os.path.abspath(directory):
60                   newpath.append(_dir)
61             pass
62         newpath[:0] = [ directory ]
63         newpath = splitsym.join(newpath)
64         os.environ[variable_name] = newpath
65         if variable_name == "PYTHONPATH":
66             sys.path[:0] = [os.path.realpath(directory)]
67
68 # -----------------------------------------------------------------------------
69
70 __lib__dir__ = None
71 def get_lib_dir():
72     global __lib__dir__
73     if __lib__dir__: return __lib__dir__
74     import platform
75     __lib__dir__ = "lib"
76     return __lib__dir__
77
78 # -----------------------------------------------------------------------------
79
80 def get_config(silent=False, exeName=None):
81     """
82     Get list of modules, paths.
83
84     Read args from launch configure xml file and command line options.
85     Check variables <module>_ROOT_DIR and set list of used modules.
86     Return args, modules_list, modules_root_dir
87     """
88
89     # read args from launch configure xml file and command line options
90
91     #*** Test additional option
92     #*** import optparse
93     #*** help_str = "Test options addition."
94     #*** o_j = optparse.Option("-j", "--join", action="store_true", dest="join", help=help_str)
95
96     import launchConfigureParser
97     args = launchConfigureParser.get_env(exeName=exeName)
98
99     #*** Test additional option
100     #*** args = launchConfigureParser.get_env([o_j])
101     #*** if args.has_key("join"): print(args["join"])
102
103     # Check variables <module>_ROOT_DIR
104     # and set list of used modules (without KERNEL)
105
106     modules_list = []
107     if "modules" in args:
108         modules_list += args["modules"]
109     # KERNEL must be last in the list to locate it at the first place in PATH
110     if args["gui"] :
111         modules_list[:0] = ["GUI"]
112     modules_list[:0] = ["KERNEL"]
113     modules_list.reverse()
114
115     modules_root_dir = {}
116
117     to_remove_list=[]
118     for module in modules_list :
119         module_variable=module+"_ROOT_DIR"
120         if module_variable not in os.environ:
121             if not silent:
122                 print("*******************************************************")
123                 print("*")
124                 print("* Environment variable",module_variable,"must be set")
125                 print("* Module", module, "will be not available")
126                 print("*")
127                 print("********************************************************")
128                 pass
129             to_remove_list.append(module)
130             continue
131             pass
132         module_root_dir = os.environ[module_variable]
133         modules_root_dir[module]=module_root_dir
134
135     for to_remove in to_remove_list:
136         modules_list.remove(to_remove)
137
138     while "KERNEL" in modules_list:
139         modules_list.remove("KERNEL")
140         pass
141
142     while "GUI" in modules_list:
143         modules_list.remove("GUI")
144         pass
145
146     return args, modules_list, modules_root_dir
147
148 # -----------------------------------------------------------------------------
149
150 def set_env(args, modules_list, modules_root_dir, silent=False):
151     """Add to the PATH-variables modules specific paths"""
152
153     import os
154     from salome_utils import getLogDir, generateFileName, makeTmpDir, getPortNumber
155
156     # create temporary directory for environment files needed by modules from the list
157     port = getPortNumber(False)
158     if port:
159         tmp_dir = getLogDir()
160         env_dir = generateFileName(tmp_dir, prefix="env", with_port=True)
161         makeTmpDir(env_dir)
162         pass
163
164     python_version="python%d.%d" % sys.version_info[0:2]
165     modules_root_dir_list = []
166     if os.getenv('SALOME_BATCH') == None:
167       os.putenv('SALOME_BATCH','0')
168     if args["gui"] :
169         modules_list = modules_list[:] + ["GUI"]
170     modules_list = modules_list[:] + ["KERNEL"]
171     for module in modules_list :
172         if module in modules_root_dir:
173             module_root_dir = modules_root_dir[module]
174             if module_root_dir not in modules_root_dir_list:
175               modules_root_dir_list[:0] = [module_root_dir]
176             if sys.platform == "win32":
177               add_path(os.path.join(module_root_dir,get_lib_dir(),salome_subdir),
178                      "PATH")
179             elif sys.platform == "darwin":
180               add_path(os.path.join(module_root_dir,get_lib_dir(),salome_subdir),
181                      "DYLD_LIBRARY_PATH")
182             else:
183               add_path(os.path.join(module_root_dir,get_lib_dir(),salome_subdir),
184                      "LD_LIBRARY_PATH")
185             add_path(os.path.join(module_root_dir,"bin",salome_subdir),
186                      "PATH")
187             if os.path.exists(os.path.join(module_root_dir, "examples")):
188                 add_path(os.path.join(module_root_dir, "examples"),
189                          "PYTHONPATH")
190                 pass
191             add_path(os.path.join(module_root_dir,"bin",salome_subdir),
192                      "PYTHONPATH")
193             # add lib before site-packages to load script instead of dll if any (win32 platform)
194             add_path(os.path.join(module_root_dir,get_lib_dir(),salome_subdir),
195                      "PYTHONPATH")
196             add_path(os.path.join(module_root_dir,get_lib_dir(),
197                                   python_version,"site-packages",
198                                   salome_subdir),
199                      "PYTHONPATH")
200             import platform
201             if platform.machine() == "x86_64":
202                 add_path(os.path.join(module_root_dir,"lib64",
203                                       python_version,"site-packages",
204                                       salome_subdir),
205                          "PYTHONPATH")
206                 pass
207             add_path(os.path.join(module_root_dir,get_lib_dir(),
208                                   python_version,"site-packages",
209                                   salome_subdir,
210                                   "shared_modules"),
211                      "PYTHONPATH")
212
213             # set environment by modules from the list
214             if port:
215                 try:
216                     mod=__import__(module.lower()+"_setenv")
217                     mod.set_env(args)
218                     pass
219                 except:
220                     pass
221                 pass
222             pass
223         pass
224
225     if sys.platform == 'win32':
226         os.environ["SALOMEPATH"]=";".join(modules_root_dir_list)
227     else:
228         os.environ["SALOMEPATH"]=":".join(modules_root_dir_list)
229
230     # set trace environment variable
231
232     if "SALOME_trace" not in os.environ:
233         os.environ["SALOME_trace"]="local"
234     if args['file']:
235         os.environ["SALOME_trace"] = "file:" + args['file'][0]
236     if args['logger']:
237         os.environ["SALOME_trace"]="with_logger"
238
239     # set resources variables if not yet set
240
241     os.environ["CSF_SALOMEDS_ResourcesDefaults"] \
242     = os.path.join(modules_root_dir["KERNEL"],"share",
243                    salome_subdir,"resources","kernel")
244
245 # -----------------------------------------------------------------------------
246
247 def main(silent=False, exeName=None):
248     args, modules_list, modules_root_dir = get_config(silent=silent, exeName=exeName)
249     set_env(args, modules_list, modules_root_dir, silent=silent)
250     return args
251
252 # -----------------------------------------------------------------------------
253
254 if __name__ == "__main__":
255    import user
256    args = main()