69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
# $Id$
|
|
|
|
import os,sys,string,re,subprocess,shlex
|
|
|
|
class Environment():
|
|
__slots__ = [ \
|
|
'rootRelation',
|
|
'pathInfo',
|
|
]
|
|
|
|
def __init__(self,rootRelation = '.'):
|
|
self.rootRelation = rootRelation
|
|
self.pathInfo = {}
|
|
self.get_pathInfo()
|
|
|
|
def relPath(self,relative = '.'):
|
|
return os.path.join(self.rootDir(),relative)
|
|
|
|
def rootDir(self):
|
|
damask_root = os.getenv('DAMASK_ROOT')
|
|
if damask_root == '' or damask_root == None: # env not set
|
|
if sys.argv[0] == '': # no caller path
|
|
cwd = os.getcwd()
|
|
else:
|
|
cwd = {False: os.path.dirname(sys.argv[0]),
|
|
True: sys.argv[0]}[os.path.isdir(sys.argv[0])]
|
|
damask_root = os.path.normpath(os.path.join(os.path.realpath(cwd),self.rootRelation))
|
|
|
|
return damask_root
|
|
|
|
def binDir(self):
|
|
damask_bin = os.getenv('DAMASK_BIN')
|
|
if damask_bin == '' or damask_bin == None: damask_bin = self.relPath('bin/')
|
|
return damask_bin
|
|
|
|
def get_pathInfo(self):
|
|
try: # check for user-defined pathinfo
|
|
file = open(self.relPath('lib/pathinfo'))
|
|
content = map(lambda string: string.strip(),file.readlines())
|
|
file.close()
|
|
for line in content:
|
|
if not (line.startswith('#') or line == ''):
|
|
items = line.split() + ['','']
|
|
if items[1] == '': # nothing specified
|
|
self.pathInfo[items[0].lower()] = ''
|
|
elif items[1].startswith(('/','$')): # absolute path specified ($shellVar is considered absolute)
|
|
self.pathInfo[items[0].lower()] = items[1]
|
|
else: # path relative to DAMASK_ROOT/lib
|
|
self.pathInfo[items[0].lower()] = os.path.normpath(os.path.join(self.relPath('lib/'),items[1]))
|
|
except:
|
|
pass
|
|
|
|
def isAvailable(self,software,noNeeded=-1):
|
|
licensesNeeded = {'abaqus' :5,
|
|
'standard':5,
|
|
'explicit':5}
|
|
if noNeeded == -1: noNeeded = licensesNeeded[software]
|
|
cmd = """ ssh mulicense2 "/Stat_Flexlm | grep 'Users of %s: ' | cut -d' ' -f7,13" """%software
|
|
process = subprocess.Popen(shlex.split(cmd),stdout = subprocess.PIPE,stderr = subprocess.PIPE)
|
|
licenses = map(int, process.stdout.readline().split())
|
|
try:
|
|
if licenses[0]-licenses[1] >= noNeeded:
|
|
return 0
|
|
else:
|
|
print(licenses[1] + noNeeded - licenses[0], 'missing licenses for %s'%software)
|
|
return licenses[1] + noNeeded - licenses[0]
|
|
except IndexError:
|
|
print('Could not retrieve license information for %s'%software)
|
|
return 127 |