src/fileEnviron.py

   1 #!/usr/bin/env python
   2 #-*- coding:utf-8 -*-
   3 #  Copyright (C) 2010-2013  CEA/DEN
   4 #
   5 #  This library is free software; you can redistribute it and/or
   6 #  modify it under the terms of the GNU Lesser General Public
   7 #  License as published by the Free Software Foundation; either
   8 #  version 2.1 of the License.
   9 #
  10 #  This library is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 #  Lesser General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU Lesser General Public
  16 #  License along with this library; if not, write to the Free Software
  17 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18
  19 import os
  20 import pprint as PP
  21 import src.debug as DBG
  22 import src.architecture
  23 import src.environment
  24
  25 def get_file_environ(output, shell, environ=None):
  26     """Instantiate correct FileEnvironment sub-class.
  27
  28     :param output file: the output file stream.
  29     :param shell str: the type of shell syntax to use.
  30     :param environ dict: a potential additional environment.
  31     """
  32     if environ == None:
  33         environ=src.environment.Environ({})
  34     if shell == "bash":
  35         return BashFileEnviron(output, environ)
  36     if shell == "tcl":
  37         return TclFileEnviron(output, environ)
  38     if shell == "bat":
  39         return BatFileEnviron(output, environ)
  40     if shell == "cfgForPy":
  41         return LauncherFileEnviron(output, environ)
  42     if shell == "cfg":
  43         return ContextFileEnviron(output, environ)
  44     raise Exception("FileEnviron: Unknown shell = %s" % shell)
  45
  46 class FileEnviron(object):
  47     """\
  48     Base class for shell environment
  49     """
  50     def __init__(self, output, environ=None):
  51         """\
  52         Initialization
  53
  54         :param output file: the output file stream.
  55         :param environ dict: SalomeEnviron.
  56         """
  57         self._do_init(output, environ)
  58
  59     def __repr__(self):
  60         """\
  61         easy non exhaustive quick resume for debug print"""
  62         res = {
  63           "output" : self.output,
  64           "environ" : self.environ,
  65         }
  66         return "%s(\n%s\n)" % (self.__class__.__name__, PP.pformat(res))
  67
  68
  69     def _do_init(self, output, environ=None):
  70         """\
  71         Initialization
  72
  73         :param output file: the output file stream.
  74         :param environ dict: a potential additional environment.
  75         """
  76         self.output = output
  77         self.init_path=True # by default we initialise all paths, except PATH
  78         if environ is not None:
  79             self.environ = environ
  80         else:
  81             self.environ = src.environment.Environ({})
  82
  83     def add_line(self, number):
  84         """\
  85         Add some empty lines in the shell file
  86
  87         :param number int: the number of lines to add
  88         """
  89         self.output.write("\n" * number)
  90
  91     def add_comment(self, comment):
  92         """\
  93         Add a comment in the shell file
  94
  95         :param comment str: the comment to add
  96         """
  97         self.output.write("# %s\n" % comment)
  98
  99     def add_echo(self, text):
 100         """\
 101         Add a "echo" in the shell file
 102
 103         :param text str: the text to echo
 104         """
 105         self.output.write('echo %s"\n' % text)
 106
 107     def add_warning(self, warning):
 108         """\
 109         Add a warning "echo" in the shell file
 110
 111         :param warning str: the text to echo
 112         """
 113         self.output.write('echo "WARNING %s"\n' % warning)
 114
 115     def append_value(self, key, value, sep=os.pathsep):
 116         """\
 117         append value to key using sep,
 118         if value contains ":" or ";" then raise error
 119
 120         :param key str: the environment variable to append
 121         :param value str: the value to append to key
 122         :param sep str: the separator string
 123         """
 124         # check that value so no contain the system separator
 125         separator=os.pathsep
 126         if separator in value:
 127             raise Exception("FileEnviron append key '%s' value '%s' contains forbidden character '%s'" % (key, value, separator))
 128         do_append=True
 129         if self.environ.is_defined(key):
 130             value_list = self.environ.get(key).split(sep)
 131             if self.environ._expandvars(value) in value_list:
 132                 do_append=False  # value is already in key path : we don't append it again
 133
 134         if do_append:
 135             self.environ.append_value(key, value,sep)
 136             self.set(key, self.get(key) + sep + value)
 137
 138     def append(self, key, value, sep=os.pathsep):
 139         """\
 140         Same as append_value but the value argument can be a list
 141
 142         :param key str: the environment variable to append
 143         :param value str or list: the value(s) to append to key
 144         :param sep str: the separator string
 145         """
 146         if isinstance(value, list):
 147             for v in value:
 148                 self.append_value(key, v, sep)
 149         else:
 150             self.append_value(key, value, sep)
 151
 152     def prepend_value(self, key, value, sep=os.pathsep):
 153         """\
 154         prepend value to key using sep,
 155         if value contains ":" or ";" then raise error
 156
 157         :param key str: the environment variable to prepend
 158         :param value str: the value to prepend to key
 159         :param sep str: the separator string
 160         """
 161         # check that value so no contain the system separator
 162         separator=os.pathsep
 163         if separator in value:
 164             raise Exception("FileEnviron append key '%s' value '%s' contains forbidden character '%s'" % (key, value, separator))
 165
 166         do_not_prepend=False
 167         if self.environ.is_defined(key):
 168             value_list = self.environ.get(key).split(sep)
 169             exp_val=self.environ._expandvars(value)
 170             if exp_val in value_list:
 171                 do_not_prepend=True
 172         if not do_not_prepend:
 173             self.environ.prepend_value(key, value,sep)
 174             self.set(key, value + sep + self.get(key))
 175
 176     def prepend(self, key, value, sep=os.pathsep):
 177         """\
 178         Same as prepend_value but the value argument can be a list
 179
 180         :param key str: the environment variable to prepend
 181         :param value str or list: the value(s) to prepend to key
 182         :param sep str: the separator string
 183         """
 184         if isinstance(value, list):
 185             for v in reversed(value): # prepend list, first item at last to stay first
 186                 self.prepend_value(key, v, sep)
 187         else:
 188             self.prepend_value(key, value, sep)
 189
 190     def is_defined(self, key):
 191         """\
 192         Check if the key exists in the environment
 193
 194         :param key str: the environment variable to check
 195         """
 196         return self.environ.is_defined(key)
 197
 198     def set(self, key, value):
 199         """\
 200         Set the environment variable 'key' to value 'value'
 201
 202         :param key str: the environment variable to set
 203         :param value str: the value
 204         """
 205         raise NotImplementedError("set is not implement for this shell!")
 206
 207     def get(self, key):
 208         """\
 209         Get the value of the environment variable "key"
 210
 211         :param key str: the environment variable
 212         """
 213         if src.architecture.is_windows():
 214             return '%' + key + '%'
 215         else:
 216             return '${%s}' % key
 217
 218     def get_value(self, key):
 219         """Get the real value of the environment variable "key"
 220         It can help env scripts
 221         :param key str: the environment variable
 222         """
 223         return self.environ.get_value(key)
 224
 225     def finish(self):
 226         """Add a final instruction in the out file (in case of file generation)
 227
 228         :param required bool: Do nothing if required is False
 229         """
 230         return
 231
 232     def set_no_init_path(self):
 233         """Set the no initialisation mode for all paths.
 234            By default only PATH is not reinitialised. All others paths are
 235            (LD_LIBRARY_PATH, PYTHONPATH, ...)
 236            After the call to these function ALL PATHS ARE NOT REINITIALISED.
 237            There initial value is inherited from the environment
 238         """
 239         self.init_path=False
 240
 241     def value_filter(self, value):
 242         res=value
 243         return res
 244
 245
 246 class TclFileEnviron(FileEnviron):
 247     """\
 248     Class for tcl shell.
 249     """
 250     def __init__(self, output, environ=None):
 251         """Initialization
 252
 253         :param output file: the output file stream.
 254         :param environ dict: a potential additional environment.
 255         """
 256         self._do_init(output, environ)
 257         self.output.write(tcl_header.replace("<module_name>",
 258                                              self.environ.get("sat_product_name")))
 259         self.output.write("\nset software %s\n" % self.environ.get("sat_product_name") )
 260         self.output.write("set version %s\n" % self.environ.get("sat_product_version") )
 261         root=os.path.join(self.environ.get("sat_product_base_path"),
 262                                   "apps",
 263                                   self.environ.get("sat_product_base_name"),
 264                                   "$software",
 265                                   "$version")
 266         self.output.write("set root %s\n" % root)
 267         modules_to_load=self.environ.get("sat_product_load_depend")
 268         if len(modules_to_load)>0:
 269             # write module load commands for product dependencies
 270             self.output.write("\n")
 271             for module_to_load in modules_to_load.split(";"):
 272                 self.output.write(module_to_load+"\n")
 273
 274     def set(self, key, value):
 275         """Set the environment variable "key" to value "value"
 276
 277         :param key str: the environment variable to set
 278         :param value str: the value
 279         """
 280         self.output.write('setenv  %s "%s"\n' % (key, value))
 281         self.environ.set(key, value)
 282
 283     def get(self, key):
 284         """\
 285         Get the value of the environment variable "key"
 286
 287         :param key str: the environment variable
 288         """
 289         return self.environ.get(key)
 290
 291     def append_value(self, key, value, sep=os.pathsep):
 292         """append value to key using sep
 293
 294         :param key str: the environment variable to append
 295         :param value str: the value to append to key
 296         :param sep str: the separator string
 297         """
 298         if sep==os.pathsep:
 299             self.output.write('append-path  %s   %s\n' % (key, value))
 300         else:
 301             self.output.write('append-path --delim=\%c %s   %s\n' % (sep, key, value))
 302
 303     def prepend_value(self, key, value, sep=os.pathsep):
 304         """prepend value to key using sep
 305
 306         :param key str: the environment variable to prepend
 307         :param value str: the value to prepend to key
 308         :param sep str: the separator string
 309         """
 310         if sep==os.pathsep:
 311             self.output.write('prepend-path  %s   %s\n' % (key, value))
 312         else:
 313             self.output.write('prepend-path --delim=\%c %s   %s\n' % (sep, key, value))
 314
 315
 316 class BashFileEnviron(FileEnviron):
 317     """\
 318     Class for bash shell.
 319     """
 320     def __init__(self, output, environ=None):
 321         """Initialization
 322
 323         :param output file: the output file stream.
 324         :param environ dict: a potential additional environment.
 325         """
 326         self._do_init(output, environ)
 327         self.output.write(bash_header)
 328
 329     def set(self, key, value):
 330         """Set the environment variable "key" to value "value"
 331
 332         :param key str: the environment variable to set
 333         :param value str: the value
 334         """
 335         self.output.write('export %s="%s"\n' % (key, value))
 336         self.environ.set(key, value)
 337
 338
 339
 340 class BatFileEnviron(FileEnviron):
 341     """\
 342     for Windows batch shell.
 343     """
 344     def __init__(self, output, environ=None):
 345         """Initialization
 346
 347         :param output file: the output file stream.
 348         :param environ dict: a potential additional environment.
 349         """
 350         self._do_init(output, environ)
 351         self.output.write(bat_header)
 352
 353     def add_comment(self, comment):
 354         """Add a comment in the shell file
 355
 356         :param comment str: the comment to add
 357         """
 358         self.output.write("rem %s\n" % comment)
 359
 360     def get(self, key):
 361         """Get the value of the environment variable "key"
 362
 363         :param key str: the environment variable
 364         """
 365         return '%%%s%%' % key
 366
 367     def set(self, key, value):
 368         """Set the environment variable "key" to value "value"
 369
 370         :param key str: the environment variable to set
 371         :param value str: the value
 372         """
 373         self.output.write('set %s=%s\n' % (key, self.value_filter(value)))
 374         self.environ.set(key, value)
 375
 376
 377 class ContextFileEnviron(FileEnviron):
 378     """Class for a salome context configuration file.
 379     """
 380     def __init__(self, output, environ=None):
 381         """Initialization
 382
 383         :param output file: the output file stream.
 384         :param environ dict: a potential additional environment.
 385         """
 386         self._do_init(output, environ)
 387         self.output.write(cfg_header)
 388
 389     def set(self, key, value):
 390         """Set the environment variable "key" to value "value"
 391
 392         :param key str: the environment variable to set
 393         :param value str: the value
 394         """
 395         self.output.write('%s="%s"\n' % (key, value))
 396         self.environ.set(key, value)
 397
 398     def get(self, key):
 399         """Get the value of the environment variable "key"
 400
 401         :param key str: the environment variable
 402         """
 403         return '%({0})s'.format(key)
 404
 405     def add_echo(self, text):
 406         """Add a comment
 407
 408         :param text str: the comment to add
 409         """
 410         self.add_comment(text)
 411
 412     def add_warning(self, warning):
 413         """Add a warning
 414
 415         :param text str: the warning to add
 416         """
 417         self.add_comment("WARNING %s"  % warning)
 418
 419     def prepend_value(self, key, value, sep=os.pathsep):
 420         """prepend value to key using sep
 421
 422         :param key str: the environment variable to prepend
 423         :param value str: the value to prepend to key
 424         :param sep str: the separator string
 425         """
 426         do_append=True
 427         if self.environ.is_defined(key):
 428             value_list = self.environ.get(key).split(sep)
 429             #value cannot be expanded (unlike bash/bat case) - but it doesn't matter.
 430             if value in value_list:
 431                 do_append=False  # value is already in key path : we don't append it again
 432
 433         if do_append:
 434             self.environ.append_value(key, value,sep)
 435             self.output.write('ADD_TO_%s: %s\n' % (key, value))
 436
 437     def append_value(self, key, value, sep=os.pathsep):
 438         """append value to key using sep
 439
 440         :param key str: the environment variable to append
 441         :param value str: the value to append to key
 442         :param sep str: the separator string
 443         """
 444         self.prepend_value(key, value)
 445
 446
 447 class LauncherFileEnviron(FileEnviron):
 448     """\
 449     Class to generate a launcher file script
 450     (in python syntax) SalomeContext API
 451     """
 452     def __init__(self, output, environ=None):
 453         """Initialization
 454
 455         :param output file: the output file stream.
 456         :param environ dict: a potential additional environment.
 457         """
 458         self._do_init(output, environ)
 459         self.python_version=self.environ.get("sat_python_version")
 460         self.bin_kernel_root_dir=self.environ.get("sat_bin_kernel_install_dir")
 461
 462         # four whitespaces for first indentation in a python script
 463         self.indent="    "
 464         self.prefix="context."
 465         self.setVarEnv="setVariable"
 466         self.begin=self.indent+self.prefix
 467
 468         # write the begining of launcher file.
 469         # choose the template version corresponding to python version
 470         # and substitute BIN_KERNEL_INSTALL_DIR (the path to salomeContext.py)
 471         if self.python_version == 2:
 472             launcher_header=launcher_header2
 473         else:
 474             launcher_header=launcher_header3
 475         # in case of Windows OS, Python scripts are not executable.  PyExe ?
 476         if src.architecture.is_windows():
 477             launcher_header = launcher_header.replace("#! /usr/bin/env python3",'')
 478         self.output.write(launcher_header\
 479                           .replace("BIN_KERNEL_INSTALL_DIR", self.bin_kernel_root_dir))
 480
 481         # for these path, we use specialired functions in salomeContext api
 482         self.specialKeys={"PATH": "Path",
 483                           "LD_LIBRARY_PATH": "LdLibraryPath",
 484                           "PYTHONPATH": "PythonPath"}
 485
 486         # we do not want to reinitialise PATH.
 487         # for that we make sure PATH is in self.environ
 488         # and therefore we will not use setVariable for PATH
 489         if not self.environ.is_defined("PATH"):
 490             self.environ.set("PATH","")
 491
 492     def add_echo(self, text):
 493         """Add a comment
 494
 495         :param text str: the comment to add
 496         """
 497         self.output.write('# %s"\n' % text)
 498
 499     def add_warning(self, warning):
 500         """Add a warning
 501
 502         :param text str: the warning to add
 503         """
 504         self.output.write('# "WARNING %s"\n' % warning)
 505
 506     def append_value(self, key, value, sep=os.pathsep):
 507         """append value to key using sep,
 508         if value contains ":" or ";" then raise error
 509
 510         :param key str: the environment variable to prepend
 511         :param value str: the value to prepend to key
 512         :param sep str: the separator string
 513         """
 514         # check that value so no contain the system separator
 515         separator=os.pathsep
 516         msg="LauncherFileEnviron append key '%s' value '%s' contains forbidden character '%s'"
 517         if separator in value:
 518             raise Exception(msg % (key, value, separator))
 519
 520         is_key_defined=self.environ.is_defined(key)
 521         conditional_reinit=False
 522         if (self.init_path and (not is_key_defined)):
 523             # reinitialisation mode set to true (the default)
 524             # for the first occurrence of key, we set it.
 525             # therefore key will not be inherited from environment
 526             self.output.write(self.indent+'if reinitialise_paths:\n'+self.indent)
 527             self.set(key, value)
 528             self.output.write(self.indent+'else:\n'+self.indent)
 529             conditional_reinit=True # in this case do not register value in self.environ a second time
 530
 531         # in all other cases we use append (except if value is already the key
 532         do_append=True
 533         if is_key_defined:
 534             value_list = self.environ.get(key).split(sep)
 535             # rem : value cannot be expanded (unlike bash/bat case) - but it doesn't matter.
 536             if value in value_list:
 537                 do_append=False  # value is already in key path : we don't append it again
 538
 539         if do_append:
 540             if not conditional_reinit:
 541                 self.environ.append_value(key, value,sep) # register value in self.environ
 542             if key in self.specialKeys.keys():
 543                 #for these special keys we use the specific salomeContext function
 544                 self.output.write(self.begin+'addTo%s(r"%s")\n' %
 545                                   (self.specialKeys[key], self.value_filter(value)))
 546             else:
 547                 # else we use the general salomeContext addToVariable function
 548                 self.output.write(self.begin+'appendVariable(r"%s", r"%s",separator="%s")\n'
 549                                   % (key, self.value_filter(value), sep))
 550
 551     def append(self, key, value, sep=":"):
 552         """Same as append_value but the value argument can be a list
 553
 554         :param key str: the environment variable to append
 555         :param value str or list: the value(s) to append to key
 556         :param sep str: the separator string
 557         """
 558         if isinstance(value, list):
 559             for v in value:
 560                 self.append_value(key, v, sep)
 561         else:
 562             self.append_value(key, value, sep)
 563
 564     def prepend_value(self, key, value, sep=os.pathsep):
 565         """prepend value to key using sep,
 566         if value contains ":" or ";" then raise error
 567
 568         :param key str: the environment variable to prepend
 569         :param value str: the value to prepend to key
 570         :param sep str: the separator string
 571         """
 572         # check that value so no contain the system separator
 573         separator=os.pathsep
 574         msg="LauncherFileEnviron append key '%s' value '%s' contains forbidden character '%s'"
 575         if separator in value:
 576             raise Exception(msg % (key, value, separator))
 577
 578         is_key_defined=self.environ.is_defined(key)
 579         conditional_reinit=False
 580         if (self.init_path and (not is_key_defined)):
 581             # reinitialisation mode set to true (the default)
 582             # for the first occurrence of key, we set it.
 583             # therefore key will not be inherited from environment
 584             self.output.write(self.indent+'if reinitialise_paths:\n'+self.indent)
 585             self.set(key, value)
 586             self.output.write(self.indent+'else:\n'+self.indent)
 587             conditional_reinit=True # in this case do not register value in self.environ a second time
 588
 589         # in all other cases we use append (except if value is already the key
 590         do_append=True
 591         if is_key_defined:
 592             value_list = self.environ.get(key).split(sep)
 593             # rem : value cannot be expanded (unlike bash/bat case) - but it doesn't matter.
 594             if value in value_list:
 595                 do_append=False  # value is already in key path : we don't append it again
 596
 597         if do_append:
 598             if not conditional_reinit:
 599                 self.environ.append_value(key, value,sep) # register value in self.environ
 600             if key in self.specialKeys.keys():
 601                 #for these special keys we use the specific salomeContext function
 602                 self.output.write(self.begin+'addTo%s(r"%s")\n' %
 603                                   (self.specialKeys[key], self.value_filter(value)))
 604             else:
 605                 # else we use the general salomeContext addToVariable function
 606                 self.output.write(self.begin+'addToVariable(r"%s", r"%s",separator="%s")\n'
 607                                   % (key, self.value_filter(value), sep))
 608
 609
 610     def prepend(self, key, value, sep=":"):
 611         """Same as prepend_value but the value argument can be a list
 612
 613         :param key str: the environment variable to prepend
 614         :param value str or list: the value(s) to prepend to key
 615         :param sep str: the separator string
 616         """
 617         if isinstance(value, list):
 618             for v in value:
 619                 self.prepend_value(key, v, sep)
 620         else:
 621             self.prepend_value(key, value, sep)
 622
 623
 624     def set(self, key, value):
 625         """Set the environment variable "key" to value "value"
 626
 627         :param key str: the environment variable to set
 628         :param value str: the value
 629         """
 630         self.output.write(self.begin+self.setVarEnv+
 631                           '(r"%s", r"%s", overwrite=True)\n' %
 632                           (key, self.value_filter(value)))
 633         self.environ.set(key,value)
 634
 635
 636     def add_comment(self, comment):
 637         # Special comment in case of the DISTENE licence
 638         if comment=="DISTENE license":
 639             self.output.write(self.indent+
 640                               "#"+
 641                               self.prefix+
 642                               self.setVarEnv+
 643                               '(r"%s", r"%s", overwrite=True)\n' %
 644                               ('DISTENE_LICENSE_FILE', 'Use global envvar: DLIM8VAR'))
 645             self.output.write(self.indent+
 646                               "#"+
 647                               self.prefix+
 648                               self.setVarEnv+
 649                               '(r"%s", r"%s", overwrite=True)\n' %
 650                               ('DLIM8VAR', '<your licence>'))
 651             return
 652         if "setting environ for" in comment:
 653             self.output.write(self.indent+"#[%s]\n" %
 654                               comment.split("setting environ for ")[1])
 655             return
 656
 657         self.output.write(self.indent+"# %s\n" % comment)
 658
 659     def finish(self):
 660         """\
 661         Add a final instruction in the out file (in case of file generation)
 662         In the particular launcher case, do nothing
 663
 664         :param required bool: Do nothing if required is False
 665         """
 666         if self.python_version == 2:
 667             launcher_tail=launcher_tail_py2
 668         else:
 669             launcher_tail=launcher_tail_py3
 670         self.output.write(launcher_tail)
 671         return
 672
 673 class ScreenEnviron(FileEnviron):
 674     def __init__(self, output, environ=None):
 675         self._do_init(output, environ)
 676         self.defined = {}
 677
 678     def add_line(self, number):
 679         pass
 680
 681     def add_comment(self, comment):
 682         pass
 683
 684     def add_echo(self, text):
 685         pass
 686
 687     def add_warning(self, warning):
 688         pass
 689
 690     def write(self, command, name, value, sign="="):
 691         import src
 692         self.output.write("  %s%s %s %s %s\n" % \
 693             (src.printcolors.printcLabel(command),
 694              " " * (12 - len(command)),
 695              src.printcolors.printcInfo(name), sign, value))
 696
 697     def is_defined(self, name):
 698         return name in self.defined
 699
 700     def get(self, name):
 701         return "${%s}" % name
 702
 703     def set(self, name, value):
 704         self.write("set", name, value)
 705         self.defined[name] = value
 706
 707     def prepend(self, name, value, sep=":"):
 708         if isinstance(value, list):
 709             value = sep.join(value)
 710         value = value + sep + self.get(name)
 711         self.write("prepend", name, value)
 712
 713     def append(self, name, value, sep=":"):
 714         if isinstance(value, list):
 715             value = sep.join(value)
 716         value = self.get(name) + sep + value
 717         self.write("append", name, value)
 718
 719     def run_env_script(self, module, script):
 720         self.write("load", script, "", sign="")
 721
 722
 723 #
 724 #  Headers
 725 #
 726 bat_header="""\
 727 @echo off
 728
 729 rem The following variables are used only in case of a sat package
 730 set out_dir_Path=%~dp0
 731 """
 732
 733 tcl_header="""\
 734 #%Module -*- tcl -*-
 735 #
 736 # <module_name> module for use with 'environment-modules' package
 737 #
 738 """
 739
 740 bash_header="""\
 741 #!/bin/bash
 742 if [ "$BASH" = "" ]
 743 then
 744   # check that the user is not using another shell
 745   echo
 746   echo "Warning! SALOME environment not initialized"
 747   echo "You must run this script in a bash shell."
 748   echo "As you are using another shell. Please first run: bash"
 749   echo
 750 fi
 751 ##########################################################################
 752 #
 753 # This line is used only in case of a sat package
 754 export out_dir_Path=$(cd $(dirname ${BASH_SOURCE[0]});pwd)
 755
 756 ###########################################################################
 757 """
 758
 759 cfg_header="""\
 760 [SALOME Configuration]
 761 """
 762
 763 launcher_header2="""\
 764 #! /usr/bin/env python
 765
 766 ################################################################
 767 # WARNING: this file is automatically generated by SalomeTools #
 768 # WARNING: and so could be overwritten at any time.            #
 769 ################################################################
 770
 771 import os
 772 import sys
 773 import subprocess
 774 import os.path
 775
 776 # Add the pwdPath to able to run the launcher after unpacking a package
 777 # Used only in case of a salomeTools package
 778 out_dir_Path=os.path.dirname(os.path.realpath(__file__))
 779
 780 # Preliminary work to initialize path to SALOME Python modules
 781 def __initialize():
 782
 783   sys.path[:0] = [ r'BIN_KERNEL_INSTALL_DIR' ]  # to get salomeContext
 784
 785   # define folder to store omniorb config (initially in virtual application folder)
 786   try:
 787     from salomeContextUtils import setOmniOrbUserPath
 788     setOmniOrbUserPath()
 789   except Exception as e:
 790     print(e)
 791     sys.exit(1)
 792 # End of preliminary work
 793
 794 # salome doc only works for virtual applications. Therefore we overwrite it with this function
 795 def _showDoc(modules):
 796     for module in modules:
 797       modulePath = os.getenv(module+"_ROOT_DIR")
 798       if modulePath != None:
 799         baseDir = os.path.join(modulePath, "share", "doc", "salome")
 800         docfile = os.path.join(baseDir, "gui", module.upper(), "index.html")
 801         if not os.path.isfile(docfile):
 802           docfile = os.path.join(baseDir, "tui", module.upper(), "index.html")
 803         if not os.path.isfile(docfile):
 804           docfile = os.path.join(baseDir, "dev", module.upper(), "index.html")
 805         if os.path.isfile(docfile):
 806           out, err = subprocess.Popen(["xdg-open", docfile]).communicate()
 807         else:
 808           print ("Online documentation is not accessible for module:", module)
 809       else:
 810         print (module+"_ROOT_DIR not found!")
 811
 812 def main(args):
 813   # Identify application path then locate configuration files
 814   __initialize()
 815
 816   if args == ['--help']:
 817     from salomeContext import usage
 818     usage()
 819     sys.exit(0)
 820
 821
 822   # Create a SalomeContext which parses configFileNames to initialize environment
 823   try:
 824     from salomeContext import SalomeContext, SalomeContextException
 825     if 'appendVariable' not in dir(SalomeContext):
 826       # check whether the appendVariable method is implemented
 827       def appendVariable(self, name, value, separator=os.pathsep):
 828         if value == '':
 829           return
 830         value = os.path.expandvars(value) # expand environment variables
 831         env = os.getenv(name, None)
 832         if env is None:
 833           os.environ[name] = value
 834         else:
 835           os.environ[name] = env + separator + value
 836         return
 837       SalomeContext.appendVariable = appendVariable
 838
 839     context = SalomeContext(None)
 840
 841     # Here set specific variables, if needed
 842     # context.addToPath('mypath')
 843     # context.addToLdLibraryPath('myldlibrarypath')
 844     # context.addToPythonPath('mypythonpath')
 845     # context.setVariable('myvarname', 'value')
 846
 847     # Logger level error
 848     context.getLogger().setLevel(40)
 849 """
 850
 851 launcher_header3="""\
 852 #! /usr/bin/env python3
 853
 854 ################################################################
 855 # WARNING: this file is automatically generated by SalomeTools #
 856 # WARNING: and so could be overwritten at any time.            #
 857 ################################################################
 858
 859 import os
 860 import sys
 861 import subprocess
 862 import os.path
 863
 864 # Add the pwdPath to able to run the launcher after unpacking a package
 865 # Used only in case of a salomeTools package
 866 out_dir_Path=os.path.dirname(os.path.realpath(__file__))
 867
 868 # Preliminary work to initialize path to SALOME Python modules
 869 def __initialize():
 870
 871   sys.path[:0] = [ r'BIN_KERNEL_INSTALL_DIR' ]
 872
 873   # define folder to store omniorb config (initially in virtual application folder)
 874   try:
 875     from salomeContextUtils import setOmniOrbUserPath
 876     setOmniOrbUserPath()
 877   except Exception as e:
 878     print(e)
 879     sys.exit(1)
 880 # End of preliminary work
 881
 882 # salome doc only works for virtual applications. Therefore we overwrite it with this function
 883 def _showDoc(modules):
 884     for module in modules:
 885       modulePath = os.getenv(module+"_ROOT_DIR")
 886       if modulePath != None:
 887         baseDir = os.path.join(modulePath, "share", "doc", "salome")
 888         docfile = os.path.join(baseDir, "gui", module.upper(), "index.html")
 889         if not os.path.isfile(docfile):
 890           docfile = os.path.join(baseDir, "tui", module.upper(), "index.html")
 891         if not os.path.isfile(docfile):
 892           docfile = os.path.join(baseDir, "dev", module.upper(), "index.html")
 893         if os.path.isfile(docfile):
 894           out, err = subprocess.Popen(["xdg-open", docfile]).communicate()
 895         else:
 896           print("Online documentation is not accessible for module:", module)
 897       else:
 898         print(module+"_ROOT_DIR not found!")
 899
 900 def main(args):
 901   # Identify application path then locate configuration files
 902   __initialize()
 903
 904   if '--help' in args:
 905     from salomeContext import usage
 906     usage()
 907     sys.exit(0)
 908
 909   reinitialise_paths=True
 910   if '--keep-paths' in args:
 911     reinitialise_paths=False
 912     args.remove('--keep-paths')
 913
 914   #from salomeContextUtils import getConfigFileNames
 915   #configFileNames, args, unexisting = getConfigFileNames( args, checkExistence=True )
 916   #if len(unexisting) > 0:
 917   #  print("ERROR: unexisting configuration file(s): " + ', '.join(unexisting))
 918   #  sys.exit(1)
 919
 920   # Create a SalomeContext which parses configFileNames to initialize environment
 921   try:
 922     from salomeContext import SalomeContext, SalomeContextException
 923     if 'appendVariable' not in dir(SalomeContext):
 924       # check whether the appendVariable method is implemented
 925       def appendVariable(self, name, value, separator=os.pathsep):
 926         if value == '':
 927           return
 928         value = os.path.expandvars(value) # expand environment variables
 929         env = os.getenv(name, None)
 930         if env is None:
 931           os.environ[name] = value
 932         else:
 933           os.environ[name] = env + separator + value
 934         return
 935       SalomeContext.appendVariable = appendVariable
 936
 937     context = SalomeContext(None)
 938     # Here set specific variables, if needed
 939     # context.addToPath('mypath')
 940     # context.addToLdLibraryPath('myldlibrarypath')
 941     # context.addToPythonPath('mypythonpath')
 942     # context.setVariable('myvarname', 'value')
 943
 944     # Logger level error
 945     context.getLogger().setLevel(40)
 946 """
 947
 948 launcher_tail_py2="""\
 949     #[hook to integrate in launcher additionnal user modules]
 950
 951     # Load all files extra.env.d/*.py and call the module's init routine]
 952
 953     extradir=out_dir_Path + r"/extra.env.d"
 954
 955     if os.path.exists(extradir):
 956         import imp
 957         sys.path.insert(0, os.path.join(os.getcwd(), extradir))
 958         for filename in sorted(
 959             filter(lambda x: os.path.isfile(os.path.join(extradir, x)),
 960                    os.listdir(extradir))):
 961
 962             if filename.endswith(".py"):
 963                 f = os.path.join(extradir, filename)
 964                 module_name = os.path.splitext(os.path.basename(f))[0]
 965                 fp, path, desc = imp.find_module(module_name)
 966                 module = imp.load_module(module_name, fp, path, desc)
 967                 module.init(context, out_dir_Path)
 968
 969     #[manage salome doc command]
 970     if len(args) >1 and args[0]=='doc':
 971         _showDoc(args[1:])
 972         return
 973
 974     # Start SALOME, parsing command line arguments
 975     out, err, status = context.runSalome(args)
 976     sys.exit(status)
 977
 978   except SalomeContextException, e:
 979     import logging
 980     logging.getLogger("salome").error(e)
 981     sys.exit(1)
 982
 983 if __name__ == "__main__":
 984   args = sys.argv[1:]
 985   main(args)
 986 #
 987 """
 988
 989 launcher_tail_py3="""\
 990     #[hook to integrate in launcher additionnal user modules]
 991
 992     # Load all files extra.env.d/*.py and call the module's init routine]
 993
 994     extradir=out_dir_Path + r"/extra.env.d"
 995
 996     if os.path.exists(extradir):
 997         import importlib
 998         import importlib.util
 999         sys.path.insert(0, os.path.join(os.getcwd(), extradir))
1000         for filename in sorted(
1001             filter(lambda x: os.path.isfile(os.path.join(extradir, x)),
1002                    os.listdir(extradir))):
1003
1004             if filename.endswith(".py"):
1005                 f = os.path.join(extradir, filename)
1006                 module_name = os.path.splitext(os.path.basename(f))[0]
1007                 _specs = importlib.util.find_spec(module_name)
1008                 _module = importlib.util.module_from_spec(_specs)
1009                 _specs.loader.exec_module(_module)
1010                 _module.init(context, out_dir_Path)
1011     #[manage salome doc command]
1012     if len(args) >1 and args[0]=='doc':
1013         _showDoc(args[1:])
1014         return
1015
1016     # Start SALOME, parsing command line arguments
1017     out, err, status = context.runSalome(args)
1018     sys.exit(status)
1019
1020   except SalomeContextException as e:
1021     import logging
1022     logging.getLogger("salome").error(e)
1023     sys.exit(1)
1024
1025
1026 if __name__ == "__main__":
1027   args = sys.argv[1:]
1028   main(args)
1029 #
1030 """
1031
1032