336 lines
16 KiB
Python
Executable File
336 lines
16 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
|
|
import re
|
|
import os
|
|
import sys
|
|
import string
|
|
import subprocess
|
|
import optparse
|
|
|
|
|
|
reCoriolisPattern = re.compile( r".*coriolis.*" )
|
|
reReleaseSharedPattern = re.compile( r".*Release\.Shared.*" )
|
|
reReleaseStaticPattern = re.compile( r".*Release\.Static.*" )
|
|
reDebugSharedPattern = re.compile( r".*Debug\.Shared.*" )
|
|
reDebugStaticPattern = re.compile( r".*Debug\.Static.*" )
|
|
|
|
|
|
def scrubPath ( pathName ):
|
|
pathEnv = os.getenv( pathName )
|
|
if not pathEnv: return ""
|
|
pathList = pathEnv.split( ':' )
|
|
scrubbedList = []
|
|
for pathElement in pathList:
|
|
if reCoriolisPattern .match(pathElement) \
|
|
or reReleaseSharedPattern.match(pathElement) \
|
|
or reReleaseStaticPattern.match(pathElement) \
|
|
or reDebugSharedPattern .match(pathElement) \
|
|
or reDebugStaticPattern .match(pathElement):
|
|
continue
|
|
scrubbedList += [ pathElement ]
|
|
if len(scrubbedList) == 0: return ""
|
|
scrubbedEnv = scrubbedList[0]
|
|
for pathElement in scrubbedList[1:]:
|
|
scrubbedEnv += ":" + pathElement
|
|
return scrubbedEnv
|
|
|
|
|
|
def guessOs ():
|
|
useDevtoolset = False
|
|
osEL9 = re.compile (".*Linux.*el9.*x86_64.*")
|
|
osSlsoc7x_64 = re.compile (".*Linux.*el7.*x86_64.*")
|
|
osSlsoc6x_64 = re.compile (".*Linux.*el6.*x86_64.*")
|
|
osSlsoc6x = re.compile (".*Linux.*(el|slsoc)6.*")
|
|
osSLSoC5x_64 = re.compile (".*Linux.*el5.*x86_64.*")
|
|
osSLSoC5x = re.compile (".*Linux.*(el5|2.6.23.13.*SoC).*")
|
|
osFedora_64 = re.compile (".*Linux.*fc.*x86_64.*")
|
|
osFedora = re.compile (".*Linux.*fc.*")
|
|
osLinux_64 = re.compile (".*Linux.*x86_64.*")
|
|
osLinux = re.compile (".*Linux.*")
|
|
osDarwin = re.compile (".*Darwin.*")
|
|
osUbuntu1004 = re.compile (".*Linux.*ubuntu.*")
|
|
osUbuntu1004_64 = re.compile (".*Linux.*ubuntu.*x86_64.*")
|
|
osFreeBSD8x_amd64 = re.compile (".*FreeBSD 8.*amd64.*")
|
|
osFreeBSD8x_64 = re.compile (".*FreeBSD 8.*x86_64.*")
|
|
osFreeBSD8x = re.compile (".*FreeBSD 8.*")
|
|
osCygwinW7_64 = re.compile (".*CYGWIN_NT-6\.1.*x86_64.*")
|
|
osCygwinW7 = re.compile (".*CYGWIN_NT-6\.1.*i686.*")
|
|
osCygwinW8_64 = re.compile (".*CYGWIN_NT-6\.[2-3].*x86_64.*")
|
|
osCygwinW8 = re.compile (".*CYGWIN_NT-6\.[2-3].*i686.*")
|
|
osCygwinW10_64 = re.compile (".*CYGWIN_NT-10\.[0-3].*x86_64.*")
|
|
osCygwinW10 = re.compile (".*CYGWIN_NT-10\.[0-3].*i686.*")
|
|
|
|
uname = subprocess.Popen ( ["uname", "-srm"], stdout=subprocess.PIPE )
|
|
lines = uname.stdout.readlines()
|
|
|
|
line = lines[0].decode( 'ascii' )
|
|
if osSlsoc7x_64.match(line): osType = "Linux.el7_64"
|
|
elif osEL9.match(line): osType = "Linux.el9"
|
|
elif osSlsoc6x_64.match(line):
|
|
osType = "Linux.slsoc6x_64"
|
|
useDevtoolset = True
|
|
elif osSlsoc6x.match(line):
|
|
osType = "Linux.slsoc6x"
|
|
useDevtoolset = True
|
|
elif osSLSoC5x_64 .match(line): osType = "Linux.SLSoC5x_64"
|
|
elif osSLSoC5x .match(line): osType = "Linux.SLSoC5x"
|
|
elif osFedora_64 .match(line): osType = "Linux.fc_64"
|
|
elif osFedora .match(line): osType = "Linux.fc"
|
|
elif osUbuntu1004 .match(line): osType = "Linux.Ubuntu1004"
|
|
elif osUbuntu1004_64 .match(line): osType = "Linux.Ubuntu1004_64"
|
|
elif osLinux_64 .match(line): osType = "Linux.x86_64"
|
|
elif osLinux .match(line): osType = "Linux.i386"
|
|
elif osFreeBSD8x_64 .match(line): osType = "FreeBSD.8x.x86_64"
|
|
elif osFreeBSD8x_amd64.match(line): osType = "FreeBSD.8x.amd64"
|
|
elif osFreeBSD8x .match(line): osType = "FreeBSD.8x.i386"
|
|
elif osDarwin .match(line): osType = "Darwin"
|
|
elif osCygwinW7_64 .match(line): osType = "Cygwin.W7_64"
|
|
elif osCygwinW7 .match(line): osType = "Cygwin.W7"
|
|
elif osCygwinW8_64 .match(line): osType = "Cygwin.W8_64"
|
|
elif osCygwinW8 .match(line): osType = "Cygwin.W8"
|
|
elif osCygwinW10_64 .match(line): osType = "Cygwin.W10_64"
|
|
elif osCygwinW10 .match(line): osType = "Cygwin.W10"
|
|
else:
|
|
uname = subprocess.Popen ( ["uname", "-sr"], stdout=subprocess.PIPE )
|
|
osType = uname.stdout.readlines()[0][:-1]
|
|
|
|
print( "[WARNING] Unrecognized OS: \"{}\".".format( line[:-1] ))
|
|
print( " (using: \"{}\")".format( osType ))
|
|
ldLibraryPath = os.getenv('LD_LIBRARY_PATH')
|
|
if ldLibraryPath and 'devtoolset' in ldLibraryPath: useDevtoolset = False
|
|
return ( osType, useDevtoolset )
|
|
|
|
|
|
def guessShell ( forcedShell ):
|
|
# This environement variable cannot be trusted as it is set once when
|
|
# the user logs in. If aftewards it changes it that variable is *not*
|
|
# affected :-(.
|
|
#if 'SHELL' in os.environ: return os.environ['SHELL']
|
|
isBourneShell = True
|
|
cshBins = [ u'/usr/bin/tcsh'
|
|
, u'/bin/tcsh'
|
|
, u'/usr/pkg/bin/tcsh'
|
|
, u'/usr/local/bin/tcsh'
|
|
, u'/usr/bin/csh'
|
|
, u'/bin/csh'
|
|
, u'/usr/pkg/bin/csh'
|
|
, u'/usr/local/bin/csh'
|
|
]
|
|
if shellName is None:
|
|
psCommand = subprocess.Popen ( ['ps', '-p', str(os.getppid()) ], stdout=subprocess.PIPE )
|
|
shell = psCommand.stdout.readlines()[1].decode('utf8')[:-1].split()[3].lstrip('-')
|
|
whichCommand = subprocess.Popen ( ['which', shell ], stdout=subprocess.PIPE )
|
|
shellPath = whichCommand.stdout.readlines()[0][:-1].decode('utf8')
|
|
#print( 'GUESSED shellPath={}'.format(shellPath) )
|
|
else:
|
|
shellPath = forcedShell
|
|
#print( 'FORCED shellPath={}'.format(shellPath) )
|
|
if shellPath in cshBins:
|
|
#print( 'Matched C-Shell' )
|
|
isBourneShell = False
|
|
#print( 'isBourneShell={}\n'.format(isBourneShell) )
|
|
return shellPath, isBourneShell
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
osType,useDevtoolset = guessOs()
|
|
buildType = "Release"
|
|
linkType = "Shared"
|
|
rootDir = None
|
|
shellName = None
|
|
|
|
parser = optparse.OptionParser()
|
|
# Build relateds.
|
|
parser.add_option ( "--query-inst-root", action="store_true" , dest="queryInstRoot" )
|
|
parser.add_option ( "--query-isys-root", action="store_true" , dest="queryISysRoot" )
|
|
parser.add_option ( "--release" , action="store_true" , dest="release" )
|
|
parser.add_option ( "--debug" , action="store_true" , dest="debug" )
|
|
parser.add_option ( "--devel" , action="store_true" , dest="devel" )
|
|
parser.add_option ( "--static" , action="store_true" , dest="static" )
|
|
parser.add_option ( "--shared" , action="store_true" , dest="shared" )
|
|
parser.add_option ( "--no-python" , action="store_true" , dest="nopython" )
|
|
parser.add_option ( "--root" , action="store" , type="string", dest="rootDir" )
|
|
parser.add_option ( "--remove" , action="store_true" , dest="remove" )
|
|
parser.add_option ( "--shell" , action="store" , type="string", dest="shell" )
|
|
( options, args ) = parser.parse_args()
|
|
|
|
if options.release: buildType = "Release"
|
|
if options.debug: buildType = "Debug"
|
|
if options.devel: buildType = "Debug"
|
|
if options.static: linkType = "Static"
|
|
if options.shared: linkType = "Shared"
|
|
if options.rootDir: rootDir = options.rootDir
|
|
if options.shell: shellName = options.shell
|
|
shellBin, isBourneShell = guessShell( shellName )
|
|
|
|
scriptPath = os.path.abspath( os.path.dirname( sys.argv[0] ))
|
|
if 'Debug.' in scriptPath: buildType = 'Debug'
|
|
if not shellBin:
|
|
print( 'echo "[ERROR] coriolisEnv.py was not able to guess/find the current shell interpeter."' )
|
|
sys.exit( 1 )
|
|
strippedPath = scrubPath( "PATH" )
|
|
strippedLibraryPath = scrubPath( "LD_LIBRARY_PATH" )
|
|
strippedPythonPath = scrubPath( "PYTHONPATH" )
|
|
if options.remove:
|
|
shellScript = 'echo "Removing Coriolis environment";'
|
|
if osType == "Darwin":
|
|
ldVar = 'DYLD_LIBRARY_PATH'
|
|
else:
|
|
ldVar = 'LD_LIBRARY_PATH'
|
|
if isBourneShell:
|
|
shellScript += 'export PATH={};hash -r;'.format(strippedPath)
|
|
shellScript += 'BOOTSTRAP_TOP="";CORIOLIS_TOP="";export -n BOOTSTRAP_TOP CORIOLIS_TOP;'
|
|
if strippedLibraryPath:
|
|
shellScript += 'export {}={};'.format(ldVar, strippedLibraryPath)
|
|
else:
|
|
shellScript += '{0}=""; export -n {0};'.format(ldVar)
|
|
else:
|
|
shellScript += 'setenv PATH {};rehash;'.format(strippedPath)
|
|
shellScript += 'unsetenv BOOTSTRAP_TOP CORIOLIS_TOP;'
|
|
if strippedLibraryPath:
|
|
shellScript += 'setenv {} {};'.format(ldVar, strippedLibraryPath)
|
|
else:
|
|
shellScript += 'unsetenv {};'.format(ldVar)
|
|
print( shellScript )
|
|
sys.exit(0)
|
|
|
|
|
|
shellScriptSh = \
|
|
'echo "Issuing commands for Bourne shell like interpreters";' \
|
|
'echo "%(MESSAGE)s";' \
|
|
'echo "Switching to Coriolis 2.x (%(buildDir)s)";' \
|
|
'PATH="%(PATH)s";' \
|
|
'BOOTSTRAP_TOP="%(BOOTSTRAP_TOP)s";' \
|
|
'CORIOLIS_TOP="%(CORIOLIS_TOP)s";' \
|
|
'export PATH BOOTSTRAP_TOP CORIOLIS_TOP STRATUS_MAPPING_NAME;'
|
|
|
|
# 'STRATUS_MAPPING_NAME="%(SYSCONF_DIR)s/stratus2sxlib.xml";' \
|
|
|
|
shellScriptCsh = \
|
|
'echo "Issuing commands for C-shell like interpreters";' \
|
|
'echo "%(MESSAGE)s";' \
|
|
'echo "Switching to Coriolis 2.x (%(buildDir)s)";' \
|
|
'setenv PATH "%(PATH)s";' \
|
|
'setenv BOOTSTRAP_TOP "%(BOOTSTRAP_TOP)s";' \
|
|
'setenv CORIOLIS_TOP "%(CORIOLIS_TOP)s";'
|
|
|
|
# 'setenv STRATUS_MAPPING_NAME "%(SYSCONF_DIR)s/stratus2sxlib.xml";' \
|
|
|
|
reDevtoolset = re.compile( r'/opt/rh/devtoolset-(?P<version>\d+)/root/etc/coriolis2.*' )
|
|
|
|
buildDir = buildType + "." + linkType
|
|
scriptDir = os.path.dirname ( os.path.abspath(__file__) )
|
|
#print( "echo \"Script Location: {}\";".format( scriptDir ))
|
|
if scriptDir.startswith("/etc/coriolis2"):
|
|
coriolisTop = "/usr"
|
|
sysconfDir = scriptDir
|
|
shellMessage = "Using system-wide Coriolis 2 (/usr)"
|
|
else:
|
|
m = reDevtoolset.match( scriptDir )
|
|
if m:
|
|
coriolisTop = "/opt/rh/devtoolset-%d/root/usr" % m.group('version')
|
|
sysconfDir = scriptDir
|
|
shellMessage = "Using system-wide devtoolset-%(v)d Coriolis 2 (/opt/rh/devtoolset-%(v)d/root/usr)" \
|
|
% { 'v':m.group('version') }
|
|
elif scriptDir.startswith(os.getenv("HOME")+"/nightly/coriolis-2.x/"):
|
|
rootDir = os.getenv("HOME") + "/nightly/coriolis-2.x"
|
|
coriolisTop = "%s/%s/%s/install" % ( rootDir, osType, buildDir )
|
|
sysconfDir = scriptDir
|
|
shellMessage = "Using Nightly build Coriolis 2 (%s)" % coriolisTop
|
|
elif scriptDir.startswith("/users/outil/coriolis/coriolis-2.x/") \
|
|
or scriptDir.startswith("/soc/coriolis2/"):
|
|
coriolisTop = "/soc/coriolis2"
|
|
sysconfDir = coriolisTop + "/etc/coriolis2"
|
|
shellMessage = "Using SoC network-wide Coriolis 2 (/soc/coriolis2)"
|
|
else:
|
|
if not rootDir:
|
|
scriptRoot = '/'.join( scriptDir.split('/')[:-2] )
|
|
if not os.path.exists(scriptRoot):
|
|
rootDir = os.getenv("HOME") + "/coriolis-2.x"
|
|
coriolisTop = "%s/%s/%s/install" % ( rootDir, osType, buildDir )
|
|
sysconfDir = coriolisTop + "/etc/coriolis2"
|
|
shellMessage = "Using user-selected Coriolis 2 (%s)" % rootDir
|
|
else:
|
|
rootDir = scriptRoot
|
|
coriolisTop = rootDir
|
|
sysconfDir = coriolisTop + "/etc/coriolis2"
|
|
shellMessage = "Using script location Coriolis 2 (%s)" % rootDir
|
|
|
|
if osType.startswith("Cygwin"):
|
|
strippedPath = "%s/lib:%s" % ( coriolisTop, libDir, strippedPath )
|
|
if not os.path.exists(coriolisTop):
|
|
print( 'echo "[ERROR] coriolisEnv.py, top directory "{}" do not exists."'.format( coriolisTop ))
|
|
sys.exit( 1 )
|
|
for lib in [ 'lib64', 'lib' ]:
|
|
libDir = lib
|
|
absLibDir = '{0}/{1}'.format( coriolisTop, lib )
|
|
if os.path.isdir(absLibDir): break
|
|
libDir = None
|
|
if libDir is None:
|
|
print( 'echo "[ERROR] coriolisEnv.py, library directory not found."' )
|
|
sys.exit( 1 )
|
|
|
|
strippedPath = "%s/bin:%s" % ( coriolisTop, strippedPath )
|
|
strippedLibraryPath = "%s:%s" % ( absLibDir , strippedLibraryPath )
|
|
if not options.nopython:
|
|
pyVersion = sys.version_info
|
|
version = "%d.%d" % (pyVersion[0],pyVersion[1])
|
|
sitePackagesDir = "sitePackageDir_has_been_not_found"
|
|
for pyPackageDir in [ "%s/python%s.%s/site-packages" % (absLibDir,pyVersion[0],pyVersion[1])
|
|
, "%s/python%s.%s/dist-packages" % (absLibDir,pyVersion[0],pyVersion[1])
|
|
, "%s/%s.%s/site-packages" % (absLibDir,pyVersion[0],pyVersion[1])
|
|
, "%s/python%s/site-packages" % (absLibDir,pyVersion[0])
|
|
, "%s/python%s/dist-packages" % (absLibDir,pyVersion[0])
|
|
, "%s/%s/site-packages" % (absLibDir,pyVersion[0])
|
|
]:
|
|
if os.path.isdir(pyPackageDir):
|
|
sitePackagesDir = pyPackageDir
|
|
break
|
|
strippedPythonPath = "%s:" % (sitePackagesDir) + strippedPythonPath
|
|
strippedPythonPath = "%s/crlcore:" % (sitePackagesDir) + strippedPythonPath
|
|
strippedPythonPath = "%s/cumulus:" % (sitePackagesDir) + strippedPythonPath
|
|
strippedPythonPath = "%s/cumulus/plugins:" % (sitePackagesDir) + strippedPythonPath
|
|
strippedPythonPath = "%s/stratus:" % (sitePackagesDir) + strippedPythonPath
|
|
strippedPythonPath = "%s:" % (sysconfDir) + strippedPythonPath
|
|
shellScriptSh += 'PYTHONPATH="%(PYTHONPATH)s";' \
|
|
'export PYTHONPATH;'
|
|
shellScriptCsh += 'setenv PYTHONPATH "%(PYTHONPATH)s";'
|
|
if osType == "Darwin":
|
|
shellScriptSh += 'DYLD_LIBRARY_PATH="%(LD_LIBRARY_PATH)s";' \
|
|
'export DYLD_LIBRARY_PATH;'
|
|
shellScriptCsh += 'setenv DYLD_LIBRARY_PATH "%(LD_LIBRARY_PATH)s";'
|
|
else:
|
|
shellScriptSh += 'LD_LIBRARY_PATH="%(LD_LIBRARY_PATH)s";' \
|
|
'export LD_LIBRARY_PATH;'
|
|
shellScriptCsh += 'setenv LD_LIBRARY_PATH "%(LD_LIBRARY_PATH)s";'
|
|
|
|
shellScriptSh += "hash -r;"
|
|
shellScriptCsh += "rehash;"
|
|
if isBourneShell: shellScript = shellScriptSh
|
|
else: shellScript = shellScriptCsh
|
|
if useDevtoolset:
|
|
shellScript += \
|
|
' echo "Launching a devtoolset-8 subshell though scl (CTRL+D to exit).";' \
|
|
' scl enable devtoolset-8 %(SHELL)s'
|
|
|
|
evalScript = shellScript % { "PATH" : strippedPath
|
|
, "LD_LIBRARY_PATH" : strippedLibraryPath
|
|
, "PYTHONPATH" : strippedPythonPath
|
|
, "BOOTSTRAP_TOP" : coriolisTop
|
|
, "CORIOLIS_TOP" : coriolisTop
|
|
, "SYSCONF_DIR" : sysconfDir
|
|
, "MESSAGE" : shellMessage
|
|
, "buildDir" : buildDir
|
|
, 'SHELL' : shellBin
|
|
}
|
|
if options.queryISysRoot:
|
|
print( '{}/{}'.format( rootDir, osType ))
|
|
sys.exit( 0 )
|
|
if options.queryInstRoot:
|
|
print( coriolisTop )
|
|
sys.exit( 0 )
|
|
print( evalScript )
|
|
sys.exit( 0 )
|