geom/bin/geom_setenv.py

#! /usr/bin/env python
#  -*- coding: iso-8859-1 -*-
# Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
#
# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
#

import os, sys, string
from salome_utils import getTmpDir, generateFileName, uniteFiles
from setenv import add_path, get_lib_dir, salome_subdir

# -----------------------------------------------------------------------------

def set_env( args ):
    """Add to the PATH-variables modules specific paths"""
    psep = os.pathsep
    python_version="python%d.%d" % sys.version_info[0:2]
    
    tmp_dir = getTmpDir()
    env_dir = generateFileName( tmp_dir, prefix="env", with_port=True )
    res_dir = os.path.join( os.getenv( "GEOM_ROOT_DIR" ), "share", salome_subdir, "resources", "geom" )

    csf_list = ["Plugin", "GEOMDS_Resources", "ShHealing"]

    for csf_file in csf_list:
       uniteFiles( os.path.join( res_dir, csf_file ), os.path.join( env_dir, csf_file ) )
       pass

    for csf_string in csf_list:
        csf_var = "CSF_" + csf_string + "Defaults"
        if not os.getenv( csf_var ):
            os.environ[ csf_var ] = env_dir
            pass
        pass

    # find plugins
    plugin_list = []
    resource_path_list = []
    plugins_dir_var = "GEOM_ENGINE_RESOURCES_DIR"
    if os.environ.has_key(plugins_dir_var):
        # reverse the user's paths list, because the used 'add_path' prepends a new path,
        # but we want to append it: [a, b, c] => [c, b, a]
        plugins_dirs = os.environ[plugins_dir_var].split(os.pathsep)
        plugins_dirs.reverse()
        os.environ[plugins_dir_var] = string.join(plugins_dirs, os.pathsep)
        pass
    for env_var in os.environ.keys():
        value = os.environ[env_var]
        if env_var[-9:] == "_ROOT_DIR" and value:
            plugin_root = value
            plugin = env_var[:-9] # plugin name may have wrong case

            # look for NAMEOFPlugin.xml file among resource files
            resource_dir = os.path.join(plugin_root,"share",salome_subdir,"resources",plugin.lower())
            if not os.access( resource_dir, os.F_OK ): continue
            for resource_file in os.listdir( resource_dir ):
                if resource_file.endswith( ".xml") and \
                   resource_file.lower() == plugin.lower() + ".xml":
                    # use "name" attribute of "geom-plugin" as name of plugin in a right case
                    from xml.dom.minidom import parse
                    try:
                        xml_doc = parse( os.path.join( resource_dir, resource_file ))
                        plugin_nodes = xml_doc.getElementsByTagName("geom-plugin")
                    except:
                        continue
                    if not plugin_nodes or not plugin_nodes[0].hasAttribute("name"): continue
                    plugin = plugin_nodes[0].getAttribute("name")
                    if plugin in plugin_list: continue

                    # add paths of plugin
                    plugin_list.append(plugin)
                    if not os.environ.has_key("SALOME_"+plugin+"Resources"):
                        resource_path = os.path.join(plugin_root,"share",salome_subdir,"resources",plugin.lower())
                        os.environ["SALOME_"+plugin+"Resources"] = resource_path
                        resource_path_list.append( resource_path )
                        add_path(os.path.join(plugin_root,get_lib_dir(),python_version, "site-packages",salome_subdir), "PYTHONPATH")
                        add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "PYTHONPATH")

                        if sys.platform == "win32":
                            add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "PATH")
                            add_path(os.path.join(plugin_root,"bin",salome_subdir), "PYTHONPATH")
                        else:
                            add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "LD_LIBRARY_PATH")
                            add_path(os.path.join(plugin_root,"bin",salome_subdir), "PYTHONPATH")
                            add_path(os.path.join(plugin_root,"bin",salome_subdir), "PATH")
                            pass
                        pass
                    pass
                elif resource_file == "ImportExport" and plugin.upper() != "GEOM":
                    # add 'ImportExport' plugin file path into variable
                    add_path(resource_dir, plugins_dir_var)
                    # add plugin's library path into environment
                    if sys.platform == "win32":
                        add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "PATH")
                    else:
                        add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "LD_LIBRARY_PATH")
                        pass
                    pass
                pass
    plugin_list.append("GEOMActions")
    os.environ["GEOM_PluginsList"] = ":".join(plugin_list)
    os.environ["SalomeAppConfig"] = os.environ["SalomeAppConfig"] + psep + psep.join(resource_path_list)

    if os.environ.has_key(plugins_dir_var):
        # reverse back the plugin paths:
        # [f, e, d, c, b, a] => [a, b, c, d, e, f]
        plugins_dirs = os.environ[plugins_dir_var].split(os.pathsep)
        plugins_dirs.reverse()
        os.environ[plugins_dir_var] = string.join(plugins_dirs, os.pathsep)
        pass
    pass