2 # -*- coding: iso-8859-1 -*-
3 # Copyright (C) 2007-2010 CEA/DEN, EDF R&D, OPEN CASCADE
5 # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
6 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
22 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
25 import sys, os, string, glob, time, pickle
27 from launchConfigureParser import verbose
29 # this file is extraction of set_env from runSalome.py
30 # for reusage in automated tests
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"
38 # -----------------------------------------------------------------------------
40 def add_path(directory, variable_name):
41 """Function helper to add environment variables"""
42 if sys.platform == "win32":
46 if not os.environ.has_key(variable_name):
47 os.environ[variable_name] = ""
49 if os.path.exists(directory):
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):
59 if os.path.abspath(_dir) != os.path.abspath(directory):
63 newpath[:0] = [ directory ]
64 newpath = string.join(newpath, splitsym)
65 os.environ[variable_name] = newpath
66 if variable_name == "PYTHONPATH":
67 sys.path[:0] = [directory]
69 # -----------------------------------------------------------------------------
74 if __lib__dir__: return __lib__dir__
79 # -----------------------------------------------------------------------------
81 def get_config(silent=False):
83 Get list of modules, paths.
85 Read args from launch configure xml file and command line options.
86 Check variables <module>_ROOT_DIR and set list of used modules.
87 Return args, modules_list, modules_root_dir
90 # read args from launch configure xml file and command line options
92 #*** Test additional option
94 #*** help_str = "Test options addition."
95 #*** o_j = optparse.Option("-j", "--join", action="store_true", dest="join", help=help_str)
97 import launchConfigureParser
98 args = launchConfigureParser.get_env()
100 #*** Test additional option
101 #*** args = launchConfigureParser.get_env([o_j])
102 #*** if args.has_key("join"): print args["join"]
104 # Check variables <module>_ROOT_DIR
105 # and set list of used modules (without KERNEL)
108 if args.has_key("modules"):
109 modules_list += args["modules"]
110 # KERNEL must be last in the list to locate it at the first place in PATH
112 modules_list[:0] = ["GUI"]
113 modules_list[:0] = ["KERNEL"]
114 modules_list.reverse()
116 modules_root_dir = {}
119 for module in modules_list :
120 module_variable=module+"_ROOT_DIR"
121 if not os.environ.has_key(module_variable):
123 print "*******************************************************"
125 print "* Environment variable",module_variable,"must be set"
126 print "* Module", module, "will be not available"
128 print "********************************************************"
130 to_remove_list.append(module)
133 module_root_dir = os.environ[module_variable]
134 modules_root_dir[module]=module_root_dir
136 for to_remove in to_remove_list:
137 modules_list.remove(to_remove)
139 while "KERNEL" in modules_list:
140 modules_list.remove("KERNEL")
143 while "GUI" in modules_list:
144 modules_list.remove("GUI")
147 return args, modules_list, modules_root_dir
149 # -----------------------------------------------------------------------------
151 def set_env(args, modules_list, modules_root_dir, silent=False):
152 """Add to the PATH-variables modules specific paths"""
155 from salome_utils import getTmpDir, generateFileName, makeTmpDir, getPortNumber
157 # create temporary directory for environment files needed by modules from the list
158 port = getPortNumber(False)
160 tmp_dir = getTmpDir()
161 env_dir = generateFileName(tmp_dir, prefix="env", with_port=True)
165 python_version="python%d.%d" % sys.version_info[0:2]
166 modules_root_dir_list = []
167 if os.getenv('SALOME_BATCH') == None:
168 os.putenv('SALOME_BATCH','0')
170 modules_list = modules_list[:] + ["GUI"]
171 modules_list = modules_list[:] + ["KERNEL"]
172 for module in modules_list :
173 if modules_root_dir.has_key(module):
174 module_root_dir = modules_root_dir[module]
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),
180 add_path(os.path.join(module_root_dir,get_lib_dir(),salome_subdir),
182 add_path(os.path.join(module_root_dir,"bin",salome_subdir),
184 if os.path.exists(module_root_dir + "/examples") :
185 add_path(os.path.join(module_root_dir,"examples"),
188 add_path(os.path.join(module_root_dir,"bin",salome_subdir),
190 # add lib before site-packages to load script instead of dll if any (win32 platform)
191 add_path(os.path.join(module_root_dir,get_lib_dir(),salome_subdir),
193 add_path(os.path.join(module_root_dir,get_lib_dir(),
194 python_version,"site-packages",
198 if platform.machine() == "x86_64":
199 add_path(os.path.join(module_root_dir,"lib64",
200 python_version,"site-packages",
204 add_path(os.path.join(module_root_dir,get_lib_dir(),
205 python_version,"site-packages",
210 # set environment by modules from the list
213 mod=__import__(module.lower()+"_setenv")
222 if sys.platform == 'win32':
223 os.environ["SALOMEPATH"]=";".join(modules_root_dir_list)
225 os.environ["SALOMEPATH"]=":".join(modules_root_dir_list)
227 # set trace environment variable
229 if not os.environ.has_key("SALOME_trace"):
230 os.environ["SALOME_trace"]="local"
232 os.environ["SALOME_trace"]="file:"+args['file'][0]
234 os.environ["SALOME_trace"]="with_logger"
236 # set resources variables if not yet set
238 os.environ["CSF_SALOMEDS_ResourcesDefaults"] \
239 = os.path.join(modules_root_dir["KERNEL"],"share",
240 salome_subdir,"resources","kernel")
242 # -----------------------------------------------------------------------------
244 def main(silent=False):
245 args, modules_list, modules_root_dir = get_config(silent=silent)
246 set_env(args, modules_list, modules_root_dir, silent=silent)
249 # -----------------------------------------------------------------------------
251 if __name__ == "__main__":